summaryrefslogtreecommitdiffstats
path: root/testing/web-platform/mozilla/tests/webdriver/classic/new_session/binary.py
blob: 79d1f842ed31fdd591806c1ebf639493df555a5e (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
import os

from tests.support.asserts import assert_error, assert_success


def test_bad_binary(new_session, configuration):
    # skipif annotations are forbidden in wpt
    if os.path.exists("/bin/echo"):
        capabilities = configuration["capabilities"].copy()
        capabilities["moz:firefoxOptions"]["binary"] = "/bin/echo"

        response, _ = new_session({"capabilities": {"alwaysMatch": capabilities}})
        assert_error(response, "invalid argument")


def test_shell_script_binary(new_session, configuration):
    # skipif annotations are forbidden in wpt
    if os.path.exists("/bin/bash"):
        capabilities = configuration["capabilities"].copy()
        binary = configuration["browser"]["binary"]

        path = os.path.abspath("firefox.sh")
        assert not os.path.exists(path)
        try:
            script = f"""#!/bin/bash\n\n"{binary}" $@\n"""
            with open("firefox.sh", "w") as f:
                f.write(script)
            os.chmod(path, 0o744)
            capabilities["moz:firefoxOptions"]["binary"] = path
            response, _ = new_session({"capabilities": {"alwaysMatch": capabilities}})
            assert_success(response)
        finally:
            os.unlink(path)