claude-dev-env 8.37.3 → 8.38.0

package/scripts/test_account_broker_support.py

@@ -0,0 +1,107 @@
+from __future__ import annotations
+
+import json
+import subprocess
+from datetime import datetime, timezone
+from pathlib import Path
+from types import SimpleNamespace
+
+import pytest
+
+import account_broker_support as support
+from account_broker_support import Account, Meters, Product, ProductAdapter
+
+
+def test_should_load_claude_roster_from_both_lists(
+    monkeypatch: pytest.MonkeyPatch, tmp_path: Path
+) -> None:
+    monkeypatch.setattr(Path, "home", lambda: tmp_path)
+    monkeypatch.setattr(support, "default_profile_home", lambda name="extra": tmp_path / "profiles" / name)
+    main_home = tmp_path / ".claude"
+    main_home.mkdir()
+    (main_home / "claude-chain.json").write_text(
+        json.dumps({"chain": [{"command": "claude"}, {"command": "alternate", "credentials_path": str(tmp_path / "alternate" / ".credentials.json") }]}),
+        encoding="utf-8",
+    )
+    (main_home / "extra-profiles.json").write_text('["extra"]', encoding="utf-8")
+
+    accounts = support.load_claude_accounts()
+
+    assert [each_account.name for each_account in accounts] == ["main", "alternate", "extra"]
+    assert accounts[1].home == (tmp_path / "alternate").resolve()
+
+
+def test_should_load_codex_roster_without_meter_reads(
+    monkeypatch: pytest.MonkeyPatch, tmp_path: Path
+) -> None:
+    monkeypatch.setattr(support.codex_account_choice, "default_profiles_root", lambda: tmp_path)
+    monkeypatch.setattr(support.codex_account_choice, "codex_account_names", lambda _: ("one",))
+    monkeypatch.delenv("CODEX_ACCOUNT_PROFILES", raising=False)
+    (tmp_path / "account-launchers.json").write_text('["one"]', encoding="utf-8")
+
+    assert [each_account.name for each_account in support.load_codex_accounts()] == ["one"]
+
+
+def test_should_read_claude_meters_through_injected_probe(
+    monkeypatch: pytest.MonkeyPatch, tmp_path: Path
+) -> None:
+    usage = SimpleNamespace(
+        session_utilization=20.0,
+        session_resets_at=datetime(2026, 10, 3, tzinfo=timezone.utc),
+        weekly_utilization=30.0,
+        weekly_resets_at=None,
+    )
+    monkeypatch.setattr(support, "probe_account_meters", lambda _: usage)
+
+    meters = support.read_claude_meters(Account(Product.CLAUDE, "one", tmp_path))
+
+    assert meters.session_percent_left == 80.0
+    assert meters.weekly_percent_left == 70.0
+
+
+def test_should_read_codex_meters_through_injected_reader(
+    monkeypatch: pytest.MonkeyPatch, tmp_path: Path
+) -> None:
+    short = SimpleNamespace(duration_minutes=300, used_percent=20.0, resets_at=None)
+    weekly = SimpleNamespace(duration_minutes=10080, used_percent=30.0, resets_at=None)
+    usage = SimpleNamespace(all_windows=(short, weekly), short_window_percent_left=80.0)
+    monkeypatch.setattr(support.codex_account_meters, "resolve_codex_path", lambda _: tmp_path / "codex")
+    monkeypatch.setattr(support.codex_account_meters, "read_codex_meters", lambda *_: usage)
+
+    meters = support.read_codex_account_meters(Account(Product.CODEX, "one", tmp_path))
+
+    assert meters.session_percent_left == 80.0
+    assert meters.weekly_percent_left == 70.0
+
+
+def test_should_cache_account_reading_under_injected_state_path(
+    monkeypatch: pytest.MonkeyPatch, tmp_path: Path
+) -> None:
+    monkeypatch.setattr(support, "broker_state_path", lambda: tmp_path / "broker" / "state.json")
+    account = Account(Product.CODEX, "one", tmp_path / "one")
+    seen: list[str] = []
+
+    def reader(selected: Account) -> Meters:
+        seen.append(selected.name)
+        return Meters(80.0, None, 80.0, None)
+
+    adapter = ProductAdapter(lambda: (account,), reader, "CODEX_HOME", (), False)
+    state = support._load_state(support.broker_state_path())
+    now = datetime(2026, 10, 3, tzinfo=timezone.utc)
+
+    assert support.read_accounts(Product.CODEX, adapter, all_state=state, now=now)[0].meters is not None
+    assert support.read_accounts(Product.CODEX, adapter, all_state=state, now=now)[0].meters is not None
+    assert seen == ["one"]
+    assert support.broker_state_path().is_file()
+
+
+def test_should_extract_session_id_and_restore_runner() -> None:
+    original = support.subprocess_runner
+
+    def runner(argv: object, **options: object) -> subprocess.CompletedProcess[str]:
+        return subprocess.CompletedProcess(argv, 0, "", "")
+
+    with support.override_subprocess_runner(runner):
+        assert support.subprocess_runner is runner
+    assert support.subprocess_runner is original
+    assert support.extract_session_id_from_stdout('{"session_id":"one"}') == "one"