summaryrefslogtreecommitdiffstats
path: root/testing/raptor/test/test_print_tests.py
blob: 739d3b0d6c52e09b3a5ca4af88ce289cba33f12a (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.ini 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.ini")
    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()