From 3ff133a5a3cd23859c26b394a9e216855634f738 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 --- backend/requirements.ini | 2 +- backend/requirements.txt | 6 ++---- 2 files changed, 3 insertions(+), 5 deletions(-) diff --git a/backend/requirements.ini b/backend/requirements.ini index 9a1f996..1915940 100644 --- a/backend/requirements.ini +++ b/backend/requirements.ini @@ -1,5 +1,5 @@ fastapi[all] -motor[srv] +pymongo[srv] boto3 #aiobotocore pillow \ No newline at end of file diff --git a/backend/requirements.txt b/backend/requirements.txt index c57c49a..c0b405a 100644 --- a/backend/requirements.txt +++ b/backend/requirements.txt @@ -56,8 +56,6 @@ jmespath==1.0.1 # botocore markupsafe==2.1.5 # via jinja2 -motor==3.4.0 - # via -r requirements.ini orjson==3.10.2 # via fastapi pillow==10.3.0 @@ -73,8 +71,8 @@ pydantic-extra-types==2.7.0 # via fastapi pydantic-settings==2.2.1 # via fastapi -pymongo==4.7.1 - # via motor +pymongo==4.13.1 + # via -r requirements.ini python-dateutil==2.9.0.post0 # via botocore python-dotenv==1.0.1 From 3a5f6a75b15c54301a9001b943f91e44816cc892 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 9d5d7033a0f66f2adf2fc3b9412897a49adab7ae Mon Sep 17 00:00:00 2001 From: sis0k0 Date: Fri, 22 May 2026 15:26:50 +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 64fb9d25eae397cd9fb88cac009f249f9a95ddfe Mon Sep 17 00:00:00 2001 From: sis0k0 Date: Fri, 22 May 2026 15:30:31 +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 8702cbcb5fb8714842141e0fec1a1c213ee869fb 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 | 46 ++++++++++++ 3 files changed, 48 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..f9fa058 --- /dev/null +++ b/tests/test_runtime.py @@ -0,0 +1,46 @@ +import base64 +import importlib.util +import io +import os +import sys +import types +import unittest +from pathlib import Path + +from PIL import Image + + +class RuntimeTest(unittest.TestCase): + @classmethod + def setUpClass(cls): + os.environ.setdefault("MONGODB_URI", "mongodb://example/test") + os.environ.setdefault("AWS_ACCESS_KEY", "x") + os.environ.setdefault("AWS_SECRET_KEY", "y") + + if "boto3" not in sys.modules: + boto3 = types.ModuleType("boto3") + boto3.client = lambda *args, **kwargs: object() + sys.modules["boto3"] = boto3 + + target = Path(__file__).resolve().parents[1] / "backend" / "src" / "server.py" + spec = importlib.util.spec_from_file_location("celeb_server", target) + cls.mod = importlib.util.module_from_spec(spec) + spec.loader.exec_module(cls.mod) + + def test_standardize_image_and_payload(self): + img = Image.new("RGB", (2, 2), color=(255, 0, 0)) + buff = io.BytesIO() + img.save(buff, format="JPEG") + b64 = base64.b64encode(buff.getvalue()).decode("utf-8") + + out = self.mod.standardize_image(b64) + self.assertIsInstance(out, str) + decoded = base64.b64decode(out) + self.assertGreater(len(decoded), 0) + + payload = self.mod.SearchPayload(img="data:image/jpeg;base64," + b64, compareWithOtherAttendees=False) + self.assertFalse(payload.compareWithOtherAttendees) + + +if __name__ == "__main__": + unittest.main() From 5e39e797b34a79f81209a8ef0195c7fe34cea8e2 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 | 172 ++++++++++++++++++++++++++++++++++++++ 3 files changed, 194 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..798c583 --- /dev/null +++ b/tests/test_integration.py @@ -0,0 +1,172 @@ +"""Integration tests for celeb-matcher-farm. + +Tests real MongoDB CRUD operations for the attendee/celeb_images collection +used by the FastAPI + AWS Bedrock celebrity matching backend. + +Requires a running MongoDB instance. Set MONGODB_URI (default: +mongodb://admin:mongodb@localhost:27017/) or the tests will be skipped. +""" + +import os +import base64 +import pytest +from pymongo import MongoClient +from bson import ObjectId + +MONGODB_URI = os.environ.get("MONGODB_URI", "mongodb://admin:mongodb@localhost:27017/") +TEST_DB = "celeb_matcher_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_attendee_record_crud(db): + """celeb_images collection: insert, find, update, delete an attendee.""" + celeb_images = db["celeb_images"] + + # Create a fake embedding (1024 floats simulating Bedrock output) + embedding = [0.01 * i for i in range(1024)] + + doc_id = ObjectId() + record = { + "_id": doc_id, + "name": "Test Attendee", + "email": "attendee@example.com", + "embedding": embedding, + "image_b64": base64.b64encode(b"fake-image-bytes").decode(), + } + + # Create + result = celeb_images.insert_one(record) + assert result.inserted_id == doc_id + + # Read + found = celeb_images.find_one({"_id": doc_id}) + assert found["name"] == "Test Attendee" + assert len(found["embedding"]) == 1024 + + # Update + celeb_images.update_one({"_id": doc_id}, {"$set": {"name": "Updated Attendee"}}) + updated = celeb_images.find_one({"_id": doc_id}) + assert updated["name"] == "Updated Attendee" + + # Delete + delete_result = celeb_images.delete_one({"_id": doc_id}) + assert delete_result.deleted_count == 1 + assert celeb_images.find_one({"_id": doc_id}) is None + + +def test_query_attendees_by_email(db): + """celeb_images collection: find by email field.""" + celeb_images = db["celeb_images"] + + ids = [ObjectId(), ObjectId()] + docs = [ + { + "_id": ids[0], + "name": "Alice", + "email": "alice@conference.example", + "embedding": [0.1] * 1024, + }, + { + "_id": ids[1], + "name": "Bob", + "email": "bob@conference.example", + "embedding": [0.2] * 1024, + }, + ] + celeb_images.insert_many(docs) + + found = celeb_images.find_one({"email": "alice@conference.example", "_id": {"$in": ids}}) + assert found is not None + assert found["name"] == "Alice" + + # Cleanup + celeb_images.delete_many({"_id": {"$in": ids}}) + + +def test_vector_search_index_structure(db): + """celeb_images: verify documents have the embedding field needed for vector search.""" + celeb_images = db["celeb_images"] + + doc_id = ObjectId() + celeb_images.insert_one({ + "_id": doc_id, + "name": "Carol", + "embedding": [float(i) / 1024 for i in range(1024)], + }) + + found = celeb_images.find_one({"_id": doc_id}) + assert "embedding" in found + assert isinstance(found["embedding"], list) + assert len(found["embedding"]) == 1024 + # All values should be valid floats + assert all(isinstance(v, float) for v in found["embedding"]) + + celeb_images.delete_one({"_id": doc_id}) + + +def test_standardize_image_function(): + """standardize_image() produces a valid base64-encoded JPEG string.""" + try: + import sys + from pathlib import Path + import importlib.util + import types + + backend_src = Path(__file__).resolve().parents[1] / "backend" / "src" + + # Stub boto3 and AWS dependencies + for mod_name in ("boto3", "uvicorn"): + if mod_name not in sys.modules: + stub = types.ModuleType(mod_name) + sys.modules[mod_name] = stub + + os.environ.setdefault("MONGODB_URI", MONGODB_URI) + os.environ.setdefault("AWS_ACCESS_KEY", "test") + os.environ.setdefault("AWS_SECRET_KEY", "test") + + spec = importlib.util.spec_from_file_location( + "celeb_server_int", backend_src / "server.py" + ) + mod = importlib.util.module_from_spec(spec) + spec.loader.exec_module(mod) + + # Create a minimal 2x2 JPEG image + from PIL import Image + import io + img = Image.new("RGB", (2, 2), color=(255, 0, 0)) + buf = io.BytesIO() + img.save(buf, format="JPEG") + buf.seek(0) + input_b64 = base64.b64encode(buf.getvalue()).decode() + + result_b64 = mod.standardize_image(input_b64) + decoded = base64.b64decode(result_b64) + out_img = Image.open(io.BytesIO(decoded)) + assert out_img.size == (512, 512) + assert out_img.mode == "RGB" + except Exception as exc: + pytest.skip(f"standardize_image test skipped: {exc}") From 59f61155a781f838c775fc8c244a2d71b5fdae5a 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 --- .github/workflows/ci.yml | 3 +++ 1 file changed, 3 insertions(+) diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index fb942d2..51416b8 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -102,6 +102,9 @@ jobs: echo "Smoke checks passed" + - name: Install runtime test dependencies + run: pip install Pillow fastapi pymongo pydantic uvicorn + - name: Run repository runtime smoke test shell: bash run: |