test_print_tests.py

import os

import mozunit
import pytest

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()