komorebi/e2e/lib/tests/test_tiling.py

"""The tiling scenario against a platform that tiles correctly, and against ones that do not."""

from PIL import Image

from komorebi_vm.core.checks import Checks
from komorebi_vm.core.geometry import Rect
from komorebi_vm.core.platform import Client, OsWindow, Pointer
from komorebi_vm.core.scenarios import tiling
from komorebi_vm.core.scenarios.resize import outside_grab
from komorebi_vm.core.snapshot import Snapshot, parse


class FakePlatform:
    """A desktop in memory: `spawn` opens a window, and the frames are the ones it is given."""

    tolerance = 24

    def __init__(self, frames: list[Rect]) -> None:
        self.frames = frames
        self.opened: list[str] = []

    def wait_for(self, predicate, timeout: float = 10) -> bool:  # noqa: ANN001
        return predicate()

    def spawn(
        self,
        title: str,
        app_id: str | None = None,
        exe: str | None = None,
        rename_to: str | None = None,
        after: float = 0,
    ) -> Client:
        self.opened.append(title)
        return Client(title)

    def state(self) -> Snapshot:
        return parse(_state(self.opened))

    def os_window(self, client: Client) -> OsWindow | None:
        i = self.opened.index(client.title)
        return OsWindow(client.title, self.frames[i], focused=False)

    def hidden(self, client: Client) -> bool:
        return False

    def close(self, client: Client) -> None:
        pass

    def displays(self) -> list[Rect]:
        return [Rect(0, 0, 1920, 1080)]

    def add_display(self) -> None:
        pass

    def remove_display(self) -> None:
        pass

    def resizable_display(self) -> int | None:
        return None

    def resize_display(self, index: int, width: int, height: int) -> None:
        pass

    def komorebic_status(self, *args: str) -> tuple[int, str]:
        return 0, ""

    def has(self, feature: str) -> bool:
        return False

    @property
    def pointer(self) -> Pointer:
        raise AssertionError("no pointer")

    def work_area(self) -> Rect:
        return Rect(0, 0, 1920, 1080)

    def start_komorebi(self) -> None:
        pass

    def restart_with(self, config: dict) -> None:
        pass

    def crash(self) -> None:
        pass

    def komorebic(self, *args: str) -> str:
        return ""

    def exe_id(self, program: str) -> str:
        return program

    def temp_path(self, name: str) -> str:
        return f"/tmp/{name}"

    def minimize(self, client: Client) -> None:
        pass

    def rewrite_config(self, config: dict) -> None:
        pass

    def lock(self) -> None:
        pass

    def unlock(self) -> None:
        pass

    def put_file(self, path: str, data: bytes) -> None:  # noqa: ARG002
        pass

    def wallpaper(self, display: int = 0) -> str:  # noqa: ARG002
        return ""

    def set_wallpaper(self, path: str, display: int = 0) -> None:
        pass

    def resize_grab(self, frame: Rect, grab: str) -> tuple[int, int]:
        return outside_grab(frame, grab)

    def move_window(self, client: Client, x: int, y: int) -> None:
        pass

    def activate(self, client: Client) -> None:
        pass

    def fullscreen(self, client: Client, on: bool = True) -> None:
        pass

    def user_maximize(self, client: Client, on: bool = True) -> None:
        pass

    def screenshot(self, name: str) -> Image.Image:
        return Image.new("RGB", (4, 4))


def _state(titles: list[str]) -> dict:
    width = 1920 // len(titles)
    rects = [
        {"left": i * width, "top": 0, "right": width, "bottom": 1080} for i in range(len(titles))
    ]
    windows = [
        {"id": i + 1, "rect": rects[i], "details": {"title": t, "exe": "", "class": "", "path": ""}}
        for i, t in enumerate(titles)
    ]
    workspace = {
        "name": None,
        "containers": {
            "elements": [
                {"id": str(i), "locked": False, "windows": {"elements": [w], "focused": 0}}
                for i, w in enumerate(windows)
            ],
            "focused": 0,
        },
        "monocle_container": None,
        "maximized_window": None,
        "floating_windows": {"elements": [], "focused": 0},
        "layout": {"Default": "Columns"},
        "tile": True,
        "latest_layout": rects,
    }
    monitor = {
        "id": 1,
        "size": {"left": 0, "top": 0, "right": 1920, "bottom": 1080},
        "workspaces": {"elements": [workspace], "focused": 0},
    }
    return {"monitors": {"elements": [monitor], "focused": 0}}


def test_windows_that_are_tiled_pass() -> None:
    platform = FakePlatform(
        [Rect(0, 0, 640, 1080), Rect(640, 0, 640, 1080), Rect(1280, 0, 640, 1080)]
    )
    checks = Checks()
    tiling.run(platform, checks)
    assert checks.failed == 0
    assert checks.passed > 0


def test_windows_over_each_other_fail() -> None:
    platform = FakePlatform(
        [Rect(0, 0, 700, 1080), Rect(640, 0, 640, 1080), Rect(1280, 0, 640, 1080)]
    )
    checks = Checks()
    tiling.run(platform, checks)
    assert any("overlap" in f for f in checks.failures)


def test_a_window_off_the_screen_fails() -> None:
    platform = FakePlatform(
        [Rect(0, 0, 640, 1080), Rect(640, 0, 640, 1080), Rect(1500, 0, 640, 1080)]
    )
    checks = Checks()
    tiling.run(platform, checks)
    assert any("inside the work area" in f for f in checks.failures)