Check uniqueness

This commit is contained in:
Beto Dealmeida
2026-03-10 15:46:54 -04:00
parent 3730bc0f20
commit ef24357de7
2 changed files with 125 additions and 0 deletions

View File

@@ -33,6 +33,7 @@ from superset.commands.semantic_layer.exceptions import (
from superset.daos.semantic_layer import SemanticViewDAO
from superset.exceptions import SupersetSecurityException
from superset.semantic_layers.models import SemanticView
from superset.utils import json
from superset.utils.decorators import on_error, transaction
logger = logging.getLogger(__name__)
@@ -65,3 +66,20 @@ class UpdateSemanticViewCommand(BaseCommand):
security_manager.raise_for_ownership(self._model)
except SupersetSecurityException as ex:
raise SemanticViewForbiddenError() from ex
name = self._properties.get("name", self._model.name)
layer_uuid = str(self._model.semantic_layer_uuid)
configuration = self._properties.get(
"configuration",
json.loads(self._model.configuration),
)
if not SemanticViewDAO.validate_update_uniqueness(
view_uuid=str(self._model.uuid),
name=name,
layer_uuid=layer_uuid,
configuration=configuration,
):
raise ValueError(
f"A semantic view with name '{name}' and the same "
"configuration already exists in this semantic layer."
)

View File

@@ -32,6 +32,7 @@ def test_update_semantic_view_success(mocker: MockerFixture) -> None:
"""Test successful update of a semantic view."""
mock_model = MagicMock()
mock_model.id = 1
mock_model.configuration = "{}"
dao = mocker.patch(
"superset.commands.semantic_layer.update.SemanticViewDAO",
@@ -86,6 +87,7 @@ def test_update_semantic_view_forbidden(mocker: MockerFixture) -> None:
def test_update_semantic_view_copies_data(mocker: MockerFixture) -> None:
"""Test that the command copies input data and does not mutate it."""
mock_model = MagicMock()
mock_model.configuration = "{}"
dao = mocker.patch(
"superset.commands.semantic_layer.update.SemanticViewDAO",
@@ -102,3 +104,108 @@ def test_update_semantic_view_copies_data(mocker: MockerFixture) -> None:
# The original dict should not have been modified
assert original_data == {"description": "Original"}
def _make_view_model(
uuid: str = "view-uuid-1",
name: str = "my_view",
layer_uuid: str = "layer-uuid-1",
configuration: str = '{"schema": "prod"}',
) -> MagicMock:
model = MagicMock()
model.uuid = uuid
model.name = name
model.semantic_layer_uuid = layer_uuid
model.configuration = configuration
return model
def test_update_uniqueness_different_config_same_name(
mocker: MockerFixture,
) -> None:
"""Same name but different configuration is allowed."""
mock_model = _make_view_model(configuration='{"schema": "prod"}')
dao = mocker.patch(
"superset.commands.semantic_layer.update.SemanticViewDAO",
)
dao.find_by_id.return_value = mock_model
dao.update.return_value = mock_model
dao.validate_update_uniqueness.return_value = True
mocker.patch(
"superset.commands.semantic_layer.update.security_manager",
)
# Update to a config that differs from an existing view
data = {"name": "my_view", "configuration": {"schema": "testing"}}
result = UpdateSemanticViewCommand(1, data).run()
assert result == mock_model
dao.validate_update_uniqueness.assert_called_once_with(
view_uuid="view-uuid-1",
name="my_view",
layer_uuid="layer-uuid-1",
configuration={"schema": "testing"},
)
def test_update_uniqueness_same_config_different_name(
mocker: MockerFixture,
) -> None:
"""Same configuration but different name is allowed."""
mock_model = _make_view_model(configuration='{"schema": "prod"}')
dao = mocker.patch(
"superset.commands.semantic_layer.update.SemanticViewDAO",
)
dao.find_by_id.return_value = mock_model
dao.update.return_value = mock_model
dao.validate_update_uniqueness.return_value = True
mocker.patch(
"superset.commands.semantic_layer.update.security_manager",
)
data = {"name": "renamed_view", "configuration": {"schema": "prod"}}
result = UpdateSemanticViewCommand(1, data).run()
assert result == mock_model
dao.validate_update_uniqueness.assert_called_once_with(
view_uuid="view-uuid-1",
name="renamed_view",
layer_uuid="layer-uuid-1",
configuration={"schema": "prod"},
)
def test_update_uniqueness_same_config_same_name_fails(
mocker: MockerFixture,
) -> None:
"""Same name and same configuration is a duplicate."""
mock_model = _make_view_model(configuration='{"schema": "prod"}')
dao = mocker.patch(
"superset.commands.semantic_layer.update.SemanticViewDAO",
)
dao.find_by_id.return_value = mock_model
dao.validate_update_uniqueness.return_value = False
mocker.patch(
"superset.commands.semantic_layer.update.security_manager",
)
from superset.commands.semantic_layer.exceptions import (
SemanticViewUpdateFailedError,
)
data = {"name": "my_view", "configuration": {"schema": "prod"}}
with pytest.raises(SemanticViewUpdateFailedError):
UpdateSemanticViewCommand(1, data).run()
dao.validate_update_uniqueness.assert_called_once_with(
view_uuid="view-uuid-1",
name="my_view",
layer_uuid="layer-uuid-1",
configuration={"schema": "prod"},
)