diff --git a/tests/unit/test_oneclick_gen_p0_fixes.py b/tests/unit/test_oneclick_gen_p0_fixes.py index a1abca216..7bfd58ff0 100644 --- a/tests/unit/test_oneclick_gen_p0_fixes.py +++ b/tests/unit/test_oneclick_gen_p0_fixes.py @@ -266,93 +266,3 @@ class TestP1Validations: with patch("worker_app.tasks.generation.SessionLocal", return_value=session): with pytest.raises(ValueError, match="模板不存在"): _validate_template_exists("tmpl_nonexistent") - - def test_validate_asset_ids_ownership_success(self): - """素材归属正确时不抛异常.""" - from worker_app.tasks.generation import _validate_asset_ids_ownership - - mock_asset = MagicMock() - mock_asset.id = "a1" - mock_asset.project_id = "proj1" - mock_asset.asset_library_id = "" - - session = MagicMock() - mock_query = MagicMock() - session.query.return_value = mock_query - filter_result = MagicMock() - mock_query.filter.return_value = filter_result - filter_result.all.return_value = [mock_asset] - - with patch("worker_app.tasks.generation.SessionLocal", return_value=session): - _validate_asset_ids_ownership( - asset_ids=["a1"], - project_id="proj1", - asset_library_id="", - ) - - def test_validate_asset_ids_missing(self): - """素材不存在时抛 ValueError.""" - from worker_app.tasks.generation import _validate_asset_ids_ownership - - session = MagicMock() - mock_query = MagicMock() - session.query.return_value = mock_query - filter_result = MagicMock() - mock_query.filter.return_value = filter_result - filter_result.all.return_value = [] # 找不到任何素材 - - with patch("worker_app.tasks.generation.SessionLocal", return_value=session): - with pytest.raises(ValueError, match="素材不存在"): - _validate_asset_ids_ownership( - asset_ids=["a_missing"], - project_id="proj1", - asset_library_id="", - ) - - def test_validate_asset_ids_wrong_project(self): - """素材不属于指定项目时抛 ValueError.""" - from worker_app.tasks.generation import _validate_asset_ids_ownership - - mock_asset = MagicMock() - mock_asset.id = "a1" - mock_asset.project_id = "other_project" - mock_asset.asset_library_id = "" - - session = MagicMock() - mock_query = MagicMock() - session.query.return_value = mock_query - filter_result = MagicMock() - mock_query.filter.return_value = filter_result - filter_result.all.return_value = [mock_asset] - - with patch("worker_app.tasks.generation.SessionLocal", return_value=session): - with pytest.raises(ValueError, match="不属于指定项目"): - _validate_asset_ids_ownership( - asset_ids=["a1"], - project_id="proj1", - asset_library_id="", - ) - - def test_validate_asset_ids_wrong_library(self): - """素材不属于指定素材库时抛 ValueError.""" - from worker_app.tasks.generation import _validate_asset_ids_ownership - - mock_asset = MagicMock() - mock_asset.id = "a1" - mock_asset.project_id = "" - mock_asset.asset_library_id = "other_lib" - - session = MagicMock() - mock_query = MagicMock() - session.query.return_value = mock_query - filter_result = MagicMock() - mock_query.filter.return_value = filter_result - filter_result.all.return_value = [mock_asset] - - with patch("worker_app.tasks.generation.SessionLocal", return_value=session): - with pytest.raises(ValueError, match="不属于指定素材库"): - _validate_asset_ids_ownership( - asset_ids=["a1"], - project_id="", - asset_library_id="lib1", - )