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