diff --git a/tests/unit/test_generation_preview.py b/tests/unit/test_generation_preview.py index 11b584146..fa64cf48f 100755 --- a/tests/unit/test_generation_preview.py +++ b/tests/unit/test_generation_preview.py @@ -1173,9 +1173,11 @@ class TestInferVideoRatioFromTemplate: mock_template.mode = "pip" mock_db = MagicMock() - with patch("app.api.routes.generation_preview.SQLAlchemyTemplateRepository") as MockRepo: - MockRepo.return_value.get_by_id.return_value = mock_template - result = _infer_video_ratio_from_template("tpl_001", mock_db) + with patch( + "app.api.routes.generation_preview.SQLAlchemyTemplateRepository" + ) as MockRepo: + MockRepo.return_value.get.return_value = mock_template + result = _infer_video_ratio_from_template("tpl_001", mock_db, "user_1") assert result == "9:16" def test_standard_mode_returns_16_9(self): @@ -1186,9 +1188,11 @@ class TestInferVideoRatioFromTemplate: mock_template.mode = "standard" mock_db = MagicMock() - with patch("app.api.routes.generation_preview.SQLAlchemyTemplateRepository") as MockRepo: - MockRepo.return_value.get_by_id.return_value = mock_template - result = _infer_video_ratio_from_template("tpl_001", mock_db) + with patch( + "app.api.routes.generation_preview.SQLAlchemyTemplateRepository" + ) as MockRepo: + MockRepo.return_value.get.return_value = mock_template + result = _infer_video_ratio_from_template("tpl_001", mock_db, "user_1") assert result == "16:9" def test_unknown_mode_returns_empty(self): @@ -1199,9 +1203,11 @@ class TestInferVideoRatioFromTemplate: mock_template.mode = "unknown_mode" mock_db = MagicMock() - with patch("app.api.routes.generation_preview.SQLAlchemyTemplateRepository") as MockRepo: - MockRepo.return_value.get_by_id.return_value = mock_template - result = _infer_video_ratio_from_template("tpl_001", mock_db) + with patch( + "app.api.routes.generation_preview.SQLAlchemyTemplateRepository" + ) as MockRepo: + MockRepo.return_value.get.return_value = mock_template + result = _infer_video_ratio_from_template("tpl_001", mock_db, "user_1") assert result == "" def test_template_not_found_returns_empty(self): @@ -1210,9 +1216,11 @@ class TestInferVideoRatioFromTemplate: mock_db = MagicMock() - with patch("app.api.routes.generation_preview.SQLAlchemyTemplateRepository") as MockRepo: - MockRepo.return_value.get_by_id.return_value = None - result = _infer_video_ratio_from_template("nonexistent", mock_db) + with patch( + "app.api.routes.generation_preview.SQLAlchemyTemplateRepository" + ) as MockRepo: + MockRepo.return_value.get.return_value = None + result = _infer_video_ratio_from_template("nonexistent", mock_db, "user_1") assert result == "" def test_empty_template_id_returns_empty(self): @@ -1220,7 +1228,7 @@ class TestInferVideoRatioFromTemplate: from app.api.routes.generation_preview import _infer_video_ratio_from_template mock_db = MagicMock() - result = _infer_video_ratio_from_template("", mock_db) + result = _infer_video_ratio_from_template("", mock_db, "user_1") assert result == "" def test_db_exception_returns_empty(self): @@ -1229,9 +1237,11 @@ class TestInferVideoRatioFromTemplate: mock_db = MagicMock() - with patch("app.api.routes.generation_preview.SQLAlchemyTemplateRepository") as MockRepo: + with patch( + "app.api.routes.generation_preview.SQLAlchemyTemplateRepository" + ) as MockRepo: MockRepo.side_effect = Exception("db connection error") - result = _infer_video_ratio_from_template("tpl_001", mock_db) + result = _infer_video_ratio_from_template("tpl_001", mock_db, "user_1") assert result == "" @@ -1264,9 +1274,13 @@ class TestPreviewRouteAutoInfersVideoRatio: bgm_config={}, ) - with patch("app.api.routes.generation_preview.SQLAlchemyTemplateRepository") as MockRepo: - MockRepo.return_value.get_by_id.return_value = mock_template - with patch("app.api.routes.generation_preview.CreateGenerationTaskUseCase") as MockUC: + with patch( + "app.api.routes.generation_preview.SQLAlchemyTemplateRepository" + ) as MockRepo: + MockRepo.return_value.get.return_value = mock_template + with patch( + "app.api.routes.generation_preview.CreateGenerationTaskUseCase" + ) as MockUC: MockUC.return_value.execute.return_value = task with patch( "app.api.routes.generation_preview.safe_enqueue_generation_task",