xiaoxia-saas/tests/unit/test_entities_domain.py

"""Entities 领域层单元测试 - entities.py"""

import pytest

from packages.domain.classification import (
    AssetLibraryKind,
    ClassificationStatus,
    IngestJobStatus,
)
from packages.domain.entities import (
    Asset,
    AssetLibrary,
    AssetStatus,
    IngestJob,
    Project,
    User,
)


class TestAssetStatus:
    """AssetStatus 枚举 + _missing_ 兼容测试"""

    def test_normal_values(self):
        """正常枚举值"""
        assert AssetStatus.UPLOADING == "uploading"
        assert AssetStatus.READY == "ready"
        assert AssetStatus.PROCESSING == "processing"
        assert AssetStatus.ERROR == "error"
        assert AssetStatus.DELETED == "deleted"

    def test_missing_uploaded_maps_to_ready(self):
        """历史值 uploaded → READY"""
        assert AssetStatus("uploaded") == AssetStatus.READY

    def test_missing_success_maps_to_ready(self):
        assert AssetStatus("success") == AssetStatus.READY

    def test_missing_done_maps_to_ready(self):
        assert AssetStatus("done") == AssetStatus.READY

    def test_missing_upload_maps_to_uploading(self):
        assert AssetStatus("upload") == AssetStatus.UPLOADING

    def test_missing_failed_maps_to_error(self):
        assert AssetStatus("failed") == AssetStatus.ERROR

    def test_missing_process_maps_to_processing(self):
        assert AssetStatus("process") == AssetStatus.PROCESSING

    def test_missing_running_maps_to_processing(self):
        assert AssetStatus("running") == AssetStatus.PROCESSING

    def test_missing_unknown_defaults_to_ready(self):
        """完全未知的值兜底为 READY（不阻塞业务）"""
        assert AssetStatus("completely_unknown") == AssetStatus.READY

    def test_missing_case_insensitive(self):
        """大小写不敏感"""
        assert AssetStatus("UPLOADED") == AssetStatus.READY
        assert AssetStatus("Success") == AssetStatus.READY

    def test_missing_with_spaces(self):
        """带空格也能处理"""
        assert AssetStatus("  uploaded  ") == AssetStatus.READY

    def test_missing_non_string_defaults_to_ready(self):
        """非字符串输入兜底为 READY"""
        assert AssetStatus(None) == AssetStatus.READY
        assert AssetStatus(123) == AssetStatus.READY


class TestUser:
    """User 实体测试"""

    def test_create_minimal(self):
        """最小化创建"""
        user = User(
            id="user-1",
            email="test@example.com",
            display_name="Test User",
        )
        assert user.id == "user-1"
        assert user.email == "test@example.com"
        assert user.display_name == "Test User"
        assert user.username == ""
        assert user.email_verified is False
        assert user.subscription_plan == "free"
        assert user.is_admin is False
        assert user.created_at is not None

    def test_create_with_all_fields(self):
        """全字段创建"""
        user = User(
            id="user-1",
            email="test@example.com",
            display_name="Test",
            username="testuser",
            password_hash="hash123",
            email_verified=True,
            subscription_plan="pro",
            is_admin=True,
            wechat_openid="wx_openid",
            phone="13800138000",
            phone_verified=True,
        )
        assert user.username == "testuser"
        assert user.subscription_plan == "pro"
        assert user.is_admin is True
        assert user.wechat_openid == "wx_openid"
        assert user.phone == "13800138000"
        assert user.phone_verified is True

    def test_default_subscription_is_free(self):
        user = User(id="u1", email="a@b.com", display_name="A")
        assert user.subscription_plan == "free"

    def test_default_max_projects(self):
        user = User(id="u1", email="a@b.com", display_name="A")
        assert user.max_projects == 3

    def test_default_storage(self):
        user = User(id="u1", email="a@b.com", display_name="A")
        assert user.max_storage_gb == 10
        assert user.used_storage_gb == 0.0


