diff --git a/tests/unit/test_domain_entities_extended.py b/tests/unit/test_domain_entities_extended.py new file mode 100755 index 000000000..d638320fa --- /dev/null +++ b/tests/unit/test_domain_entities_extended.py @@ -0,0 +1,523 @@ +"""领域实体测试 — Project + Asset + AssetStatus + IngestJob.""" + +from __future__ import annotations + +import pytest + +from packages.domain.entities import ( + Asset, + AssetLibrary, + AssetStatus, + ClassificationStatus, + IngestJob, + Project, + User, +) + + +class TestUser: + """User 实体测试.""" + + def test_create_user(self): + """创建用户.""" + user = User( + id="u1", + email="test@example.com", + display_name="测试用户", + username="testuser", + ) + assert user.id == "u1" + assert user.email == "test@example.com" + assert user.display_name == "测试用户" + assert user.username == "testuser" + + def test_default_subscription_free(self): + """默认订阅免费版.""" + user = User(id="u1", email="a@b.com", display_name="A") + assert user.subscription_plan == "free" + assert user.subscription_status == "active" + assert user.max_projects == 3 + assert user.max_storage_gb == 10 + + def test_default_not_admin(self): + """默认不是管理员.""" + user = User(id="u1", email="a@b.com", display_name="A") + assert user.is_admin is False + + +class TestProject: + """Project 实体测试.""" + + def test_create_project_success(self): + """创建项目成功.""" + project = Project.create( + owner_user_id="user_1", + name="我的项目", + description="测试项目描述", + ) + assert project.id is not None + assert len(project.id) == 32 # uuid hex + assert project.owner_user_id == "user_1" + assert project.name == "我的项目" + assert project.description == "测试项目描述" + assert project.shared_users == [] + + def test_create_project_name_stripped(self): + """项目名称首尾空白被去除.""" + project = Project.create(owner_user_id="u1", name=" 测试项目 ") + assert project.name == "测试项目" + + def test_create_project_empty_name_raises(self): + """空项目名抛异常.""" + with pytest.raises(ValueError, match="项目名称不能为空"): + Project.create(owner_user_id="u1", name="") + + def test_create_project_whitespace_name_raises(self): + """全空白项目名抛异常.""" + with pytest.raises(ValueError, match="项目名称不能为空"): + Project.create(owner_user_id="u1", name=" ") + + def test_is_owner_true(self): + """是项目所有者.""" + project = Project.create(owner_user_id="owner_1", name="项目") + assert project.is_owner("owner_1") is True + + def test_is_owner_false(self): + """不是项目所有者.""" + project = Project.create(owner_user_id="owner_1", name="项目") + assert project.is_owner("other_user") is False + + def test_is_shared_with_true(self): + """项目已共享给用户.""" + project = Project.create(owner_user_id="u1", name="项目") + project.shared_users.append("user_2") + assert project.is_shared_with("user_2") is True + + def test_is_shared_with_false(self): + """项目未共享给用户.""" + project = Project.create(owner_user_id="u1", name="项目") + assert project.is_shared_with("nobody") is False + + def test_can_access_owner(self): + """所有者可以访问.""" + project = Project.create(owner_user_id="u1", name="项目") + assert project.can_access("u1") is True + + def test_can_access_shared_user(self): + """共享用户可以访问.""" + project = Project.create(owner_user_id="u1", name="项目") + project.shared_users.append("u2") + assert project.can_access("u2") is True + + def test_can_access_outsider(self): + """无关用户不能访问.""" + project = Project.create(owner_user_id="u1", name="项目") + assert project.can_access("stranger") is False + + +class TestAssetLibrary: + """AssetLibrary 实体测试.""" + + def test_create_library(self): + """创建素材库.""" + from packages.domain.classification import AssetLibraryKind + + lib = AssetLibrary.create( + project_id="p1", + name="默认库", + kind=AssetLibraryKind.VIDEO, + ) + assert lib.id is not None + assert lib.project_id == "p1" + assert lib.name == "默认库" + assert lib.kind == AssetLibraryKind.VIDEO + assert lib.asset_count == 0 + + def test_create_default_counts(self): + """默认素材数量和大小为0.""" + from packages.domain.classification import AssetLibraryKind + + lib = AssetLibrary.create( + project_id="p1", + name="我的素材", + kind=AssetLibraryKind.VOICE, + ) + assert lib.asset_count == 0 + assert lib.total_size == 0 + + def test_empty_name_raises(self): + """空名称抛异常.""" + from packages.domain.classification import AssetLibraryKind + + with pytest.raises(ValueError, match="素材库名称不能为空"): + AssetLibrary.create(project_id="p1", name="", kind=AssetLibraryKind.VIDEO) + + +class TestAssetStatus: + """AssetStatus 枚举兼容测试.""" + + def test_direct_values(self): + """直接枚举值.""" + assert AssetStatus.UPLOADING.value == "uploading" + assert AssetStatus.READY.value == "ready" + assert AssetStatus.PROCESSING.value == "processing" + assert AssetStatus.ERROR.value == "error" + assert AssetStatus.DELETED.value == "deleted" + + def test_missing_uploaded_maps_to_ready(self): + """历史值 uploaded → READY.""" + assert AssetStatus("uploaded") == AssetStatus.READY + + def test_missing_success_maps_to_ready(self): + """success → READY.""" + assert AssetStatus("success") == AssetStatus.READY + + def test_missing_ok_maps_to_ready(self): + """ok → READY.""" + assert AssetStatus("ok") == AssetStatus.READY + + def test_missing_done_maps_to_ready(self): + """done → READY.""" + assert AssetStatus("done") == AssetStatus.READY + + def test_missing_upload_maps_to_uploading(self): + """upload → UPLOADING.""" + assert AssetStatus("upload") == AssetStatus.UPLOADING + + def test_missing_uploading_start_maps_to_uploading(self): + """uploading_start → UPLOADING.""" + assert AssetStatus("uploading_start") == AssetStatus.UPLOADING + + def test_missing_failed_maps_to_error(self): + """failed → ERROR.""" + assert AssetStatus("failed") == AssetStatus.ERROR + + def test_missing_fail_maps_to_error(self): + """fail → ERROR.""" + assert AssetStatus("fail") == AssetStatus.ERROR + + def test_missing_process_maps_to_processing(self): + """process → PROCESSING.""" + assert AssetStatus("process") == AssetStatus.PROCESSING + + def test_missing_running_maps_to_processing(self): + """running → PROCESSING.""" + assert AssetStatus("running") == AssetStatus.PROCESSING + + def test_unknown_value_fallback_to_ready(self): + """完全未知值 → READY兜底.""" + assert AssetStatus("completely_unknown") == AssetStatus.READY + + def test_case_insensitive(self): + """大小写不敏感.""" + assert AssetStatus("UPLOADED") == AssetStatus.READY + assert AssetStatus("Failed") == AssetStatus.ERROR + + def test_whitespace_stripped(self): + """首尾空白被去除.""" + assert AssetStatus(" ready ") == AssetStatus.READY + + def test_none_value_fallback(self): + """None值 → READY兜底(不抛异常).""" + assert AssetStatus(None) == AssetStatus.READY + + +class TestAsset: + """Asset 实体测试.""" + + def test_create_asset_success(self): + """创建素材成功.""" + asset = Asset.create( + project_id="p1", + library_id="lib1", + name="测试视频.mp4", + storage_key="projects/p1/assets/test.mp4", + mime_type="video/mp4", + file_size=1024000, + duration=30.5, + width=1920, + height=1080, + ) + assert asset.id is not None + assert len(asset.id) == 32 + assert asset.project_id == "p1" + assert asset.name == "测试视频.mp4" + assert asset.storage_key == "projects/p1/assets/test.mp4" + assert asset.mime_type == "video/mp4" + assert asset.file_size == 1024000 + assert asset.duration == pytest.approx(30.5) + assert asset.width == 1920 + assert asset.height == 1080 + assert asset.status == AssetStatus.UPLOADING + assert asset.tag_ids == [] + + def test_file_type_video(self): + """video类型从mime_type推导.""" + asset = Asset.create( + project_id="p1", + library_id="lib1", + name="v.mp4", + storage_key="k", + mime_type="video/mp4", + ) + assert asset.file_type == "video" + + def test_file_type_image(self): + """image类型.""" + asset = Asset.create( + project_id="p1", + library_id="lib1", + name="i.jpg", + storage_key="k", + mime_type="image/jpeg", + ) + assert asset.file_type == "image" + + def test_file_type_audio(self): + """audio类型.""" + asset = Asset.create( + project_id="p1", + library_id="lib1", + name="a.mp3", + storage_key="k", + mime_type="audio/mpeg", + ) + assert asset.file_type == "audio" + + def test_file_type_no_slash(self): + """mime_type没有斜杠时返回原值.""" + asset = Asset.create( + project_id="p1", + library_id="lib1", + name="f.txt", + storage_key="k", + mime_type="text", + ) + assert asset.file_type == "text" + + def test_empty_name_raises(self): + """空名称抛异常.""" + with pytest.raises(ValueError, match="素材名称不能为空"): + Asset.create( + project_id="p1", + library_id="lib1", + name="", + storage_key="k", + mime_type="video/mp4", + ) + + def test_empty_storage_key_raises(self): + """空storage_key抛异常.""" + with pytest.raises(ValueError, match="storage_key"): + Asset.create( + project_id="p1", + library_id="lib1", + name="v.mp4", + storage_key=" ", + mime_type="video/mp4", + ) + + def test_empty_mime_type_raises(self): + """空mime_type抛异常.""" + with pytest.raises(ValueError, match="mime_type"): + Asset.create( + project_id="p1", + library_id="lib1", + name="v.mp4", + storage_key="k", + mime_type="", + ) + + def test_name_stripped(self): + """名称首尾空白被去除.""" + asset = Asset.create( + project_id="p1", + library_id="lib1", + name=" 视频.mp4 ", + storage_key="k", + mime_type="video/mp4", + ) + assert asset.name == "视频.mp4" + + def test_add_tag_success(self): + """添加标签成功.""" + asset = Asset.create( + project_id="p1", + library_id="lib1", + name="v.mp4", + storage_key="k", + mime_type="video/mp4", + ) + asset.add_tag("tag_1") + assert "tag_1" in asset.tag_ids + assert len(asset.tag_ids) == 1 + + def test_add_duplicate_tag_deduped(self): + """重复添加标签自动去重.""" + asset = Asset.create( + project_id="p1", + library_id="lib1", + name="v.mp4", + storage_key="k", + mime_type="video/mp4", + ) + asset.add_tag("tag_1") + asset.add_tag("tag_1") + assert asset.tag_ids.count("tag_1") == 1 + + def test_add_empty_tag_raises(self): + """空标签ID抛异常.""" + asset = Asset.create( + project_id="p1", + library_id="lib1", + name="v.mp4", + storage_key="k", + mime_type="video/mp4", + ) + with pytest.raises(ValueError, match="标签 ID 不能为空"): + asset.add_tag(" ") + + def test_remove_tag_success(self): + """删除标签成功.""" + asset = Asset.create( + project_id="p1", + library_id="lib1", + name="v.mp4", + storage_key="k", + mime_type="video/mp4", + ) + asset.add_tag("tag_1") + asset.remove_tag("tag_1") + assert "tag_1" not in asset.tag_ids + + def test_remove_nonexistent_tag_no_error(self): + """删除不存在的标签不报错(幂等).""" + asset = Asset.create( + project_id="p1", + library_id="lib1", + name="v.mp4", + storage_key="k", + mime_type="video/mp4", + ) + # 不抛异常 + asset.remove_tag("nonexistent_tag") + + def test_default_status_uploading(self): + """默认状态UPLOADING.""" + asset = Asset.create( + project_id="p1", + library_id="lib1", + name="v.mp4", + storage_key="k", + mime_type="video/mp4", + ) + assert asset.status == AssetStatus.UPLOADING + + def test_custom_status(self): + """自定义状态.""" + asset = Asset.create( + project_id="p1", + library_id="lib1", + name="v.mp4", + storage_key="k", + mime_type="video/mp4", + status=AssetStatus.READY, + ) + assert asset.status == AssetStatus.READY + + def test_metadata_default_empty_dict(self): + """metadata默认为空dict.""" + asset = Asset.create( + project_id="p1", + library_id="lib1", + name="v.mp4", + storage_key="k", + mime_type="video/mp4", + ) + assert asset.metadata == {} + + def test_metadata_none_becomes_empty(self): + """metadata=None → {}.""" + asset = Asset.create( + project_id="p1", + library_id="lib1", + name="v.mp4", + storage_key="k", + mime_type="video/mp4", + metadata=None, + ) + assert asset.metadata == {} + + +class TestIngestJob: + """IngestJob 实体测试.""" + + def test_create_ingest_job(self): + """创建入库任务.""" + job = IngestJob.create( + project_id="p1", + library_id="lib1", + storage_key="projects/p1/uploads/temp.mp4", + file_hash="abc123", + ) + assert job.id is not None + assert job.project_id == "p1" + assert job.library_id == "lib1" + assert job.storage_key == "projects/p1/uploads/temp.mp4" + assert job.file_hash == "abc123" + + def test_default_status_pending(self): + """默认状态PENDING.""" + from packages.domain.classification import IngestJobStatus + + job = IngestJob.create( + project_id="p1", + library_id="lib1", + storage_key="projects/p1/v.mp4", + ) + assert job.status == IngestJobStatus.PENDING + + def test_empty_project_id_raises(self): + """空project_id抛异常.""" + with pytest.raises(ValueError, match="project_id"): + IngestJob.create( + project_id=" ", + library_id="lib1", + storage_key="k", + ) + + def test_empty_library_id_raises(self): + """空library_id抛异常.""" + with pytest.raises(ValueError, match="library_id"): + IngestJob.create( + project_id="p1", + library_id="", + storage_key="k", + ) + + def test_empty_storage_key_raises(self): + """空storage_key抛异常.""" + with pytest.raises(ValueError, match="storage_key"): + IngestJob.create( + project_id="p1", + library_id="lib1", + storage_key=" ", + ) + + def test_default_result_asset_id_empty(self): + """默认result_asset_id为空.""" + job = IngestJob.create( + project_id="p1", + library_id="lib1", + storage_key="k", + ) + assert job.result_asset_id == "" + + def test_error_message_empty_by_default(self): + """默认error_message为空.""" + job = IngestJob.create( + project_id="p1", + library_id="lib1", + storage_key="k", + ) + assert job.error_message == ""