stop.py - mozsearch

Enable keyboard shortcuts

This WPT test may be referenced by the following Test IDs:
- /webdriver/bidi/moz/profiler/stop/stop.py - WPT Dashboard Interop Dashboard

import json

import os

import pytest

from .. import DEFAULT_START_OPTIONS

pytestmark = pytest.mark.asyncio

async def test_stop_basic(bidi_session, is_profiler_active):

    await bidi_session.moz.profiler.start(**DEFAULT_START_OPTIONS)

    assert await is_profiler_active() is True

    await bidi_session.moz.profiler.stop(discard=True)

    assert await is_profiler_active() is False

async def test_stop_when_not_running(bidi_session, is_profiler_active):

    assert await is_profiler_active() is False

    result = await bidi_session.moz.profiler.stop()

    assert result["path"] is None

    assert await is_profiler_active() is False

async def test_stop_twice(bidi_session, is_profiler_active):

    await bidi_session.moz.profiler.start(**DEFAULT_START_OPTIONS)

    assert await is_profiler_active() is True

    await bidi_session.moz.profiler.stop(discard=True)

    await bidi_session.moz.profiler.stop(discard=True)

    assert await is_profiler_active() is False

async def test_stop_discard(bidi_session, is_profiler_active):

    await bidi_session.moz.profiler.start(**DEFAULT_START_OPTIONS)

    assert await is_profiler_active() is True

    result = await bidi_session.moz.profiler.stop(discard=True)

    assert await is_profiler_active() is False

    assert result["path"] is None

async def test_stop_saves_profile(bidi_session, is_profiler_active):

    await bidi_session.moz.profiler.start(**DEFAULT_START_OPTIONS)

    assert await is_profiler_active() is True

    result = await bidi_session.moz.profiler.stop()

    assert await is_profiler_active() is False

    path = result["path"]

    assert path is not None

    assert os.path.exists(path)

    assert os.path.getsize(path) > 0

    try:

        with open(path) as f:

            profile = json.load(f)

        assert "meta" in profile

        assert "threads" in profile

        assert isinstance(profile["threads"], list)

    finally:

        os.remove(path)

async def test_stop_saved_profile_reflects_start_options(bidi_session):

    options = {

        "entries": 1000000,

        "interval": 2,

        "features": ["js", "stackwalk"],

        "threads": ["GeckoMain"],

    await bidi_session.moz.profiler.start(**options)

    result = await bidi_session.moz.profiler.stop()

    path = result["path"]

    assert path is not None

    try:

        with open(path) as f:

            profile = json.load(f)

        configuration = profile["meta"]["configuration"]

        assert configuration["interval"] == options["interval"]

        for thread in options["threads"]:

            assert thread in configuration["threads"]

        for feature in options["features"]:

            assert feature in configuration["features"]

    finally:

        os.remove(path)