[30388f42] feat: Add automated test infrastructure for core services

2026-03-08 13:41:58 +00:00
parent 3a6183a85e
commit c467d62580
23 changed files with 420 additions and 8 deletions
--- a/company-explorer/backend/tests/init.py
+++ b/company-explorer/backend/tests/init.py
--- a/company-explorer/backend/tests/test_api_integration.py
+++ b/company-explorer/backend/tests/test_api_integration.py
@@ -0,0 +1,107 @@
+import pytest
+from fastapi.testclient import TestClient
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker
+from sqlalchemy.pool import StaticPool
+from unittest.mock import MagicMock, patch
+import os
+import sys
+
+# Add backend to sys.path to resolve imports
+sys.path.append(os.path.join(os.path.dirname(__file__), "../.."))
+
+from backend.app import app, get_db, authenticate_user
+from backend.database import Base
+
+# --- Test Database Setup ---
+SQLALCHEMY_DATABASE_URL = "sqlite:///:memory:"
+
+engine = create_engine(
+    SQLALCHEMY_DATABASE_URL,
+    connect_args={"check_same_thread": False},
+    poolclass=StaticPool,
+)
+TestingSessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
+
+def override_get_db():
+    try:
+        db = TestingSessionLocal()
+        yield db
+    finally:
+        db.close()
+
+# Override Auth to bypass password check for tests
+def override_authenticate_user():
+    return "test_user"
+
+app.dependency_overrides[get_db] = override_get_db
+app.dependency_overrides[authenticate_user] = override_authenticate_user
+
+@pytest.fixture(scope="module")
+def client():
+    # Create tables
+    Base.metadata.create_all(bind=engine)
+    with TestClient(app) as c:
+        yield c
+    # Drop tables
+    Base.metadata.drop_all(bind=engine)
+
+# --- Tests ---
+
+def test_health_check(client):
+    response = client.get("/api/health")
+    assert response.status_code == 200
+    assert response.json()["status"] == "ok"
+
+@patch("backend.app.run_discovery_task") # Mock the background task function directly
+def test_provision_new_company(mock_discovery_task, client):
+    # Data simulating a webhook from Connector
+    payload = {
+        "so_contact_id": 12345,
+        "so_person_id": 999,
+        "crm_name": "Test Firma GmbH",
+        "crm_website": "www.test-firma.de",
+        "job_title": "CTO",
+        "campaign_tag": "standard"
+    }
+
+    # Call the API
+    response = client.post("/api/provision/superoffice-contact", json=payload)
+
+    # Assertions
+    assert response.status_code == 200
+    data = response.json()
+    assert data["company_name"] == "Test Firma GmbH"
+    
+    # Verify Background Task was triggered
+    # Note: FastAPI TestClient runs background tasks synchronously by default!
+    # So patching the function in app.py is the right way to intercept it.
+    mock_discovery_task.assert_called()
+
+def test_create_contact_manual(client):
+    # First create a company to attach contact to
+    # We can use the DB session directly or the API
+    from backend.database import Company
+    db = TestingSessionLocal()
+    company = Company(name="Manual Corp", status="NEW")
+    db.add(company)
+    db.commit()
+    db.refresh(company)
+    company_id = company.id
+    db.close()
+
+    payload = {
+        "company_id": company_id,
+        "first_name": "Hans",
+        "last_name": "Tester",
+        "email": "hans@manual.corp",
+        "job_title": "CEO",
+        "role": "Wirtschaftlicher Entscheider",
+        "is_primary": True
+    }
+
+    response = client.post("/api/contacts", json=payload)
+    assert response.status_code == 200
+    data = response.json()
+    assert data["email"] == "hans@manual.corp"
+    assert data["role"] == "Wirtschaftlicher Entscheider"
--- a/company-explorer/frontend/index.html
+++ b/company-explorer/frontend/index.html
@@ -3,7 +3,8 @@
  <head>
    <meta charset="UTF-8" />
    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
-    <title>Company Explorer (Robotics)</title>
+    <link rel="icon" href="data:image/svg+xml,<svg xmlns=%22http://www.w3.org/2000/svg%22 viewBox=%220 0 100 100%22><text y=%22.9em%22 font-size=%2290%22>🤖</text></svg>">
+    <title>Company Explorer</title>
  </head>
  <body class="bg-slate-950 text-slate-100">
    <div id="root"></div>
--- a/company-explorer/requirements.txt
+++ b/company-explorer/requirements.txt
@@ -9,4 +9,6 @@ python-multipart
 thefuzz
 wikipedia
 pydantic
-pydantic-settings
+pydantic-settings
+pytest
+httpx
--- a/company-explorer/run_tests.sh
+++ b/company-explorer/run_tests.sh
@@ -0,0 +1,3 @@
+#!/bin/bash
+export PYTHONPATH=$PYTHONPATH:/app
+python3 -m pytest -v /app/backend/tests/test_api_integration.py