Initial curated ITWorx_MCP_Hub source from 368fba691ea8d23403b4d56b53b5ec670d7c5020

tests/test_ingestion_activation.py

@@ -0,0 +1,70 @@
+from __future__ import annotations
+
+import importlib
+import json
+
+
+def test_ingestion_catalog_project_and_gateway_are_opt_in(monkeypatch):
+    from pathlib import Path
+
+    from services.runtime import state_sync
+
+    catalog = Path(__file__).resolve().parents[1] / "catalog"
+    monkeypatch.delenv("INGESTION_CONNECTOR_ENABLED", raising=False)
+    module = importlib.reload(state_sync)
+    assert all(slug != "itworx-ingestion" for slug, _ in module.CONNECTOR_GATEWAYS)
+    assert all(
+        item["metadata"]["slug"] != "knowledge-hub"
+        for item in module._project_documents(catalog)
+    )
+
+    monkeypatch.setenv("INGESTION_CONNECTOR_ENABLED", "true")
+    module = importlib.reload(state_sync)
+    assert ("itworx-ingestion", "http://connector-ingestion:8080/mcp") in module.CONNECTOR_GATEWAYS
+    assert any(
+        item["metadata"]["slug"] == "knowledge-hub"
+        for item in module._project_documents(catalog)
+    )
+
+    monkeypatch.delenv("INGESTION_CONNECTOR_ENABLED", raising=False)
+    importlib.reload(state_sync)
+
+
+def test_disabled_ingestion_binding_is_ignored_while_feature_is_off(monkeypatch):
+    from pathlib import Path
+
+    from services.control_api.persistence import Base, create_engine_from_url, session_factory
+    from services.control_api.persistence.models import ClientBindingRow, ClientRow
+    from services.runtime import state_sync
+
+    catalog = Path(__file__).resolve().parents[1] / "catalog"
+    binding = [{
+        "clientId": "11111111-1111-4111-8111-111111111111",
+        "subject": "disabled-knowledge-intake",
+        "project": "knowledge-hub",
+        "environment": "prod",
+        "enabled": False,
+    }]
+    with monkeypatch.context() as disabled_feature:
+        disabled_feature.setenv("INGESTION_CONNECTOR_ENABLED", "false")
+        disabled_feature.setenv("MCP_CLIENT_BINDINGS_JSON", json.dumps(binding))
+        module = importlib.reload(state_sync)
+        engine = create_engine_from_url("sqlite:///:memory:")
+        Base.metadata.create_all(engine)
+        sessions = session_factory(engine)
+        try:
+            with sessions() as session, session.begin():
+                module.sync_runtime_state(
+                    session,
+                    catalog,
+                    "https://auth.itworx.test/application/o/itworx/",
+                )
+            with sessions() as session:
+                assert session.query(ClientRow).filter_by(
+                    subject="disabled-knowledge-intake"
+                ).one_or_none() is None
+                assert session.query(ClientBindingRow).one_or_none() is None
+        finally:
+            engine.dispose()
+
+    importlib.reload(state_sync)