feat: initialize resume agent with OfferPai sync

2026-08-05 20:20:18 +08:00
commit 61ec750031
197 changed files with 33291 additions and 0 deletions
@@ -0,0 +1,87 @@
+from __future__ import annotations
+
+import os
+from uuid import uuid4
+
+from fastapi.testclient import TestClient
+from sqlalchemy import create_engine, text
+
+from app.main import create_app
+from app.offerpai_auth import OfferPaiIdentity
+from app.postgres_database import PostgresDatabase
+from app.services import RuleBasedEntryExpander, RuleBasedExperienceExtractor, RuleBasedResumeRewriter
+from app.settings import Settings
+
+
+class StaticIdentityProvider:
+    def authenticate(self, _token: str) -> OfferPaiIdentity:
+        return OfferPaiIdentity(
+            user_id="2081575100391407617",
+            mobile_number="13421012384",
+            nick="用户2384",
+        )
+
+
+def test_create_app_uses_postgres_when_database_path_is_not_supplied(monkeypatch) -> None:
+    schema = f"test_runtime_{uuid4().hex}"
+    database_url = os.environ["RESUME_AGENT_TEST_DATABASE_URL"]
+    monkeypatch.setenv("RESUME_AGENT_DATABASE_SCHEMA", schema)
+    monkeypatch.setenv("RESUME_AGENT_DEFAULT_TIER", "free")
+    application = create_app(
+        extractor=RuleBasedExperienceExtractor(),
+        rewriter=RuleBasedResumeRewriter(),
+        expander=RuleBasedEntryExpander(),
+        settings=Settings(
+            llm_provider="rule",
+            database_url=database_url,
+            offerpai_auth_required=False,
+        ),
+    )
+    try:
+        assert isinstance(application.state.database, PostgresDatabase)
+        with TestClient(application) as client:
+            response = client.post("/ai-api/resume-agent/sessions", json={})
+            assert response.status_code == 201
+            session_id = response.json()["session_id"]
+            assert application.state.database.get_session(session_id) is not None
+    finally:
+        application.state.database.engine.dispose()
+        engine = create_engine(database_url)
+        try:
+            with engine.begin() as connection:
+                connection.execute(text(f'DROP SCHEMA IF EXISTS "{schema}" CASCADE'))
+        finally:
+            engine.dispose()
+
+
+def test_postgres_restores_session_by_external_user_id(monkeypatch) -> None:
+    schema = f"test_external_identity_{uuid4().hex}"
+    database_url = os.environ["RESUME_AGENT_TEST_DATABASE_URL"]
+    monkeypatch.setenv("RESUME_AGENT_DATABASE_SCHEMA", schema)
+    application = create_app(
+        extractor=RuleBasedExperienceExtractor(),
+        rewriter=RuleBasedResumeRewriter(),
+        expander=RuleBasedEntryExpander(),
+        settings=Settings(
+            llm_provider="rule",
+            database_url=database_url,
+            offerpai_auth_required=True,
+        ),
+        offerpai_identity_provider=StaticIdentityProvider(),
+    )
+    headers = {"Authorization": "Bearer header.payload.signature-value"}
+    try:
+        with TestClient(application) as client:
+            first = client.post("/ai-api/resume-agent/sessions", json={}, headers=headers)
+            second = client.post("/ai-api/resume-agent/sessions", json={}, headers=headers)
+            assert first.status_code == 201
+            assert second.status_code == 201
+            assert second.json()["session_id"] == first.json()["session_id"]
+    finally:
+        application.state.database.engine.dispose()
+        engine = create_engine(database_url)
+        try:
+            with engine.begin() as connection:
+                connection.execute(text(f'DROP SCHEMA IF EXISTS "{schema}" CASCADE'))
+        finally:
+            engine.dispose()