summaryrefslogtreecommitdiffstats
path: root/testing/raptor/test/test_print_tests.py
blob: 53f759fa2725dd97c531bfd89e27f1296a466821 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
import os

import mozunit
import pytest

# need this so raptor imports work both from /raptor and via mach
here = os.path.abspath(os.path.dirname(__file__))
from raptor import cmdline


def test_pageload_subtests(capsys, monkeypatch, tmpdir):
    # cmdline.py is hard-coded to use raptor.toml from the same directory so we need
    # to monkey patch os.dirname, which is not ideal. If we could make --print-tests
    # respect the --test path that would be much better.
    def mock(path):
        return str(tmpdir)

    monkeypatch.setattr(os.path, "dirname", mock)
    manifest = tmpdir.join("raptor.toml")
    manifest.write(
        """
[DEFAULT]
type = "pageload"
apps = "firefox"

["raptor-subtest-1"]
measure = ["foo", "bar"]

["raptor-subtest-2"]
"""
    )
    with pytest.raises(SystemExit):
        cmdline.parse_args(["--print-tests"])
    captured = capsys.readouterr()
    assert (
        captured.out
        == """
Raptor Tests Available for Firefox Desktop
==========================================

raptor
  type: pageload
  subtests:
    raptor-subtest-1 (foo, bar)
    raptor-subtest-2

Done.
"""
    )


if __name__ == "__main__":
    mozunit.main()