collab-runtime 0.2.9__py3-none-any.whl

tests/backend/unit/lock_client/test_lock_client_active.py

@@ -0,0 +1,104 @@
+"""Active-lock related tests for LockClient."""
+
+from __future__ import annotations
+
+from ._helpers import (
+    FakeClient,
+    FakeResponse,
+    load_lock_client_module,
+    make_create_client,
+)
+
+mod = load_lock_client_module()
+
+
+def test_active_locks(monkeypatch):
+    """Test retrieving all active locks."""
+    monkeypatch.setenv("SUPABASE_URL", "https://test.supabase.co")
+    monkeypatch.setenv("SUPABASE_ANON_KEY", "test_key")
+
+    locks_data = [
+        {"file_path": "src/app.py", "developer_id": "user1"},
+        {"file_path": "src/routes.py", "developer_id": "user2"},
+    ]
+    response = FakeResponse(status=200, data=locks_data)
+    monkeypatch.setattr(mod, "_get_create_client", lambda: make_create_client(response))
+
+    lc = mod.LockClient(developer_id="test_user")
+    locks = lc.active()
+    assert len(locks) == 2
+    assert locks[0]["file_path"] == "src/app.py"
+
+
+def test_active_locks_exception(monkeypatch):
+    """Test active() returns empty list on API exception."""
+    monkeypatch.setenv("SUPABASE_URL", "https://test.supabase.co")
+    monkeypatch.setenv("SUPABASE_ANON_KEY", "test_key")
+
+    class ExplodingClient(FakeClient):
+        def execute(self):
+            raise RuntimeError("Network error")
+
+    monkeypatch.setattr(
+        mod,
+        "_get_create_client",
+        lambda: (lambda url, key: ExplodingClient(FakeResponse())),
+    )
+
+    lc = mod.LockClient(developer_id="test_user")
+    locks = lc.active()
+    assert locks == []
+
+
+def test_active_locks_with_error(monkeypatch):
+    """Test active() returns empty list when response has error."""
+    monkeypatch.setenv("SUPABASE_URL", "https://test.supabase.co")
+    monkeypatch.setenv("SUPABASE_ANON_KEY", "test_key")
+
+    response = FakeResponse(status=500, data=None, error="Error")
+    monkeypatch.setattr(mod, "_get_create_client", lambda: make_create_client(response))
+
+    lc = mod.LockClient(developer_id="test_user")
+    locks = lc.active()
+    assert locks == []
+
+
+def test_get_lock_status_expired(monkeypatch):
+    """Test get_lock_status marks expired locks as unlocked (server-side expiry not
+    enforced)."""
+    monkeypatch.setenv("SUPABASE_URL", "https://test.supabase.co")
+    monkeypatch.setenv("SUPABASE_ANON_KEY", "test_key")
+
+    from datetime import datetime, timedelta, timezone
+
+    past = (datetime.now(timezone.utc) - timedelta(hours=1)).isoformat()
+    lock_data = {
+        "file_path": "src/app.py",
+        "developer_id": "other_user",
+        "acquired_at": "2025-01-01T10:00:00+00:00",
+        "expires_at": past,
+    }
+    response = FakeResponse(status=200, data=[lock_data])
+    monkeypatch.setattr(mod, "_get_create_client", lambda: make_create_client(response))
+
+    lc = mod.LockClient(developer_id="test_user")
+    status = lc.get_lock_status("src/app.py")
+    # With server-side expiry disabled, presence of a DB row implies an active
+    # lock until explicitly released. The client does not evaluate expires_at.
+    assert status["is_locked"] is True
+    assert status["locked_by"] == "other_user"
+    assert status["can_edit"] is False
+
+
+# RESTORED: test_get_lock_status_no_lock
+def test_get_lock_status_no_lock(monkeypatch):
+    monkeypatch.setenv("SUPABASE_URL", "https://example.invalid")
+    monkeypatch.setenv("SUPABASE_ANON_KEY", "anon:fake")
+    response = FakeResponse(status=200, data=[])
+    monkeypatch.setattr(mod, "_get_create_client", lambda: make_create_client(response))
+
+    lc = mod.LockClient(developer_id="tester")
+    status = lc.get_lock_status("some/file.py")
+    assert isinstance(status, dict)
+    assert status.get("is_locked") is False
+    assert status.get("can_edit") is True

tests/backend/unit/lock_client/test_lock_client_api.py

@@ -0,0 +1,63 @@
+from ._helpers import FakeClient, FakeResponse, load_lock_client_module
+
+# acquire tests moved to test_lock_client_acquire.py
+# force_release tests moved to test_lock_client_force_release.py
+# release tests moved to test_lock_client_release.py
+# This file now covers only active() and get_lock_status()
+
+
+def test_active_and_get_lock_status(monkeypatch):
+    mod = load_lock_client_module()
+    client = mod.LockClient(local_only=True, developer_id="frank")
+    client._client = FakeClient(
+        FakeResponse(
+            status=200,
+            data=[
+                {
+                    "file_path": "a",
+                    "developer_id": "frank",
+                    "acquired_at": "t",
+                    "reason": "r",
+                }
+            ],
+        )
+    )
+    act = client.active()
+    assert isinstance(act, list) and act
+    info = client.get_lock_status("a")
+    assert info.get("is_locked") is True
+    assert info.get("locked_by") == "frank"
+
+
+def test_force_release_all_chunking(monkeypatch):
+    mod = load_lock_client_module()
+    client = mod.LockClient(local_only=True, developer_id="root")
+    client._is_admin = True
+
+    # Simulate many active locks
+    monkeypatch.setattr(
+        client, "active", lambda: [{"file_path": f"f{i}"} for i in range(5)]
+    )
+
+    class RespNone:
+        status = 200
+        data = None
+        error = None
+
+    class CustomClient:
+        def table(self, *a, **k):
+            return self
+
+        def delete(self, *a, **k):
+            return self
+
+        def in_(self, *a, **k):
+            return self
+
+        def execute(self):
+            return RespNone()
+
+    client._client = CustomClient()
+    released = client.force_release_all()
+    assert isinstance(released, int)
+    assert released >= 0