class TestProject:
    """Project 实体测试"""

    def test_create_minimal(self):
        project = Project.create(
            owner_user_id="user-1",
            name="My Project",
        )
        assert project.id
        assert len(project.id) == 32
        assert project.owner_user_id == "user-1"
        assert project.name == "My Project"
        assert project.description == ""
        assert project.shared_users == []
        assert project.created_at is not None

    def test_create_with_description(self):
        project = Project.create(
            owner_user_id="user-1",
            name="Test",
            description="A test project",
        )
        assert project.description == "A test project"

    def test_create_empty_name_raises(self):
        with pytest.raises(ValueError, match="项目名称不能为空"):
            Project.create(owner_user_id="user-1", name="")

    def test_create_whitespace_name_raises(self):
        with pytest.raises(ValueError, match="项目名称不能为空"):
            Project.create(owner_user_id="user-1", name="   ")

    def test_create_name_stripped(self):
        project = Project.create(owner_user_id="user-1", name="  My Project  ")
        assert project.name == "My Project"

    def test_create_description_stripped(self):
        project = Project.create(
            owner_user_id="user-1",
            name="Test",
            description="  desc  ",
        )
        assert project.description == "desc"

    def test_is_owner_true(self):
        project = Project.create(owner_user_id="user-1", name="Test")
        assert project.is_owner("user-1") is True

    def test_is_owner_false(self):
        project = Project.create(owner_user_id="user-1", name="Test")
        assert project.is_owner("user-2") is False

    def test_is_shared_with_true(self):
        project = Project.create(owner_user_id="user-1", name="Test")
        project.shared_users = ["user-2", "user-3"]
        assert project.is_shared_with("user-2") is True

    def test_is_shared_with_false(self):
        project = Project.create(owner_user_id="user-1", name="Test")
        assert project.is_shared_with("user-2") is False

    def test_can_access_owner(self):
        project = Project.create(owner_user_id="user-1", name="Test")
        assert project.can_access("user-1") is True

    def test_can_access_shared_user(self):
        project = Project.create(owner_user_id="user-1", name="Test")
        project.shared_users = ["user-2"]
        assert project.can_access("user-2") is True

    def test_can_access_stranger(self):
        project = Project.create(owner_user_id="user-1", name="Test")
        assert project.can_access("user-3") is False


class TestAssetLibrary:
    """AssetLibrary 实体测试"""

    def test_create_video_library(self):
        library = AssetLibrary.create(
            project_id="proj-1",
            name="视频素材",
            kind=AssetLibraryKind.VIDEO,
        )
        assert library.id
        assert len(library.id) == 32
        assert library.project_id == "proj-1"
        assert library.name == "视频素材"
        assert library.kind == AssetLibraryKind.VIDEO
        assert library.asset_count == 0
        assert library.total_size == 0
        assert library.created_at is not None

    def test_create_audio_library(self):
        library = AssetLibrary.create(
            project_id="proj-1",
            name="音频素材",
            kind=AssetLibraryKind.VOICE,
        )
        assert library.kind == AssetLibraryKind.VOICE

    def test_create_image_library(self):
        library = AssetLibrary.create(
            project_id="proj-1",
            name="图片素材",
            kind=AssetLibraryKind.IMAGE,
        )
        assert library.kind == AssetLibraryKind.IMAGE

    def test_create_empty_name_raises(self):
        with pytest.raises(ValueError, match="素材库名称不能为空"):
            AssetLibrary.create(
                project_id="proj-1",
                name="",
                kind=AssetLibraryKind.VIDEO,
            )

    def test_create_whitespace_name_raises(self):
        with pytest.raises(ValueError, match="素材库名称不能为空"):
            AssetLibrary.create(
                project_id="proj-1",
                name="   ",
                kind=AssetLibraryKind.VIDEO,
            )

    def test_create_name_stripped(self):
        library = AssetLibrary.create(
            project_id="proj-1",
            name="  视频库  ",
            kind=AssetLibraryKind.VIDEO,
        )
        assert library.name == "视频库"


