stagehand-python/tests/test_sea_binary.py at 0be87c729e8ffc6fea5e49068a6eeef29d0534e3 · browserbase/stagehand-python · GitHub

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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
from __future__ import annotations

import importlib.util
from pathlib import Path

from stagehand.lib import sea_binary
from stagehand._version import __version__


def _load_download_binary_module():
    script_path = Path(__file__).resolve().parents[1] / "scripts" / "download-binary.py"
    spec = importlib.util.spec_from_file_location("download_binary_script", script_path)
    assert spec is not None
    assert spec.loader is not None

    module = importlib.util.module_from_spec(spec)
    spec.loader.exec_module(module)
    return module


download_binary = _load_download_binary_module()


def test_resolve_binary_path_defaults_cache_version_to_package_version(
    monkeypatch,
    tmp_path: Path,
) -> None:
    resource_path = tmp_path / "stagehand-test"
    resource_path.write_bytes(b"binary")

    captured: dict[str, object] = {}

    monkeypatch.delenv("STAGEHAND_VERSION", raising=False)
    monkeypatch.setattr(sea_binary, "_resource_binary_path", lambda _filename: resource_path)

    def _fake_copy_to_cache(*, src: Path, filename: str, version: str) -> Path:
        captured["src"] = src
        captured["filename"] = filename
        captured["version"] = version
        return tmp_path / "cache" / filename

    monkeypatch.setattr(sea_binary, "_copy_to_cache", _fake_copy_to_cache)

    resolved = sea_binary.resolve_binary_path()

    assert resolved == tmp_path / "cache" / sea_binary.default_binary_filename()
    assert captured["src"] == resource_path
    assert captured["filename"] == sea_binary.default_binary_filename()
    assert captured["version"] == __version__


def test_parse_server_tag_rejects_prerelease_tags() -> None:
    assert download_binary._parse_server_tag("stagehand-server-v3/v3.20.0-dev") is None
    assert download_binary._parse_server_tag("stagehand-server-v3/v3.20.0+build.1") is None


def test_resolve_latest_server_tag_ignores_dev_releases(monkeypatch) -> None:
    releases = [
        {"tag_name": "stagehand-server-v3/v3.20.0-dev"},
        {"tag_name": "stagehand-server-v3/v3.19.1"},
        {"tag_name": "stagehand-server-v3/v3.19.0"},
    ]

    monkeypatch.setattr(download_binary, "_http_get_json", lambda _url: releases)

    assert download_binary.resolve_latest_server_tag() == "stagehand-server-v3/v3.19.1"