diff --git a/tests/unit/test_duplicate_rate_scope.py b/tests/unit/test_duplicate_rate_scope.py index 13c6c6d28..845bfe6ad 100644 --- a/tests/unit/test_duplicate_rate_scope.py +++ b/tests/unit/test_duplicate_rate_scope.py @@ -301,8 +301,8 @@ class TestListByUserRepository: from sqlalchemy import create_engine from sqlalchemy.orm import sessionmaker - from packages.adapters.sqlalchemy_impl.models import Base, GeneratedVideoModel from packages.adapters.sqlalchemy_impl.generated_video_repository import SQLAlchemyGeneratedVideoRepository + from packages.adapters.sqlalchemy_impl.models import Base, GeneratedVideoModel engine = create_engine("sqlite:///:memory:") Base.metadata.create_all(engine) @@ -315,11 +315,17 @@ class TestListByUserRepository: from packages.adapters.sqlalchemy_impl.models import GeneratedVideoModel row = GeneratedVideoModel( - id=video_id, user_id=user_id, project_id=project_id, + id=video_id, + user_id=user_id, + project_id=project_id, generation_task_id=f"task-{video_id[:8]}", name=f"video-{video_id[:8]}.mp4", file_url=f"https://example.com/{video_id}.mp4", - file_size=1024, duration=duration, width=1280, height=720, fps=25.0, + file_size=1024, + duration=duration, + width=1280, + height=720, + fps=25.0, status="completed", ) session.add(row)