From bae9fb1b7b176acd955583f774a8a4df4e15b8a3 Mon Sep 17 00:00:00 2001 From: sis0k0 Date: Fri, 22 May 2026 15:16:34 +0300 Subject: [PATCH 1/7] chore: migrate from Motor to PyMongo async --- app.py | 4 ++-- mongodb/checkpointer.py | 6 +++--- requirements.txt | 3 +-- 3 files changed, 6 insertions(+), 7 deletions(-) diff --git a/app.py b/app.py index 3730656..d661e74 100644 --- a/app.py +++ b/app.py @@ -10,7 +10,7 @@ from graph import create_workflow, AgentState from mongodb.connect import get_mongo_client from mongodb import checkpointer -from motor.motor_asyncio import AsyncIOMotorClient +from pymongo import AsyncMongoClient from langchain_core.messages import HumanMessage, AIMessage from utilities import sanitize_name from langchain.schema.runnable import Runnable @@ -41,7 +41,7 @@ async def on_chat_start(): workflow = create_workflow(chatbot_agent, tools) - mongo_client = AsyncIOMotorClient(MONGO_URI) + mongo_client = AsyncMongoClient(MONGO_URI) mongodb_checkpointer = checkpointer.MongoDBSaver(mongo_client, DATABASE_NAME, "checkpoints_collection") graph = workflow.compile(checkpointer=mongodb_checkpointer) diff --git a/mongodb/checkpointer.py b/mongodb/checkpointer.py index 66f424d..a0d9980 100644 --- a/mongodb/checkpointer.py +++ b/mongodb/checkpointer.py @@ -14,7 +14,7 @@ SerializerProtocol, ) from langgraph.serde.jsonplus import JsonPlusSerializer -from motor.motor_asyncio import AsyncIOMotorClient +from pymongo import AsyncMongoClient class JsonPlusSerializerCompat(JsonPlusSerializer): def loads(self, data: bytes) -> Any: @@ -25,13 +25,13 @@ def loads(self, data: bytes) -> Any: class MongoDBSaver(AbstractContextManager, BaseCheckpointSaver): serde = JsonPlusSerializerCompat() - client: AsyncIOMotorClient + client: AsyncMongoClient db_name: str collection_name: str def __init__( self, - client: AsyncIOMotorClient, + client: AsyncMongoClient, db_name: str, collection_name: str, *, diff --git a/requirements.txt b/requirements.txt index 8856143..fd83e15 100644 --- a/requirements.txt +++ b/requirements.txt @@ -53,7 +53,6 @@ langsmith==0.1.84 Lazify==0.4.0 literalai==0.0.607 marshmallow==3.21.3 -motor==3.5.1 multidict==6.0.5 mypy-extensions==1.0.0 nest-asyncio==1.6.0 @@ -77,7 +76,7 @@ pyasn1_modules==0.4.0 pydantic==2.8.2 pydantic_core==2.20.1 PyJWT==2.8.0 -pymongo==4.8.0 +pymongo==4.13.1 pyparsing==3.1.2 python-dateutil==2.9.0.post0 python-dotenv==1.0.1 From e6638707a1290ec000ecaae16a32d01e55cfe973 Mon Sep 17 00:00:00 2001 From: sis0k0 Date: Fri, 22 May 2026 15:24:18 +0300 Subject: [PATCH 2/7] ci: add standard GitHub Actions smoke workflow --- .github/workflows/ci.yml | 89 ++++++++++++++++++++++++++++++++++++++++ 1 file changed, 89 insertions(+) create mode 100644 .github/workflows/ci.yml diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml new file mode 100644 index 0000000..a3cc6bd --- /dev/null +++ b/.github/workflows/ci.yml @@ -0,0 +1,89 @@ +name: CI + +on: + push: + pull_request: + workflow_dispatch: + +jobs: + smoke: + runs-on: ubuntu-latest + + steps: + - name: Checkout + uses: actions/checkout@v4 + + - name: Set up Python + uses: actions/setup-python@v5 + with: + python-version: "3.12" + + - name: Set up Node + uses: actions/setup-node@v4 + with: + node-version: "20" + + - name: Repo smoke checks + shell: bash + run: | + set -euo pipefail + + echo "Validating repository structure and basic runnable signals" + + has_signal=0 + + if find . -maxdepth 4 -type f -name "package.json" | grep -q .; then + has_signal=1 + while IFS= read -r pkg; do + [ -z "$pkg" ] && continue + node -e "const fs=require('fs'); JSON.parse(fs.readFileSync(process.argv[1],'utf8'));" "$pkg" + done < <(find . -maxdepth 4 -type f -name "package.json") + fi + + if find . -maxdepth 4 -type f \( -name "pyproject.toml" -o -name "requirements.txt" -o -name "setup.py" -o -name "manage.py" \) | grep -q .; then + has_signal=1 + fi + + if find . -maxdepth 4 -type f \( -name "app.py" -o -name "main.py" -o -name "wsgi.py" -o -name "asgi.py" \) | grep -q .; then + has_signal=1 + fi + + if find . -maxdepth 4 -type f \( -name "pom.xml" -o -name "build.gradle" -o -name "build.gradle.kts" -o -name "gradlew" \) | grep -q .; then + has_signal=1 + fi + + if find . -maxdepth 4 -type f -name "go.mod" | grep -q .; then + has_signal=1 + fi + + if find . -maxdepth 4 -type f -name "Cargo.toml" | grep -q .; then + has_signal=1 + fi + + if find . -maxdepth 4 -type f \( -name "*.csproj" -o -name "*.sln" \) | grep -q .; then + has_signal=1 + fi + + if find . -maxdepth 4 -type f \( -name "Dockerfile" -o -name "docker-compose.yml" -o -name "docker-compose.yaml" \) | grep -q .; then + has_signal=1 + fi + + if find . -maxdepth 4 -type f -name "Makefile" | grep -q .; then + has_signal=1 + fi + + if [ "$has_signal" -ne 1 ]; then + echo "No runnable/build signals found in repository" + exit 1 + fi + + echo "Running Python syntax smoke check" + python_files="$(find . -type f -name '*.py' -not -path './.git/*' 2>/dev/null || true)" + if [ -n "$python_files" ]; then + while IFS= read -r f; do + [ -z "$f" ] && continue + python -m py_compile "$f" + done <<< "$python_files" + fi + + echo "Smoke checks passed" From a8154c85e0234ee50efe21674906203c09783037 Mon Sep 17 00:00:00 2001 From: sis0k0 Date: Fri, 22 May 2026 15:26:49 +0300 Subject: [PATCH 3/7] test: add runtime smoke test and run it in CI --- .github/workflows/ci.yml | 11 +++++ tests/run_runtime_smoke.py | 92 ++++++++++++++++++++++++++++++++++++++ 2 files changed, 103 insertions(+) create mode 100755 tests/run_runtime_smoke.py diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index a3cc6bd..986c994 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -87,3 +87,14 @@ jobs: fi echo "Smoke checks passed" + + - name: Run repository runtime smoke test + shell: bash + run: | + set -euo pipefail + if [ -f tests/run_runtime_smoke.py ]; then + python tests/run_runtime_smoke.py + else + echo "No runtime smoke test file found" + exit 1 + fi diff --git a/tests/run_runtime_smoke.py b/tests/run_runtime_smoke.py new file mode 100755 index 0000000..12e3437 --- /dev/null +++ b/tests/run_runtime_smoke.py @@ -0,0 +1,92 @@ +#!/usr/bin/env python3 +"""Repository runtime smoke test. + +This test intentionally avoids external services and validates: +1) repo has at least one runnable/build marker +2) Python source files in repo compile successfully +""" + +from __future__ import annotations + +from pathlib import Path +import py_compile +import sys + +REPO_ROOT = Path(__file__).resolve().parents[1] + +SKIP_DIRS = { + ".git", + ".venv", + "venv", + "node_modules", + "dist", + "build", + "__pycache__", +} + + +def has_runnable_marker(root: Path) -> bool: + markers = [ + "package.json", + "pyproject.toml", + "requirements.txt", + "setup.py", + "manage.py", + "pom.xml", + "build.gradle", + "build.gradle.kts", + "go.mod", + "Cargo.toml", + "Dockerfile", + "docker-compose.yml", + "docker-compose.yaml", + "Makefile", + ] + for marker in markers: + if list(root.rglob(marker)): + return True + + if list(root.rglob("*.csproj")) or list(root.rglob("*.sln")): + return True + + return False + + +def iter_python_files(root: Path): + for py_file in root.rglob("*.py"): + rel = py_file.relative_to(root) + if any(part in SKIP_DIRS for part in rel.parts): + continue + yield py_file + + +def main() -> int: + if not has_runnable_marker(REPO_ROOT): + print("FAIL: no runnable/build marker found") + return 1 + + py_files = list(iter_python_files(REPO_ROOT)) + if not py_files: + print("PASS: no python files to compile; marker checks passed") + return 0 + + failures = [] + for py_file in py_files: + try: + py_compile.compile(str(py_file), doraise=True) + except py_compile.PyCompileError as exc: + failures.append((py_file, exc.msg)) + + if failures: + print(f"FAIL: {len(failures)} python files failed to compile") + for py_file, msg in failures[:20]: + rel = py_file.relative_to(REPO_ROOT) + print(f" - {rel}: {msg}") + return 1 + + print(f"PASS: compiled {len(py_files)} python files") + return 0 + + +if __name__ == "__main__": + sys.exit(main()) From bf77464759f3a09e62178bc6d693c502aafacac2 Mon Sep 17 00:00:00 2001 From: sis0k0 Date: Fri, 22 May 2026 15:30:30 +0300 Subject: [PATCH 4/7] test: replace generic smoke check with repo-specific runtime contract test --- tests/run_runtime_smoke.py | 203 ++++++++++++++++++++++++------------- 1 file changed, 130 insertions(+), 73 deletions(-) diff --git a/tests/run_runtime_smoke.py b/tests/run_runtime_smoke.py index 12e3437..596d0af 100755 --- a/tests/run_runtime_smoke.py +++ b/tests/run_runtime_smoke.py @@ -1,9 +1,9 @@ #!/usr/bin/env python3 -"""Repository runtime smoke test. +"""Repo-specific runtime contract test. -This test intentionally avoids external services and validates: -1) repo has at least one runnable/build marker -2) Python source files in repo compile successfully +This validates that the repository's primary runtime module(s): +- compile successfully +- keep expected runtime entrypoints/routes/contracts """ from __future__ import annotations @@ -12,80 +12,137 @@ import py_compile import sys -REPO_ROOT = Path(__file__).resolve().parents[1] - -SKIP_DIRS = { - ".git", - ".venv", - "venv", - "node_modules", - "dist", - "build", - "__pycache__", -} - - -def has_runnable_marker(root: Path) -> bool: - markers = [ - "package.json", - "pyproject.toml", - "requirements.txt", - "setup.py", - "manage.py", - "pom.xml", - "build.gradle", - "build.gradle.kts", - "go.mod", - "Cargo.toml", - "Dockerfile", - "docker-compose.yml", - "docker-compose.yaml", - "Makefile", - ] - for marker in markers: - if list(root.rglob(marker)): - return True - - if list(root.rglob("*.csproj")) or list(root.rglob("*.sln")): - return True - - return False - - -def iter_python_files(root: Path): - for py_file in root.rglob("*.py"): - rel = py_file.relative_to(root) - if any(part in SKIP_DIRS for part in rel.parts): - continue - yield py_file +ROOT = Path(__file__).resolve().parents[1] + + +def require_file(path: str) -> Path: + p = ROOT / path + if not p.exists(): + raise AssertionError(f"Missing expected file: {path}") + return p + + +def compile_file(path: str) -> None: + p = require_file(path) + py_compile.compile(str(p), doraise=True) + + +def read_text(path: str) -> str: + return require_file(path).read_text() + + +def require_contains(path: str, needle: str) -> None: + text = read_text(path) + if needle not in text: + raise AssertionError(f"Expected marker not found in {path}: {needle}") + + +def ok(msg: str) -> None: + print(f"PASS: {msg}") + + +def fail(msg: str) -> int: + print(f"FAIL: {msg}") + return 1 def main() -> int: - if not has_runnable_marker(REPO_ROOT): - print("FAIL: no runnable/build marker found") - return 1 + try: + repo = ROOT.name + + if repo == "FARM-Auth": + compile_file("backend/main.py") + require_contains("backend/main.py", "app = FastAPI()") + require_contains("backend/main.py", "@app.on_event(\"startup\")") + require_contains("backend/main.py", "get_users_router") + require_contains("backend/main.py", "get_todo_router") + ok("FARM-Auth runtime contracts") + + elif repo == "FARM-Intro": + compile_file("backend/main.py") + require_contains("backend/main.py", "app = FastAPI()") + require_contains("backend/main.py", "app.include_router(todo_router") + require_contains("backend/main.py", "prefix=\"/task\"") + ok("FARM-Intro runtime contracts") + + elif repo == "a2a-mcp-mongodb-multiagents": + compile_file("mcp/main.py") + require_contains("mcp/main.py", "mcp = FastMCP(") + require_contains("mcp/main.py", "@mcp.tool") + require_contains("mcp/main.py", "async def connect_to_mongo") + ok("a2a MCP runtime contracts") + + elif repo == "beanie-example": + compile_file("src/beaniecocktails/__init__.py") + compile_file("src/beaniecocktails/scripts/init_db.py") + require_contains("src/beaniecocktails/__init__.py", "app = FastAPI(lifespan=app_lifespan)") + require_contains("src/beaniecocktails/__init__.py", "init_beanie(") + require_contains("src/beaniecocktails/__init__.py", "app.include_router(cocktail_router") + ok("beanie-example runtime contracts") + + elif repo == "docbridge": + compile_file("examples/why/why/__init__.py") + require_contains("examples/why/why/__init__.py", "app = FastAPI(lifespan=db_lifespan)") + require_contains("examples/why/why/__init__.py", "@app.get(\"/profiles/{user_id}\")") + require_contains("examples/why/why/__init__.py", "class Profile(Document)") + ok("docbridge runtime contracts") + + elif repo == "farm-stack-to-do-app": + compile_file("backend/src/todo/server.py") + require_contains("backend/src/todo/server.py", "app = FastAPI(lifespan=lifespan") + require_contains("backend/src/todo/server.py", "@app.get(\"/api/lists\")") + require_contains("backend/src/todo/server.py", "@app.post(\"/api/lists\"") + require_contains("backend/src/todo/server.py", "@app.patch(\"/api/lists/{list_id}/checked_state\")") + ok("farm-stack-to-do-app runtime contracts") + + elif repo == "hr_agentic_chatbot": + compile_file("app.py") + require_contains("app.py", "@cl.on_chat_start") + require_contains("app.py", "@cl.on_message") + require_contains("app.py", "create_workflow(") + ok("hr_agentic_chatbot runtime contracts") + + elif repo == "mongodb-atlas-fastapi": + compile_file("app/main.py") + require_contains("app/main.py", "app = FastAPI()") + require_contains("app/main.py", "@app.get(\"/\", response_description=\"Student API HealthCheck\")") + require_contains("app/main.py", "@app.get(\"/students\"") + ok("mongodb-atlas-fastapi runtime contracts") + + elif repo == "mongodb-with-starlette": + compile_file("app.py") + require_contains("app.py", "app = Starlette(") + require_contains("app.py", "Route(\"/\", create_student, methods=[\"POST\"])") + require_contains("app.py", "Route(\"/{id}\", delete_student, methods=[\"DELETE\"])") + ok("mongodb-with-starlette runtime contracts") + + elif repo == "mongodb-with-tornado": + compile_file("app.py") + require_contains("app.py", "class MainHandler(tornado.web.RequestHandler)") + require_contains("app.py", "app = tornado.web.Application(") + require_contains("app.py", "(r\"/(?P\\w+)\", MainHandler)") + ok("mongodb-with-tornado runtime contracts") + + elif repo == "mongodb-with-sanic": + compile_file("app.py") + require_contains("app.py", "app = Sanic(__name__)") + require_contains("app.py", "@app.route(\"/\", methods=[\"POST\"])") + require_contains("app.py", "@app.route(\"/\", methods=[\"DELETE\"])") + ok("mongodb-with-sanic runtime contracts") + + elif repo == "celeb-matcher-farm": + compile_file("backend/src/server.py") + require_contains("backend/src/server.py", "app = FastAPI(lifespan=lifespan") + require_contains("backend/src/server.py", "@app.post(\"/api/search\")") + require_contains("backend/src/server.py", "class SearchPayload(BaseModel)") + ok("celeb-matcher-farm runtime contracts") + + else: + raise AssertionError(f"No repository-specific contract defined for: {repo}") - py_files = list(iter_python_files(REPO_ROOT)) - if not py_files: - print("PASS: no python files to compile; marker checks passed") return 0 - - failures = [] - for py_file in py_files: - try: - py_compile.compile(str(py_file), doraise=True) - except py_compile.PyCompileError as exc: - failures.append((py_file, exc.msg)) - - if failures: - print(f"FAIL: {len(failures)} python files failed to compile") - for py_file, msg in failures[:20]: - rel = py_file.relative_to(REPO_ROOT) - print(f" - {rel}: {msg}") - return 1 - - print(f"PASS: compiled {len(py_files)} python files") - return 0 + except Exception as exc: + return fail(str(exc)) if __name__ == "__main__": From d3fddde62c8bf1e4bc878a223aac909d7a62d3e6 Mon Sep 17 00:00:00 2001 From: sis0k0 Date: Fri, 22 May 2026 15:41:56 +0300 Subject: [PATCH 5/7] test: replace smoke checks with repo-local runtime tests --- .github/workflows/ci.yml | 4 +- tests/run_runtime_smoke.py | 149 ------------------------------------- tests/test_runtime.py | 60 +++++++++++++++ 3 files changed, 62 insertions(+), 151 deletions(-) delete mode 100755 tests/run_runtime_smoke.py create mode 100644 tests/test_runtime.py diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index 986c994..b8014ec 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -92,8 +92,8 @@ jobs: shell: bash run: | set -euo pipefail - if [ -f tests/run_runtime_smoke.py ]; then - python tests/run_runtime_smoke.py + if [ -f tests/test_runtime.py ]; then + python tests/test_runtime.py else echo "No runtime smoke test file found" exit 1 diff --git a/tests/run_runtime_smoke.py b/tests/run_runtime_smoke.py deleted file mode 100755 index 596d0af..0000000 --- a/tests/run_runtime_smoke.py +++ /dev/null @@ -1,149 +0,0 @@ -#!/usr/bin/env python3 -"""Repo-specific runtime contract test. - -This validates that the repository's primary runtime module(s): -- compile successfully -- keep expected runtime entrypoints/routes/contracts -""" - -from __future__ import annotations - -from pathlib import Path -import py_compile -import sys - -ROOT = Path(__file__).resolve().parents[1] - - -def require_file(path: str) -> Path: - p = ROOT / path - if not p.exists(): - raise AssertionError(f"Missing expected file: {path}") - return p - - -def compile_file(path: str) -> None: - p = require_file(path) - py_compile.compile(str(p), doraise=True) - - -def read_text(path: str) -> str: - return require_file(path).read_text() - - -def require_contains(path: str, needle: str) -> None: - text = read_text(path) - if needle not in text: - raise AssertionError(f"Expected marker not found in {path}: {needle}") - - -def ok(msg: str) -> None: - print(f"PASS: {msg}") - - -def fail(msg: str) -> int: - print(f"FAIL: {msg}") - return 1 - - -def main() -> int: - try: - repo = ROOT.name - - if repo == "FARM-Auth": - compile_file("backend/main.py") - require_contains("backend/main.py", "app = FastAPI()") - require_contains("backend/main.py", "@app.on_event(\"startup\")") - require_contains("backend/main.py", "get_users_router") - require_contains("backend/main.py", "get_todo_router") - ok("FARM-Auth runtime contracts") - - elif repo == "FARM-Intro": - compile_file("backend/main.py") - require_contains("backend/main.py", "app = FastAPI()") - require_contains("backend/main.py", "app.include_router(todo_router") - require_contains("backend/main.py", "prefix=\"/task\"") - ok("FARM-Intro runtime contracts") - - elif repo == "a2a-mcp-mongodb-multiagents": - compile_file("mcp/main.py") - require_contains("mcp/main.py", "mcp = FastMCP(") - require_contains("mcp/main.py", "@mcp.tool") - require_contains("mcp/main.py", "async def connect_to_mongo") - ok("a2a MCP runtime contracts") - - elif repo == "beanie-example": - compile_file("src/beaniecocktails/__init__.py") - compile_file("src/beaniecocktails/scripts/init_db.py") - require_contains("src/beaniecocktails/__init__.py", "app = FastAPI(lifespan=app_lifespan)") - require_contains("src/beaniecocktails/__init__.py", "init_beanie(") - require_contains("src/beaniecocktails/__init__.py", "app.include_router(cocktail_router") - ok("beanie-example runtime contracts") - - elif repo == "docbridge": - compile_file("examples/why/why/__init__.py") - require_contains("examples/why/why/__init__.py", "app = FastAPI(lifespan=db_lifespan)") - require_contains("examples/why/why/__init__.py", "@app.get(\"/profiles/{user_id}\")") - require_contains("examples/why/why/__init__.py", "class Profile(Document)") - ok("docbridge runtime contracts") - - elif repo == "farm-stack-to-do-app": - compile_file("backend/src/todo/server.py") - require_contains("backend/src/todo/server.py", "app = FastAPI(lifespan=lifespan") - require_contains("backend/src/todo/server.py", "@app.get(\"/api/lists\")") - require_contains("backend/src/todo/server.py", "@app.post(\"/api/lists\"") - require_contains("backend/src/todo/server.py", "@app.patch(\"/api/lists/{list_id}/checked_state\")") - ok("farm-stack-to-do-app runtime contracts") - - elif repo == "hr_agentic_chatbot": - compile_file("app.py") - require_contains("app.py", "@cl.on_chat_start") - require_contains("app.py", "@cl.on_message") - require_contains("app.py", "create_workflow(") - ok("hr_agentic_chatbot runtime contracts") - - elif repo == "mongodb-atlas-fastapi": - compile_file("app/main.py") - require_contains("app/main.py", "app = FastAPI()") - require_contains("app/main.py", "@app.get(\"/\", response_description=\"Student API HealthCheck\")") - require_contains("app/main.py", "@app.get(\"/students\"") - ok("mongodb-atlas-fastapi runtime contracts") - - elif repo == "mongodb-with-starlette": - compile_file("app.py") - require_contains("app.py", "app = Starlette(") - require_contains("app.py", "Route(\"/\", create_student, methods=[\"POST\"])") - require_contains("app.py", "Route(\"/{id}\", delete_student, methods=[\"DELETE\"])") - ok("mongodb-with-starlette runtime contracts") - - elif repo == "mongodb-with-tornado": - compile_file("app.py") - require_contains("app.py", "class MainHandler(tornado.web.RequestHandler)") - require_contains("app.py", "app = tornado.web.Application(") - require_contains("app.py", "(r\"/(?P\\w+)\", MainHandler)") - ok("mongodb-with-tornado runtime contracts") - - elif repo == "mongodb-with-sanic": - compile_file("app.py") - require_contains("app.py", "app = Sanic(__name__)") - require_contains("app.py", "@app.route(\"/\", methods=[\"POST\"])") - require_contains("app.py", "@app.route(\"/\", methods=[\"DELETE\"])") - ok("mongodb-with-sanic runtime contracts") - - elif repo == "celeb-matcher-farm": - compile_file("backend/src/server.py") - require_contains("backend/src/server.py", "app = FastAPI(lifespan=lifespan") - require_contains("backend/src/server.py", "@app.post(\"/api/search\")") - require_contains("backend/src/server.py", "class SearchPayload(BaseModel)") - ok("celeb-matcher-farm runtime contracts") - - else: - raise AssertionError(f"No repository-specific contract defined for: {repo}") - - return 0 - except Exception as exc: - return fail(str(exc)) - - -if __name__ == "__main__": - sys.exit(main()) diff --git a/tests/test_runtime.py b/tests/test_runtime.py new file mode 100644 index 0000000..0ab1d0c --- /dev/null +++ b/tests/test_runtime.py @@ -0,0 +1,60 @@ +import importlib.util +import sys +import types +import unittest +from pathlib import Path + + +class RuntimeTest(unittest.TestCase): + @classmethod + def setUpClass(cls): + histories = types.ModuleType("langchain_mongodb.chat_message_histories") + + class MongoDBChatMessageHistory: + def __init__(self, uri, session_id, database_name=None, collection_name=None): + self.uri = uri + self.session_id = session_id + self.database_name = database_name + self.collection_name = collection_name + + histories.MongoDBChatMessageHistory = MongoDBChatMessageHistory + sys.modules["langchain_mongodb.chat_message_histories"] = histories + + target = Path(__file__).resolve().parents[1] / "mongodb" / "connect.py" + spec = importlib.util.spec_from_file_location("hr_connect", target) + cls.mod = importlib.util.module_from_spec(spec) + spec.loader.exec_module(cls.mod) + + def test_get_mongo_client_success_and_failure(self): + class GoodAdmin: + @staticmethod + def command(_): + return {"ok": 1} + + class GoodClient: + def __init__(self, *args, **kwargs): + self.admin = GoodAdmin() + + class BadAdmin: + @staticmethod + def command(_): + raise RuntimeError("nope") + + class BadClient: + def __init__(self, *args, **kwargs): + self.admin = BadAdmin() + + self.mod.MongoClient = GoodClient + self.assertIsNotNone(self.mod.get_mongo_client("mongodb://ok")) + + self.mod.MongoClient = BadClient + self.assertIsNone(self.mod.get_mongo_client("mongodb://bad")) + + def test_get_session_history_contract(self): + history = self.mod.get_session_history("s1") + self.assertEqual(history.session_id, "s1") + self.assertEqual(history.collection_name, "history") + + +if __name__ == "__main__": + unittest.main() From 0c70072bab3ead85a15301458f96f763bd444e59 Mon Sep 17 00:00:00 2001 From: sis0k0 Date: Fri, 22 May 2026 16:02:29 +0300 Subject: [PATCH 6/7] test: add integration tests against real MongoDB - tests/test_integration.py: per-repo pytest tests that connect to a real MongoDB instance and exercise the actual collection schema with CRUD operations - tests/__init__.py: makes tests a package so pytest resolves modules correctly - .github/workflows/ci.yml: adds mongo:latest service container (health-checked, port 27017) and a step that runs pytest tests/test_integration.py with MONGODB_URI pointing at the service container credentials Pattern follows mongodb-developer/mern-stack-example PR #51. --- .github/workflows/ci.yml | 22 ++++++ tests/__init__.py | 0 tests/test_integration.py | 141 ++++++++++++++++++++++++++++++++++++++ 3 files changed, 163 insertions(+) create mode 100644 tests/__init__.py create mode 100644 tests/test_integration.py diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index b8014ec..fb942d2 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -9,6 +9,20 @@ jobs: smoke: runs-on: ubuntu-latest + services: + mongodb: + image: mongo:latest + options: >- + --health-cmd mongosh + --health-interval 10s + --health-timeout 5s + --health-retries 5 + ports: + - 27017:27017 + env: + MONGO_INITDB_ROOT_USERNAME: admin + MONGO_INITDB_ROOT_PASSWORD: mongodb + steps: - name: Checkout uses: actions/checkout@v4 @@ -98,3 +112,11 @@ jobs: echo "No runtime smoke test file found" exit 1 fi + + - name: Install integration test dependencies + run: pip install pytest pymongo + + - name: Run integration tests + env: + MONGODB_URI: mongodb://admin:mongodb@localhost:27017/ + run: pytest tests/test_integration.py -v diff --git a/tests/__init__.py b/tests/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/tests/test_integration.py b/tests/test_integration.py new file mode 100644 index 0000000..04bc355 --- /dev/null +++ b/tests/test_integration.py @@ -0,0 +1,141 @@ +"""Integration tests for hr_agentic_chatbot. + +Tests real MongoDB connectivity and session history storage +used by the HR agentic chatbot. + +Requires a running MongoDB instance. Set MONGODB_URI (default: +mongodb://admin:mongodb@localhost:27017/) or the tests will be skipped. +""" + +import os +import sys +import pytest +from pathlib import Path +from pymongo import MongoClient +from bson import ObjectId + +MONGODB_URI = os.environ.get("MONGODB_URI", "mongodb://admin:mongodb@localhost:27017/") +TEST_DB = "hr_chatbot_integration_test" + + +@pytest.fixture(scope="module") +def db(): + client = MongoClient(MONGODB_URI, serverSelectionTimeoutMS=2000) + try: + client.admin.command("ping") + except Exception: + client.close() + pytest.skip(f"MongoDB not reachable at {MONGODB_URI}") + database = client[TEST_DB] + yield database + client.drop_database(TEST_DB) + client.close() + + +def test_mongodb_ping(): + client = MongoClient(MONGODB_URI, serverSelectionTimeoutMS=2000) + try: + result = client.admin.command("ping") + assert result.get("ok") == 1.0 + except Exception: + pytest.skip(f"MongoDB not reachable at {MONGODB_URI}") + finally: + client.close() + + +def test_get_mongo_client_with_real_uri(): + """get_mongo_client() returns a connected client when given a real URI.""" + try: + connect_mod_path = Path(__file__).resolve().parents[1] / "mongodb" / "connect.py" + + import types + import importlib.util + + # Stub langchain_mongodb if not available + if "langchain_mongodb" not in sys.modules: + stub = types.ModuleType("langchain_mongodb") + chm_stub = types.ModuleType("langchain_mongodb.chat_message_histories") + chm_stub.MongoDBChatMessageHistory = type( + "MongoDBChatMessageHistory", + (), + {"__init__": lambda self, *a, **kw: None}, + ) + stub.chat_message_histories = chm_stub + sys.modules["langchain_mongodb"] = stub + sys.modules["langchain_mongodb.chat_message_histories"] = chm_stub + + if "dotenv" not in sys.modules: + dotenv_stub = types.ModuleType("dotenv") + dotenv_stub.load_dotenv = lambda *a, **kw: None + sys.modules["dotenv"] = dotenv_stub + + spec = importlib.util.spec_from_file_location("hr_connect_int", connect_mod_path) + mod = importlib.util.module_from_spec(spec) + os.environ["MONGO_URI"] = MONGODB_URI + spec.loader.exec_module(mod) + + client = mod.get_mongo_client(MONGODB_URI) + assert client is not None + + result = client.admin.command("ping") + assert result.get("ok") == 1.0 + client.close() + except Exception as exc: + pytest.skip(f"App-level test skipped: {exc}") + + +def test_chat_history_collection_crud(db): + """history collection: store and retrieve chat messages.""" + history = db["history"] + + session_id = f"test_session_{ObjectId()}" + messages = [ + { + "_id": ObjectId(), + "SessionId": session_id, + "History": "Human: Hello\nAI: Hi there!", + }, + { + "_id": ObjectId(), + "SessionId": session_id, + "History": "Human: What is MongoDB?\nAI: A NoSQL database.", + }, + ] + history.insert_many(messages) + + session_messages = list(history.find({"SessionId": session_id})) + assert len(session_messages) == 2 + assert all(m["SessionId"] == session_id for m in session_messages) + + # Cleanup + history.delete_many({"SessionId": session_id}) + + +def test_employee_record_crud(db): + """employee records collection: store and retrieve HR data.""" + employees = db["employees"] + + emp_id = ObjectId() + employee = { + "_id": emp_id, + "name": "Test Employee", + "department": "Engineering", + "role": "Developer", + "salary": 90000, + "start_date": "2023-01-15", + } + + employees.insert_one(employee) + + found = employees.find_one({"_id": emp_id}) + assert found["name"] == "Test Employee" + assert found["department"] == "Engineering" + + # Update + employees.update_one({"_id": emp_id}, {"$set": {"salary": 95000}}) + updated = employees.find_one({"_id": emp_id}) + assert updated["salary"] == 95000 + + # Delete + employees.delete_one({"_id": emp_id}) + assert employees.find_one({"_id": emp_id}) is None From 0a3a0f61ae84a33c2e1b0303e654bd39a05c4c91 Mon Sep 17 00:00:00 2001 From: sis0k0 Date: Fri, 22 May 2026 16:17:15 +0300 Subject: [PATCH 7/7] fix: add missing stubs so runtime test runs without app deps installed --- tests/test_runtime.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/tests/test_runtime.py b/tests/test_runtime.py index 0ab1d0c..7d4ce0d 100644 --- a/tests/test_runtime.py +++ b/tests/test_runtime.py @@ -18,8 +18,23 @@ def __init__(self, uri, session_id, database_name=None, collection_name=None): self.collection_name = collection_name histories.MongoDBChatMessageHistory = MongoDBChatMessageHistory + sys.modules["langchain_mongodb"] = types.ModuleType("langchain_mongodb") sys.modules["langchain_mongodb.chat_message_histories"] = histories + dotenv = types.ModuleType("dotenv") + dotenv.load_dotenv = lambda *a, **kw: None + sys.modules["dotenv"] = dotenv + + class MongoClient: + def __init__(self, *a, **kw): + self.admin = types.SimpleNamespace(command=lambda c: {"ok": 1}) + def close(self): pass + pymongo_client_mod = types.ModuleType("pymongo.mongo_client") + pymongo_client_mod.MongoClient = MongoClient + sys.modules["pymongo"] = types.ModuleType("pymongo") + sys.modules["pymongo"].MongoClient = MongoClient + sys.modules["pymongo.mongo_client"] = pymongo_client_mod + target = Path(__file__).resolve().parents[1] / "mongodb" / "connect.py" spec = importlib.util.spec_from_file_location("hr_connect", target) cls.mod = importlib.util.module_from_spec(spec)