CI Bot
142 lines
5.0 KiB
Python
Executable File
142 lines
5.0 KiB
Python
Executable File
"""generated_video 领域模型单元测试."""
|
|
|
|
import pytest
|
|
from domain.generated_video import GeneratedVideo
|
|
|
|
|
|
class TestGeneratedVideoCreate:
|
|
"""GeneratedVideo.create 工厂方法测试."""
|
|
|
|
def test_create_with_required_fields(self):
|
|
video = GeneratedVideo.create(
|
|
project_id="proj_001",
|
|
generation_task_id="task_001",
|
|
name="测试视频",
|
|
file_url="https://example.com/out.mp4",
|
|
)
|
|
assert video.id
|
|
assert len(video.id) == 32
|
|
assert video.project_id == "proj_001"
|
|
assert video.generation_task_id == "task_001"
|
|
assert video.name == "测试视频"
|
|
assert video.file_url == "https://example.com/out.mp4"
|
|
# 默认值
|
|
assert video.user_id == ""
|
|
assert video.file_size == 0
|
|
assert video.duration == 0.0
|
|
assert video.width == 0
|
|
assert video.height == 0
|
|
assert video.fps == 0.0
|
|
assert video.thumbnail_url is None
|
|
assert video.status == "completed"
|
|
assert video.review_status == "pending_review"
|
|
assert video.generation_params == {}
|
|
assert video.video_fingerprint is None
|
|
assert video.is_duplicate is False
|
|
assert video.duplicate_of is None
|
|
assert video.generated_at is not None
|
|
assert video.created_at is not None
|
|
|
|
def test_create_with_all_fields(self):
|
|
video = GeneratedVideo.create(
|
|
project_id="proj_002",
|
|
generation_task_id="task_002",
|
|
name="完整视频",
|
|
file_url="https://example.com/full.mp4",
|
|
user_id="user_001",
|
|
file_size=1024000,
|
|
duration=30.5,
|
|
width=1920,
|
|
height=1080,
|
|
fps=30.0,
|
|
thumbnail_url="https://example.com/thumb.jpg",
|
|
generation_params={"quality": "high"},
|
|
)
|
|
assert video.user_id == "user_001"
|
|
assert video.file_size == 1024000
|
|
assert video.duration == 30.5
|
|
assert video.width == 1920
|
|
assert video.height == 1080
|
|
assert video.fps == 30.0
|
|
assert video.thumbnail_url == "https://example.com/thumb.jpg"
|
|
assert video.generation_params == {"quality": "high"}
|
|
|
|
def test_create_strips_strings(self):
|
|
video = GeneratedVideo.create(
|
|
project_id=" proj_003 ",
|
|
generation_task_id=" task_003 ",
|
|
name=" 测试视频 ",
|
|
file_url=" https://example.com/out.mp4 ",
|
|
user_id=" user_003 ",
|
|
)
|
|
assert video.project_id == "proj_003"
|
|
assert video.generation_task_id == "task_003"
|
|
assert video.name == "测试视频"
|
|
assert video.file_url == "https://example.com/out.mp4"
|
|
assert video.user_id == "user_003"
|
|
|
|
def test_create_empty_project_id_raises(self):
|
|
with pytest.raises(ValueError, match="project_id"):
|
|
GeneratedVideo.create(
|
|
project_id="",
|
|
generation_task_id="t",
|
|
name="n",
|
|
file_url="u",
|
|
)
|
|
|
|
def test_create_whitespace_project_id_raises(self):
|
|
with pytest.raises(ValueError, match="project_id"):
|
|
GeneratedVideo.create(
|
|
project_id=" ",
|
|
generation_task_id="t",
|
|
name="n",
|
|
file_url="u",
|
|
)
|
|
|
|
def test_create_empty_generation_task_id_raises(self):
|
|
with pytest.raises(ValueError, match="generation_task_id"):
|
|
GeneratedVideo.create(
|
|
project_id="p",
|
|
generation_task_id="",
|
|
name="n",
|
|
file_url="u",
|
|
)
|
|
|
|
def test_create_empty_name_raises(self):
|
|
with pytest.raises(ValueError, match="name"):
|
|
GeneratedVideo.create(
|
|
project_id="p",
|
|
generation_task_id="t",
|
|
name="",
|
|
file_url="u",
|
|
)
|
|
|
|
def test_create_empty_file_url_raises(self):
|
|
with pytest.raises(ValueError, match="file_url"):
|
|
GeneratedVideo.create(
|
|
project_id="p",
|
|
generation_task_id="t",
|
|
name="n",
|
|
file_url="",
|
|
)
|
|
|
|
def test_create_none_generation_params_defaults_to_empty_dict(self):
|
|
video = GeneratedVideo.create(
|
|
project_id="p",
|
|
generation_task_id="t",
|
|
name="n",
|
|
file_url="u",
|
|
generation_params=None,
|
|
)
|
|
assert video.generation_params == {}
|
|
|
|
def test_create_ids_are_unique(self):
|
|
v1 = GeneratedVideo.create(project_id="p", generation_task_id="t1", name="n1", file_url="u1")
|
|
v2 = GeneratedVideo.create(project_id="p", generation_task_id="t2", name="n2", file_url="u2")
|
|
assert v1.id != v2.id
|
|
|
|
def test_create_timestamps_are_utc(self):
|
|
video = GeneratedVideo.create(project_id="p", generation_task_id="t", name="n", file_url="u")
|
|
assert video.created_at.tzinfo is not None
|
|
assert video.generated_at.tzinfo is not None
|