summaryrefslogtreecommitdiffstats
path: root/testing/web-platform/mozilla/tests/webdriver/classic/protocol/marionette_port.py
blob: 09951abc43a9cadc52265d6df3c6a981bfe38563 (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
import os
from copy import deepcopy

import pytest


@pytest.mark.parametrize("port", ["0", "2828"], ids=["system allocated", "fixed"])
def test_marionette_port(geckodriver, port):
    extra_args = ["--marionette-port", port]

    driver = geckodriver(extra_args=extra_args)
    driver.new_session()
    driver.delete_session()


def test_marionette_port_outdated_active_port_file(
    configuration, geckodriver, custom_profile
):
    config = deepcopy(configuration)
    extra_args = ["--marionette-port", "0"]

    # Prepare a Marionette active port file that contains a port which will
    # never be used when requesting a system allocated port.
    active_port_file = os.path.join(custom_profile.profile, "MarionetteActivePort")
    with open(active_port_file, "wb") as f:
        f.write(b"53")

    config["capabilities"]["moz:firefoxOptions"]["args"] = [
        "--profile",
        custom_profile.profile,
    ]

    driver = geckodriver(config=config, extra_args=extra_args)

    driver.new_session()
    with open(active_port_file, "rb") as f:
        assert f.readline() != b"53"

    driver.delete_session()
    with pytest.raises(FileNotFoundError):
        open(active_port_file, "rb")