class TestAsset:
    """Asset 实体测试"""

    def test_create_minimal(self):
        asset = Asset.create(
            project_id="proj-1",
            library_id="lib-1",
            name="test.mp4",
            storage_key="assets/test.mp4",
            mime_type="video/mp4",
        )
        assert asset.id
        assert len(asset.id) == 32
        assert asset.project_id == "proj-1"
        assert asset.library_id == "lib-1"
        assert asset.name == "test.mp4"
        assert asset.storage_key == "assets/test.mp4"
        assert asset.mime_type == "video/mp4"
        assert asset.status == AssetStatus.UPLOADING
        assert asset.classification_status == ClassificationStatus.PENDING
        assert asset.tag_ids == []
        assert asset.metadata == {}

    def test_create_empty_name_raises(self):
        with pytest.raises(ValueError, match="素材名称不能为空"):
            Asset.create(
                project_id="p1",
                library_id="l1",
                name="",
                storage_key="k",
                mime_type="video/mp4",
            )

    def test_create_empty_storage_key_raises(self):
        with pytest.raises(ValueError, match="storage_key 不能为空"):
            Asset.create(
                project_id="p1",
                library_id="l1",
                name="test",
                storage_key="  ",
                mime_type="video/mp4",
            )

    def test_create_empty_mime_type_raises(self):
        with pytest.raises(ValueError, match="mime_type 不能为空"):
            Asset.create(
                project_id="p1",
                library_id="l1",
                name="test",
                storage_key="k",
                mime_type="  ",
            )

    def test_file_type_video(self):
        asset = Asset.create(
            project_id="p1",
            library_id="l1",
            name="test.mp4",
            storage_key="k",
            mime_type="video/mp4",
        )
        assert asset.file_type == "video"

    def test_file_type_audio(self):
        asset = Asset.create(
            project_id="p1",
            library_id="l1",
            name="test.mp3",
            storage_key="k",
            mime_type="audio/mpeg",
        )
        assert asset.file_type == "audio"

    def test_file_type_image(self):
        asset = Asset.create(
            project_id="p1",
            library_id="l1",
            name="test.jpg",
            storage_key="k",
            mime_type="image/jpeg",
        )
        assert asset.file_type == "image"

    def test_file_type_no_slash(self):
        asset = Asset.create(
            project_id="p1",
            library_id="l1",
            name="test",
            storage_key="k",
            mime_type="unknown",
        )
        assert asset.file_type == "unknown"

    def test_add_tag(self):
        asset = Asset.create(
            project_id="p1",
            library_id="l1",
            name="test.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_tag_deduplicates(self):
        """重复添加标签自动去重"""
        asset = Asset.create(
            project_id="p1",
            library_id="l1",
            name="test.mp4",
            storage_key="k",
            mime_type="video/mp4",
        )
        asset.add_tag("tag-1")
        asset.add_tag("tag-1")
        assert len(asset.tag_ids) == 1

    def test_add_empty_tag_raises(self):
        asset = Asset.create(
            project_id="p1",
            library_id="l1",
            name="test.mp4",
            storage_key="k",
            mime_type="video/mp4",
        )
        with pytest.raises(ValueError, match="标签 ID 不能为空"):
            asset.add_tag("")

    def test_add_tag_strips(self):
        asset = Asset.create(
            project_id="p1",
            library_id="l1",
            name="test.mp4",
            storage_key="k",
            mime_type="video/mp4",
        )
        asset.add_tag("  tag-1  ")
        assert asset.tag_ids == ["tag-1"]

    def test_remove_tag(self):
        asset = Asset.create(
            project_id="p1",
            library_id="l1",
            name="test.mp4",
            storage_key="k",
            mime_type="video/mp4",
        )
        asset.add_tag("tag-1")
        asset.add_tag("tag-2")
        asset.remove_tag("tag-1")
        assert "tag-1" not in asset.tag_ids
        assert "tag-2" in asset.tag_ids
        assert len(asset.tag_ids) == 1

    def test_remove_nonexistent_tag_no_error(self):
        """删除不存在的标签不报错（幂等）"""
        asset = Asset.create(
            project_id="p1",
            library_id="l1",
            name="test.mp4",
            storage_key="k",
            mime_type="video/mp4",
        )
        asset.remove_tag("nonexistent")  # 不抛异常
        assert len(asset.tag_ids) == 0

    def test_add_tag_updates_updated_at(self):
        asset = Asset.create(
            project_id="p1",
            library_id="l1",
            name="test.mp4",
            storage_key="k",
            mime_type="video/mp4",
        )
        old_updated = asset.updated_at
        import time

        time.sleep(0.001)
        asset.add_tag("tag-1")
        assert asset.updated_at >= old_updated

    def test_remove_tag_updates_updated_at(self):
        asset = Asset.create(
            project_id="p1",
            library_id="l1",
            name="test.mp4",
            storage_key="k",
            mime_type="video/mp4",
        )
        asset.add_tag("tag-1")
        old_updated = asset.updated_at
        import time

        time.sleep(0.001)
        asset.remove_tag("tag-1")
        assert asset.updated_at >= old_updated

    def test_create_with_file_size(self):
        asset = Asset.create(
            project_id="p1",
            library_id="l1",
            name="test.mp4",
            storage_key="k",
            mime_type="video/mp4",
            file_size=1024,
        )
        assert asset.file_size == 1024

    def test_create_with_video_properties(self):
        asset = Asset.create(
            project_id="p1",
            library_id="l1",
            name="test.mp4",
            storage_key="k",
            mime_type="video/mp4",
            duration=60.5,
            width=1920,
            height=1080,
            fps=30.0,
            codec="h264",
        )
        assert asset.duration == 60.5
        assert asset.width == 1920
        assert asset.height == 1080
        assert asset.fps == 30.0
        assert asset.codec == "h264"

    def test_create_with_metadata(self):
        asset = Asset.create(
            project_id="p1",
            library_id="l1",
            name="test.mp4",
            storage_key="k",
            mime_type="video/mp4",
            metadata={"source": "upload"},
        )
        assert asset.metadata == {"source": "upload"}

    def test_create_none_metadata_defaults_to_empty_dict(self):
        asset = Asset.create(
            project_id="p1",
            library_id="l1",
            name="test.mp4",
            storage_key="k",
            mime_type="video/mp4",
            metadata=None,
        )
        assert asset.metadata == {}

    def test_name_stripped(self):
        asset = Asset.create(
            project_id="p1",
            library_id="l1",
            name="  test.mp4  ",
            storage_key="k",
            mime_type="video/mp4",
        )
        assert asset.name == "test.mp4"


class TestIngestJob:
    """IngestJob 实体测试"""

    def test_create_minimal(self):
        job = IngestJob.create(
            project_id="proj-1",
            library_id="lib-1",
            storage_key="assets/test.mp4",
        )
        assert job.id
        assert len(job.id) == 32
        assert job.project_id == "proj-1"
        assert job.library_id == "lib-1"
        assert job.storage_key == "assets/test.mp4"
        assert job.status == IngestJobStatus.PENDING
        assert job.error_message == ""
        assert job.result_asset_id == ""
        assert job.created_at is not None

    def test_create_with_file_hash(self):
        job = IngestJob.create(
            project_id="proj-1",
            library_id="lib-1",
            storage_key="assets/test.mp4",
            file_hash="abc123",
        )
        assert job.file_hash == "abc123"

    def test_create_empty_project_id_raises(self):
        with pytest.raises(ValueError, match="project_id 不能为空"):
            IngestJob.create(
                project_id="",
                library_id="lib-1",
                storage_key="k",
            )

    def test_create_empty_library_id_raises(self):
        with pytest.raises(ValueError, match="library_id 不能为空"):
            IngestJob.create(
                project_id="p1",
                library_id="",
                storage_key="k",
            )

    def test_create_empty_storage_key_raises(self):
        with pytest.raises(ValueError, match="storage_key 不能为空"):
            IngestJob.create(
                project_id="p1",
                library_id="l1",
                storage_key="",
            )

    def test_create_whitespace_project_id_raises(self):
        with pytest.raises(ValueError, match="project_id 不能为空"):
            IngestJob.create(
                project_id="   ",
                library_id="lib-1",
                storage_key="k",
            )

    def test_create_strips_fields(self):
        job = IngestJob.create(
            project_id="  proj-1  ",
            library_id="  lib-1  ",
            storage_key="  assets/test.mp4  ",
            file_hash="  hash123  ",
        )
        assert job.project_id == "proj-1"
        assert job.library_id == "lib-1"
        assert job.storage_key == "assets/test.mp4"
        assert job.file_hash == "hash123"