summaryrefslogtreecommitdiffstats
path: root/testing/web-platform/mozilla/tests/webdriver/classic/protocol/marionette_port.py
blob: 20b4e0332431cc33cd05a398d7b951c61dfd03e4 (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
42
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, create_custom_profile, geckodriver
):
    config = deepcopy(configuration)
    custom_profile = create_custom_profile()
    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")