diff --git a/tests/unit/test_ai_avatar_render_service.py b/tests/unit/test_ai_avatar_render_service.py index a23ac997d..62cf82691 100644 --- a/tests/unit/test_ai_avatar_render_service.py +++ b/tests/unit/test_ai_avatar_render_service.py @@ -549,13 +549,16 @@ class TestAiAvatarRenderService: patch.object(svc, "_upload_to_oss", side_effect=lambda path, key: f"https://oss/{key}"), patch("os.system", return_value=0), patch("tempfile.TemporaryDirectory") as tmpdir_mock, - patch("app.services.ai_avatar_cover_service.generate_smart_cover", return_value="https://oss/smart_cover.jpg"), + patch( + "app.services.ai_avatar_cover_service.generate_smart_cover", return_value="https://oss/smart_cover.jpg" + ), patch("packages.domain.generated_video.GeneratedVideo.create") as gv_create, patch( "packages.adapters.sqlalchemy_impl.generated_video_repository.SQLAlchemyGeneratedVideoRepository" ) as repo_cls, ): import tempfile as _tf + tmpdir_mock.return_value.__enter__ = MagicMock(return_value="/tmp/testdir") tmpdir_mock.return_value.__exit__ = MagicMock(return_value=False) @@ -604,8 +607,7 @@ class TestAiAvatarRenderService: patch.object(svc, "_upload_to_oss", side_effect=lambda path, key: f"https://oss/{key}"), patch("os.system", return_value=0), patch("tempfile.TemporaryDirectory") as tmpdir_mock, - patch("app.services.ai_avatar_cover_service.generate_smart_cover", - side_effect=RuntimeError("DB error")), + patch("app.services.ai_avatar_cover_service.generate_smart_cover", side_effect=RuntimeError("DB error")), ): tmpdir_mock.return_value.__enter__ = MagicMock(return_value="/tmp/testdir") tmpdir_mock.return_value.__exit__ = MagicMock(return_value=False)