134 lines
5.1 KiB
Python
134 lines
5.1 KiB
Python
# [DEF:SupersetCompatibilityMatrixTests:Module]
|
|
# @COMPLEXITY: 3
|
|
# @SEMANTICS: dataset_review, superset, compatibility_matrix, preview, sql_lab, tests
|
|
# @PURPOSE: Verifies Superset preview and SQL Lab endpoint fallback strategy used by dataset-review orchestration.
|
|
# @LAYER: Tests
|
|
# @RELATION: [DEPENDS_ON] ->[backend.src.core.superset_client.SupersetClient]
|
|
# @RELATION: [DEPENDS_ON] ->[SupersetCompilationAdapter]
|
|
|
|
from types import SimpleNamespace
|
|
from unittest.mock import MagicMock
|
|
|
|
# Import models to ensure proper SQLAlchemy registration
|
|
from src.models.auth import User
|
|
from src.models.dataset_review import CompiledPreview
|
|
|
|
from src.core.utils.superset_compilation_adapter import (
|
|
PreviewCompilationPayload,
|
|
SqlLabLaunchPayload,
|
|
SupersetCompilationAdapter,
|
|
)
|
|
|
|
|
|
# [DEF:make_adapter:Function]
|
|
# @COMPLEXITY: 2
|
|
# @PURPOSE: Build an adapter with a mock Superset client and deterministic environment for compatibility tests.
|
|
# @RELATION: [DEPENDS_ON] ->[SupersetCompilationAdapter]
|
|
def make_adapter():
|
|
environment = SimpleNamespace(
|
|
id="env-1",
|
|
name="Test Env",
|
|
url="http://superset.example",
|
|
username="user",
|
|
password="pass",
|
|
verify_ssl=True,
|
|
timeout=30,
|
|
)
|
|
client = MagicMock()
|
|
client.network = MagicMock()
|
|
return SupersetCompilationAdapter(environment=environment, client=client), client
|
|
# [/DEF:make_adapter:Function]
|
|
|
|
|
|
# [DEF:test_preview_prefers_supported_client_method_before_network_fallback:Function]
|
|
# @COMPLEXITY: 2
|
|
# @PURPOSE: Confirms preview compilation uses a supported client method first when the capability exists.
|
|
# @RELATION: [DEPENDS_ON] ->[SupersetCompilationAdapter]
|
|
def test_preview_prefers_supported_client_method_before_network_fallback():
|
|
adapter, client = make_adapter()
|
|
client.compile_preview = MagicMock(return_value={"compiled_sql": "SELECT 1"})
|
|
payload = PreviewCompilationPayload(
|
|
session_id="sess-1",
|
|
dataset_id=42,
|
|
preview_fingerprint="fp-1",
|
|
template_params={"country": "RU"},
|
|
effective_filters=[{"name": "country", "value": "RU"}],
|
|
)
|
|
|
|
preview = adapter.compile_preview(payload)
|
|
|
|
assert preview.preview_status.value == "ready"
|
|
assert preview.compiled_sql == "SELECT 1"
|
|
client.compile_preview.assert_called_once()
|
|
client.network.request.assert_not_called()
|
|
# [/DEF:test_preview_prefers_supported_client_method_before_network_fallback:Function]
|
|
|
|
|
|
# [DEF:test_preview_falls_back_across_matrix_until_supported_endpoint_returns_sql:Function]
|
|
# @COMPLEXITY: 3
|
|
# @PURPOSE: Confirms preview fallback walks the compatibility matrix from preferred to legacy endpoints until one returns compiled SQL.
|
|
# @RELATION: [DEPENDS_ON] ->[SupersetCompilationAdapter]
|
|
def test_preview_falls_back_across_matrix_until_supported_endpoint_returns_sql():
|
|
adapter, client = make_adapter()
|
|
payload = PreviewCompilationPayload(
|
|
session_id="sess-2",
|
|
dataset_id=77,
|
|
preview_fingerprint="fp-2",
|
|
template_params={"region": "emea"},
|
|
effective_filters=[],
|
|
)
|
|
|
|
client.network.request.side_effect = [
|
|
RuntimeError("preview endpoint unavailable"),
|
|
{"result": {"sql": "SELECT * FROM dataset_77"}},
|
|
]
|
|
|
|
preview = adapter.compile_preview(payload)
|
|
|
|
assert preview.preview_status.value == "ready"
|
|
assert preview.compiled_sql == "SELECT * FROM dataset_77"
|
|
assert client.network.request.call_count == 2
|
|
first_call = client.network.request.call_args_list[0].kwargs
|
|
second_call = client.network.request.call_args_list[1].kwargs
|
|
assert first_call["endpoint"] == "/dataset/77/preview"
|
|
assert second_call["endpoint"] == "/dataset/77/sql"
|
|
# [/DEF:test_preview_falls_back_across_matrix_until_supported_endpoint_returns_sql:Function]
|
|
|
|
|
|
# [DEF:test_sql_lab_launch_falls_back_to_legacy_execute_endpoint:Function]
|
|
# @COMPLEXITY: 3
|
|
# @PURPOSE: Confirms SQL Lab launch falls back from modern to legacy execute endpoint and preserves canonical session reference extraction.
|
|
# @RELATION: [DEPENDS_ON] ->[SupersetCompilationAdapter]
|
|
def test_sql_lab_launch_falls_back_to_legacy_execute_endpoint():
|
|
adapter, client = make_adapter()
|
|
client.get_dataset.return_value = {
|
|
"result": {
|
|
"id": 55,
|
|
"schema": "public",
|
|
"database": {"id": 9},
|
|
}
|
|
}
|
|
client.network.request.side_effect = [
|
|
RuntimeError("sqllab execute unavailable"),
|
|
{"result": {"id": "query-123"}},
|
|
]
|
|
payload = SqlLabLaunchPayload(
|
|
session_id="sess-3",
|
|
dataset_id=55,
|
|
preview_id="preview-9",
|
|
compiled_sql="SELECT * FROM sales",
|
|
template_params={"limit": 10},
|
|
)
|
|
|
|
sql_lab_ref = adapter.create_sql_lab_session(payload)
|
|
|
|
assert sql_lab_ref == "query-123"
|
|
assert client.network.request.call_count == 2
|
|
first_call = client.network.request.call_args_list[0].kwargs
|
|
second_call = client.network.request.call_args_list[1].kwargs
|
|
assert first_call["endpoint"] == "/sqllab/execute/"
|
|
assert second_call["endpoint"] == "/sql_lab/execute/"
|
|
# [/DEF:test_sql_lab_launch_falls_back_to_legacy_execute_endpoint:Function]
|
|
|
|
|
|
# [/DEF:SupersetCompatibilityMatrixTests:Module] |