mirror of
https://github.com/mcloh/codex-chatgpt-mcp-proxy.git
synced 2026-11-05 21:34:11 +00:00
117 lines
4.8 KiB
Python
117 lines
4.8 KiB
Python
import concurrent.futures
|
|
import time
|
|
import pytest
|
|
from codex_chatgpt_mcp_proxy.store import Store, Conflict
|
|
from conftest import claim
|
|
|
|
|
|
def test_idempotency_and_payload_conflict(store):
|
|
first = store.submit("hello", "demo", "key")
|
|
assert store.submit("hello", "demo", "key")["id"] == first["id"]
|
|
with pytest.raises(Conflict):
|
|
store.submit("different", "demo", "key")
|
|
with pytest.raises(ValueError):
|
|
store.submit("hello", "/tmp", "another")
|
|
with pytest.raises(ValueError):
|
|
store.submit("hello", "demo", "another", 4000)
|
|
|
|
|
|
def test_concurrent_claim_is_atomic(store):
|
|
store.submit("hello", "demo", "key")
|
|
store.register("home", "1", "test", ["demo"])
|
|
with concurrent.futures.ThreadPoolExecutor(max_workers=8) as pool:
|
|
rows = list(pool.map(lambda _: store.claim("home"), range(8)))
|
|
assert sum(r is not None for r in rows) == 1
|
|
|
|
|
|
def test_renew_expiry_fencing_and_final_retry(store):
|
|
task = store.submit("hello", "demo", "key")
|
|
r = claim(store)
|
|
with pytest.raises(Conflict):
|
|
store.renew(task["id"], "home", "wrong")
|
|
store.progress(task["id"], "home", r["attempt"], "running", "thread", "turn")
|
|
assert store.renew(task["id"], "home", r["attempt"])["state"] == "running"
|
|
with store.tx() as c:
|
|
c.execute("UPDATE tasks SET lease_until=?", (time.time() - 1,))
|
|
assert store.claim("home") is None
|
|
assert store.get(task["id"])["state"] == "needs_reconciliation"
|
|
store.finish(task["id"], "home", r["attempt"], "succeeded", "done")
|
|
store.finish(task["id"], "home", r["attempt"], "succeeded", "done")
|
|
with pytest.raises(Conflict):
|
|
store.finish(task["id"], "home", r["attempt"], "failed", "different")
|
|
with pytest.raises(Conflict):
|
|
store.progress(task["id"], "home", r["attempt"], "running")
|
|
assert store.cancel(task["id"])["state"] == "succeeded"
|
|
|
|
|
|
def test_continuation_thread_serialization(store):
|
|
task = store.submit("hello", "demo", "key")
|
|
r = claim(store)
|
|
with pytest.raises(Conflict):
|
|
store.continue_task(task["id"], "continue", "key2")
|
|
store.progress(task["id"], "home", r["attempt"], "running", "thread", "turn")
|
|
store.finish(task["id"], "home", r["attempt"], "succeeded", "done")
|
|
second = store.continue_task(task["id"], "continue", "key2")
|
|
assert second["thread_id"] == "thread"
|
|
with pytest.raises(Conflict):
|
|
store.continue_task(task["id"], "again", "key3")
|
|
assert claim(store)["id"] == second["id"]
|
|
|
|
|
|
def test_events_persistence_and_pagination(store, config):
|
|
task = store.submit("hello", "demo", "key")
|
|
r = claim(store)
|
|
for i in range(4):
|
|
store.event(task["id"], "home", r["attempt"], str(i), {"i": i})
|
|
store.event(task["id"], "home", r["attempt"], "0", {"i": 0})
|
|
with pytest.raises(Conflict):
|
|
store.event(task["id"], "home", r["attempt"], "0", {"i": 99})
|
|
restarted = Store(config)
|
|
page = restarted.events(task["id"], limit=2)
|
|
assert len(page["items"]) == 2
|
|
page2 = restarted.events(task["id"], after=page["next_cursor"])
|
|
assert [r["payload"]["i"] for r in page2["items"]] == [2, 3]
|
|
assert restarted.get(task["id"])["attempt"] == r["attempt"]
|
|
|
|
|
|
def test_cancel_queued_running_and_terminal(store):
|
|
queued = store.submit("hello", "demo", "key")
|
|
assert store.cancel(queued["id"])["state"] == "cancelled"
|
|
task = store.submit("hello", "demo", "key2")
|
|
r = claim(store)
|
|
assert store.cancel(task["id"])["state"] == "cancel_requested"
|
|
store.progress(task["id"], "home", r["attempt"], "running")
|
|
assert store.get(task["id"])["state"] == "cancel_requested"
|
|
store.finish(task["id"], "home", r["attempt"], "cancelled")
|
|
assert store.cancel(task["id"])["state"] == "cancelled"
|
|
|
|
|
|
def test_bounds_and_retention(store):
|
|
task = store.submit("hello", "demo", "key")
|
|
r = claim(store)
|
|
with pytest.raises(ValueError):
|
|
store.event(task["id"], "home", r["attempt"], "large", {"v": "x" * 20000})
|
|
store.finish(task["id"], "home", r["attempt"], "succeeded", "done")
|
|
with store.tx() as c:
|
|
c.execute("UPDATE tasks SET updated=?", (time.time() - 86400 * 31,))
|
|
assert store.cleanup()["purged_payloads"] == 1
|
|
assert store.get(task["id"])["result"] is None
|
|
|
|
|
|
def test_manual_reconciliation_audit_and_release(store):
|
|
task = store.submit("hello", "demo", "k")
|
|
leased = claim(store)
|
|
store.finish(task["id"], "home", leased["attempt"], "needs_reconciliation", error="uncertain")
|
|
store.reconcile(task["id"], "home", leased["attempt"], "failed")
|
|
assert store.get(task["id"])["state"] == "failed"
|
|
assert [h["state"] for h in store.get(task["id"])["history"]] == [
|
|
"queued",
|
|
"leased",
|
|
"needs_reconciliation",
|
|
"failed",
|
|
]
|
|
with pytest.raises(Conflict):
|
|
store.reconcile(task["id"], "home", leased["attempt"], "failed")
|
|
store.submit("next", "demo", "k2")
|
|
assert claim(store) is not None
|