[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/b2b-marketing-assistant/tests/test_orchestrator.py
+++ b/b2b-marketing-assistant/tests/test_orchestrator.py
@@ -0,0 +1,57 @@
+import pytest
+import json
+import os
+import sys
+from unittest.mock import patch, MagicMock
+
+# Add current dir to sys.path
+sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+import b2b_marketing_orchestrator as orchestrator
+
+@pytest.fixture
+def mock_gemini_response():
+    return """## Schritt 1: Angebot (WAS)
+
+Kurzresuemee:
+* Test Summary 1
+* Test Summary 2
+
+| Produkt/Loesung | Beschreibung | Kernfunktionen | Differenzierung | Primaere Quelle (URL) |
+|---|---|---|---|---|
+| Test Bot | A great robot | AI | Faster | https://example.com/bot |
+"""
+
+@patch("b2b_marketing_orchestrator.load_api_key", return_value="fake_key")
+@patch("b2b_marketing_orchestrator.get_text_from_url", return_value="<html><body>Test Content</body></html>")
+@patch("b2b_marketing_orchestrator.find_relevant_links", return_value=[])
+@patch("b2b_marketing_orchestrator.call_gemini_api")
+def test_start_generation(mock_call, mock_links, mock_scrape, mock_key, mock_gemini_response):
+    mock_call.return_value = mock_gemini_response
+    
+    result = orchestrator.start_generation("https://example.com", "de", "DACH", "Cleaning")
+    
+    assert "offer" in result
+    assert result["offer"]["headers"] == ["Produkt/Loesung", "Beschreibung", "Kernfunktionen", "Differenzierung", "Primaere Quelle (URL)"]
+    assert len(result["offer"]["rows"]) == 1
+    assert result["offer"]["rows"][0][0] == "Test Bot"
+    # Note: Orchestrator uses a hardcoded summary for Step 1 in its return dict
+    assert "Angebotsanalyse" in result["offer"]["summary"][0]
+
+def test_parse_markdown_table():
+    md = """
+| Header 1 | Header 2 |
+|---|---|
+| Val 1 | Val 2 |
+"""
+    result = orchestrator.parse_markdown_table(md)
+    assert result["headers"] == ["Header 1", "Header 2"]
+    assert result["rows"] == [["Val 1", "Val 2"]]
+
+@patch("b2b_marketing_orchestrator.load_api_key", return_value="fake_key")
+@patch("b2b_marketing_orchestrator.call_gemini_api", return_value="Test Product, Nice Bot, AI, Best, https://test.com")
+def test_enrich_product(mock_call, mock_key):
+    result = orchestrator.enrich_product("Test Product", "https://test.com", "de")
+    assert len(result) == 5
+    assert result[0] == "Test Product"
+    assert result[1] == "Nice Bot"