ready for test

backend/tests/core/test_mapping_service.py

@@ -0,0 +1,99 @@
+# [DEF:backend.tests.core.test_mapping_service:Module]
+#
+# @TIER:      STANDARD
+# @PURPOSE:   Unit tests for the IdMappingService matching UUIDs to integer IDs.
+# @LAYER:     Domain
+# @RELATION:  VERIFIES -> backend.src.core.mapping_service
+#
+import pytest
+from datetime import datetime, timezone
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker
+import sys
+import os
+from pathlib import Path
+
+# Add backend directory to sys.path so 'src' can be resolved
+backend_dir = str(Path(__file__).parent.parent.parent.resolve())
+if backend_dir not in sys.path:
+    sys.path.insert(0, backend_dir)
+
+from src.models.mapping import Base, ResourceMapping, ResourceType
+from src.core.mapping_service import IdMappingService
+
+@pytest.fixture
+def db_session():
+    # In-memory SQLite for testing
+    engine = create_engine('sqlite:///:memory:')
+    Base.metadata.create_all(engine)
+    Session = sessionmaker(bind=engine)
+    session = Session()
+    yield session
+    session.close()
+
+class MockSupersetClient:
+    def __init__(self, resources):
+        self.resources = resources
+
+    def get_all_resources(self, endpoint):
+        return self.resources.get(endpoint, [])
+
+def test_sync_environment_upserts_correctly(db_session):
+    service = IdMappingService(db_session)
+    mock_client = MockSupersetClient({
+        "chart": [
+            {"id": 42, "uuid": "123e4567-e89b-12d3-a456-426614174000", "slice_name": "Test Chart"}
+        ]
+    })
+
+    service.sync_environment("test-env", mock_client)
+    
+    mapping = db_session.query(ResourceMapping).first()
+    assert mapping is not None
+    assert mapping.environment_id == "test-env"
+    assert mapping.resource_type == ResourceType.CHART
+    assert mapping.uuid == "123e4567-e89b-12d3-a456-426614174000"
+    assert mapping.remote_integer_id == "42"
+    assert mapping.resource_name == "Test Chart"
+
+def test_get_remote_id_returns_integer(db_session):
+    service = IdMappingService(db_session)
+    mapping = ResourceMapping(
+        environment_id="test-env",
+        resource_type=ResourceType.DATASET,
+        uuid="uuid-1",
+        remote_integer_id="99",
+        resource_name="Test DS",
+        last_synced_at=datetime.now(timezone.utc)
+    )
+    db_session.add(mapping)
+    db_session.commit()
+
+    result = service.get_remote_id("test-env", ResourceType.DATASET, "uuid-1")
+    assert result == 99
+
+def test_get_remote_ids_batch_returns_dict(db_session):
+    service = IdMappingService(db_session)
+    m1 = ResourceMapping(
+        environment_id="test-env",
+        resource_type=ResourceType.DASHBOARD,
+        uuid="uuid-1",
+        remote_integer_id="11"
+    )
+    m2 = ResourceMapping(
+        environment_id="test-env",
+        resource_type=ResourceType.DASHBOARD,
+        uuid="uuid-2",
+        remote_integer_id="22"
+    )
+    db_session.add_all([m1, m2])
+    db_session.commit()
+
+    result = service.get_remote_ids_batch("test-env", ResourceType.DASHBOARD, ["uuid-1", "uuid-2", "uuid-missing"])
+    
+    assert len(result) == 2
+    assert result["uuid-1"] == 11
+    assert result["uuid-2"] == 22
+    assert "uuid-missing" not in result
+
+# [/DEF:backend.tests.core.test_mapping_service:Module]