initial impl

tests/test_storage.py

@@ -0,0 +1,103 @@
+from pathlib import Path
+
+import pytest
+
+from agentci.adapters.storage import Storage
+from agentci.domain.models import (
+    Job,
+    JobKind,
+    JobStatus,
+    Workflow,
+    WorkflowKind,
+    WorkflowStatus,
+)
+
+
+@pytest.fixture
+async def storage(tmp_path: Path) -> Storage:
+    migrations = Path(__file__).parents[1] / "src" / "agentci" / "migrations"
+    value = Storage(tmp_path / "state.sqlite3", migrations)
+    await value.initialize()
+    return value
+
+
+def make_job(job_id: str = "job-1") -> Job:
+    return Job(
+        id=job_id,
+        kind=JobKind.PLAN,
+        target_key="alice/repo:issue:3",
+        repo_owner="alice",
+        repo_name="repo",
+        issue_number=3,
+        pr_number=None,
+        requester="alice",
+        message="",
+        comment_id=10,
+    )
+
+
+async def test_enqueue_is_idempotent_and_claims_fifo(storage: Storage) -> None:
+    assert await storage.enqueue("delivery-1", make_job())
+    assert not await storage.enqueue("delivery-1", make_job("job-2"))
+    claimed = await storage.claim_next()
+    assert claimed is not None
+    assert claimed.id == "job-1"
+    assert claimed.status is JobStatus.RUNNING
+    assert await storage.claim_next() is None
+
+
+async def test_recovers_running_job_as_failed(storage: Storage) -> None:
+    await storage.enqueue("delivery-1", make_job())
+    assert await storage.claim_next() is not None
+    recovered = await storage.recover_running()
+    assert [job.id for job in recovered] == ["job-1"]
+    assert await storage.claim_next() is None
+
+
+async def test_persists_and_finds_workflows(storage: Storage, tmp_path: Path) -> None:
+    workflow = Workflow(
+        id="workflow-1",
+        kind=WorkflowKind.PLAN,
+        repo_owner="alice",
+        repo_name="repo",
+        issue_number=3,
+        workspace_path=tmp_path / "repo",
+        base_sha="abc",
+        artifact="# Plan",
+        status=WorkflowStatus.COMPLETED,
+    )
+    await storage.create_workflow(workflow)
+    loaded = await storage.latest_workflow("alice", "repo", 3, WorkflowKind.PLAN)
+    assert loaded is not None
+    assert loaded.artifact == "# Plan"
+    assert loaded.workspace_path == tmp_path / "repo"
+
+
+async def test_tracks_operational_comments(storage: Storage) -> None:
+    await storage.enqueue("delivery-1", make_job())
+    await storage.set_job_comment("job-1", "accepted_comment_id", 21)
+    await storage.set_job_comment("job-1", "started_comment_id", 22)
+    assert await storage.operational_comment_ids("alice", "repo", 3) == {21, 22}
+
+
+async def test_failed_followup_does_not_invalidate_completed_workflow(
+    storage: Storage, tmp_path: Path
+) -> None:
+    workflow = Workflow(
+        id="workflow-1",
+        kind=WorkflowKind.PLAN,
+        repo_owner="alice",
+        repo_name="repo",
+        issue_number=3,
+        workspace_path=tmp_path / "repo",
+        base_sha="abc",
+        status=WorkflowStatus.COMPLETED,
+    )
+    await storage.create_workflow(workflow)
+    job = make_job()
+    job.workflow_id = workflow.id
+    await storage.enqueue("delivery-1", job)
+    await storage.fail_job_workflow(job.id)
+    loaded = await storage.latest_workflow("alice", "repo", 3, WorkflowKind.PLAN)
+    assert loaded is not None
+    assert loaded.status is WorkflowStatus.COMPLETED