upstream-aw-watcher-window/tests/test_main.py

import re
from types import SimpleNamespace

import pytest

import aw_watcher_window.main as main_module
from aw_watcher_window.macos_cli import build_swift_command


def test_research_mode_passes_map_to_macos_swift_strategy(monkeypatch):
    commands = []

    class FakeProcess:
        pid = 123

        def wait(self):
            return None

    class FakeClient:
        client_name = "aw-watcher-window"
        client_hostname = "host.localdomain"
        server_address = "http://localhost:5600"

        def __init__(self, *args, **kwargs):
            pass

        def create_bucket(self, *args, **kwargs):
            pass

        def wait_for_start(self):
            pass

        def __enter__(self):
            return self

        def __exit__(self, exc_type, exc, tb):
            return False

    monkeypatch.setattr(main_module.sys, "platform", "darwin")
    monkeypatch.setattr(main_module, "background_ensure_permissions", lambda: None)
    monkeypatch.setattr(main_module, "setup_logging", lambda **kwargs: None)
    monkeypatch.setattr(main_module, "ActivityWatchClient", FakeClient)
    monkeypatch.setattr(main_module.signal, "signal", lambda *args, **kwargs: None)
    monkeypatch.setattr(
        main_module.subprocess,
        "Popen",
        lambda command: commands.append(command) or FakeProcess(),
    )
    monkeypatch.setattr(
        main_module,
        "parse_args",
        lambda: SimpleNamespace(
            testing=True,
            verbose=False,
            host=None,
            port=None,
            strategy="swift",
            exclude_title=False,
            exclude_titles=[],
            research_enabled=True,
            research_category_map={"youtube": "Youtube"},
            research_app_category_map={},
        ),
    )

    main_module.main()

    assert commands[0][-4:] == ["--research", "--research-category", "youtube", "Youtube"]


def test_build_swift_command_omits_optional_filters():
    command = build_swift_command(
        "/tmp/aw-watcher-window-macos",
        "http://localhost:5600",
        "bucket",
        "host.localdomain",
        "aw-watcher-window",
    )

    assert command == [
        "/tmp/aw-watcher-window-macos",
        "http://localhost:5600",
        "bucket",
        "host.localdomain",
        "aw-watcher-window",
    ]


def test_build_swift_command_passes_title_filters():
    command = build_swift_command(
        "/tmp/aw-watcher-window-macos",
        "http://localhost:5600",
        "bucket",
        "host.localdomain",
        "aw-watcher-window",
        exclude_title=True,
        exclude_titles=["Zoom", "Slack.*huddle"],
    )

    assert command == [
        "/tmp/aw-watcher-window-macos",
        "http://localhost:5600",
        "bucket",
        "host.localdomain",
        "aw-watcher-window",
        "--exclude-title",
        "--exclude-titles",
        "Zoom",
        "--exclude-titles",
        "Slack.*huddle",
    ]


def test_build_swift_command_passes_empty_research_map():
    command = build_swift_command(
        "/tmp/aw-watcher-window-macos",
        "http://localhost:5600",
        "bucket",
        "host.localdomain",
        "aw-watcher-window",
        research_category_map={},
    )

    assert command == [
        "/tmp/aw-watcher-window-macos",
        "http://localhost:5600",
        "bucket",
        "host.localdomain",
        "aw-watcher-window",
        "--research",
    ]


def test_build_swift_command_passes_research_categories():
    command = build_swift_command(
        "/tmp/aw-watcher-window-macos",
        "http://localhost:5600",
        "bucket",
        "host.localdomain",
        "aw-watcher-window",
        research_category_map={"youtube": "Youtube", "gmail": "Email"},
    )

    assert command == [
        "/tmp/aw-watcher-window-macos",
        "http://localhost:5600",
        "bucket",
        "host.localdomain",
        "aw-watcher-window",
        "--research",
        "--research-category",
        "youtube",
        "Youtube",
        "--research-category",
        "gmail",
        "Email",
    ]


def test_research_transform_takes_precedence_over_exclude_titles():
    window = {
        "app": "Chrome",
        "title": "YouTube - Google Chrome",
        "url": "https://youtube.com/watch?v=abc",
    }

    transformed = main_module.transform_window(
        window,
        exclude_titles=[re.compile("youtube", re.IGNORECASE)],
        research_category_map={"youtube": "Youtube"},
    )

    assert transformed == {"app": "Chrome", "title": "Youtube"}


def test_research_transform_takes_precedence_over_exclude_title():
    window = {
        "app": "Chrome",
        "title": "YouTube - Google Chrome",
        "url": "https://youtube.com/watch?v=abc",
    }

    transformed = main_module.transform_window(
        window,
        exclude_title=True,
        research_category_map={"youtube": "Youtube"},
    )

    assert transformed == {"app": "Chrome", "title": "Youtube"}


def test_legacy_exclude_titles_still_apply_without_research_mode():
    window = {"app": "Chrome", "title": "YouTube - Google Chrome"}

    transformed = main_module.transform_window(
        window,
        exclude_titles=[re.compile("youtube", re.IGNORECASE)],
    )

    assert transformed == {"app": "Chrome", "title": "excluded"}


@pytest.mark.parametrize(
    "poll_time,expected_pulsetime",
    [
        (1.0, 2.0),
        (2.0, 3.0),
        (5.0, 7.5),
        (10.0, 15.0),
    ],
)
def test_pulsetime_scales_with_poll_time(
    poll_time: float, expected_pulsetime: float
):
    assert main_module.compute_pulsetime(poll_time) == expected_pulsetime