mirror of
https://github.com/apache/superset.git
synced 2026-05-12 19:35:17 +00:00
Bulk delete
This commit is contained in:
@@ -81,3 +81,35 @@ def test_delete_semantic_view_not_found(mocker: MockerFixture) -> None:
|
||||
|
||||
with pytest.raises(SemanticViewNotFoundError):
|
||||
DeleteSemanticViewCommand(999).run()
|
||||
|
||||
|
||||
def test_bulk_delete_semantic_view_success(mocker: MockerFixture) -> None:
|
||||
"""Test successful bulk deletion of semantic views."""
|
||||
mock_models = [MagicMock(), MagicMock()]
|
||||
|
||||
dao = mocker.patch(
|
||||
"superset.commands.semantic_layer.delete.SemanticViewDAO",
|
||||
)
|
||||
dao.find_by_ids.return_value = mock_models
|
||||
|
||||
from superset.commands.semantic_layer.delete import BulkDeleteSemanticViewCommand
|
||||
|
||||
BulkDeleteSemanticViewCommand([1, 2]).run()
|
||||
|
||||
dao.find_by_ids.assert_called_once_with([1, 2], id_column="id")
|
||||
dao.delete.assert_called_once_with(mock_models)
|
||||
|
||||
|
||||
def test_bulk_delete_semantic_view_not_found(mocker: MockerFixture) -> None:
|
||||
"""Test that SemanticViewNotFoundError is raised when any id is missing."""
|
||||
dao = mocker.patch(
|
||||
"superset.commands.semantic_layer.delete.SemanticViewDAO",
|
||||
)
|
||||
# Only one model returned for two requested ids
|
||||
dao.find_by_ids.return_value = [MagicMock()]
|
||||
|
||||
from superset.commands.semantic_layer.delete import BulkDeleteSemanticViewCommand
|
||||
from superset.commands.semantic_layer.exceptions import SemanticViewNotFoundError
|
||||
|
||||
with pytest.raises(SemanticViewNotFoundError):
|
||||
BulkDeleteSemanticViewCommand([1, 2]).run()
|
||||
|
||||
@@ -1718,6 +1718,73 @@ def test_delete_semantic_view_failed(
|
||||
assert response.status_code == 422
|
||||
|
||||
|
||||
# =============================================================================
|
||||
# SemanticViewRestApi.bulk_delete tests
|
||||
# =============================================================================
|
||||
|
||||
|
||||
@SEMANTIC_LAYERS_APP
|
||||
def test_bulk_delete_semantic_view(
|
||||
client: Any,
|
||||
full_api_access: None,
|
||||
mocker: MockerFixture,
|
||||
) -> None:
|
||||
"""Test DELETE / deletes multiple semantic views and returns a count message."""
|
||||
import prison as rison_lib
|
||||
|
||||
mock_command = mocker.patch(
|
||||
"superset.semantic_layers.api.BulkDeleteSemanticViewCommand",
|
||||
)
|
||||
mock_command.return_value.run.return_value = None
|
||||
|
||||
q = rison_lib.dumps([1, 2, 3])
|
||||
response = client.delete(f"/api/v1/semantic_view/?q={q}")
|
||||
|
||||
assert response.status_code == 200
|
||||
assert "3" in response.json["message"]
|
||||
mock_command.assert_called_once_with([1, 2, 3])
|
||||
|
||||
|
||||
@SEMANTIC_LAYERS_APP
|
||||
def test_bulk_delete_semantic_view_not_found(
|
||||
client: Any,
|
||||
full_api_access: None,
|
||||
mocker: MockerFixture,
|
||||
) -> None:
|
||||
"""Test DELETE / returns 404 when any id is missing."""
|
||||
import prison as rison_lib
|
||||
|
||||
mock_command = mocker.patch(
|
||||
"superset.semantic_layers.api.BulkDeleteSemanticViewCommand",
|
||||
)
|
||||
mock_command.return_value.run.side_effect = SemanticViewNotFoundError()
|
||||
|
||||
q = rison_lib.dumps([1, 999])
|
||||
response = client.delete(f"/api/v1/semantic_view/?q={q}")
|
||||
|
||||
assert response.status_code == 404
|
||||
|
||||
|
||||
@SEMANTIC_LAYERS_APP
|
||||
def test_bulk_delete_semantic_view_failed(
|
||||
client: Any,
|
||||
full_api_access: None,
|
||||
mocker: MockerFixture,
|
||||
) -> None:
|
||||
"""Test DELETE / returns 422 when deletion fails."""
|
||||
import prison as rison_lib
|
||||
|
||||
mock_command = mocker.patch(
|
||||
"superset.semantic_layers.api.BulkDeleteSemanticViewCommand",
|
||||
)
|
||||
mock_command.return_value.run.side_effect = SemanticViewDeleteFailedError()
|
||||
|
||||
q = rison_lib.dumps([1, 2])
|
||||
response = client.delete(f"/api/v1/semantic_view/?q={q}")
|
||||
|
||||
assert response.status_code == 422
|
||||
|
||||
|
||||
# =============================================================================
|
||||
# SemanticLayerRestApi.views tests
|
||||
# =============================================================================
|
||||
|
||||
Reference in New Issue
Block a user