100 lines
3.1 KiB
Python
100 lines
3.1 KiB
Python
# [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]
|