import os
import os.path
import sys
from subprocess import check_call

import mozinstall
import mozunit
import pytest
from moztest.selftest import fixtures

MOZ_AUTOMATION = bool(os.getenv("MOZ_AUTOMATION", "0") == "1")


# skip tsan while ubuntu 18.04 is the test image, for intermittent startup crashes
@pytest.mark.skip_mozinfo("tsan")
def test_grizzly_smoke():
    ffbin = fixtures.binary()

    ffbin = ffbin.replace("$MOZ_FETCHES_DIR", os.getenv("MOZ_FETCHES_DIR", "")).strip(
        '"'
    )

    if "Contents/MacOS/firefox" in ffbin and MOZ_AUTOMATION:
        mozinstall.install(
            os.path.join(os.getenv("MOZ_FETCHES_DIR", ""), "target.dmg"),
            os.getenv("MOZ_FETCHES_DIR", ""),
        )

    if MOZ_AUTOMATION:
        assert os.path.exists(
            ffbin
        ), "Missing Firefox build. Build it, or set GECKO_BINARY_PATH"

    elif not os.path.exists(ffbin):
        pytest.skip("Missing Firefox build. Build it, or set GECKO_BINARY_PATH")

    check_call(
        [
            sys.executable,
            "-m",
            "grizzly",
            ffbin,
            "no-op",
            "--headless",
            "--smoke-test",
            "--display-launch-failures",
            "--limit",
            "10",
            "--relaunch",
            "5",
        ],
    )


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