shutdown.py - mozsearch

Enable keyboard shortcuts

This WPT test may be referenced by the following Test IDs:
- /webdriver/harness/shutdown.py - WPT Dashboard Interop Dashboard

import os

import signal

import subprocess

import time

import pytest

IS_WINDOWS = os.name == "nt"

if IS_WINDOWS:

    SIGNALS = [signal.CTRL_BREAK_EVENT]

else:

    SIGNALS = [signal.SIGINT, signal.SIGTERM]

def is_process_alive(pid):

    if IS_WINDOWS:

        try:

            output = subprocess.check_output(

                ["tasklist", "/FI", f"PID eq {pid}"], text=True

            return str(pid) in output

        except subprocess.CalledProcessError:

            return False

    try:

        os.kill(pid, 0)

        return True

    except OSError:

        return False

def wait_for_process_exit(pid, timeout=10):

    end_time = time.time() + timeout

    while time.time() < end_time:

        if not is_process_alive(pid):

            return True

        time.sleep(0.1)

    return False

@pytest.mark.parametrize("signal", SIGNALS)

def test_firefox_quits_on_signal(configuration, geckodriver, signal):

    popen_kwargs = {}

    if IS_WINDOWS:

        # Start geckodriver in its own process group so that

        # CTRL_BREAK_EVENT is only delivered to it and not to the

        # parent cmd.exe, which would otherwise prompt

        # "Terminate batch job (Y/N)?" and hang.

        popen_kwargs["creationflags"] = subprocess.CREATE_NEW_PROCESS_GROUP

    driver = geckodriver(config=configuration, popen_kwargs=popen_kwargs)

    driver.new_session()

    firefox_pid = driver.session.capabilities["moz:processID"]

    driver.proc.send_signal(signal)

    assert wait_for_process_exit(firefox_pid), (

        "Firefox process still running after geckodriver was terminated"