From 2ab4eb3ef4703c6f627fa7d35dcceb17e67d9a8a Mon Sep 17 00:00:00 2001 From: xiaoxia Date: Tue, 28 Jul 2026 13:32:37 +0800 Subject: [PATCH 1/2] =?UTF-8?q?test(wave151):=20quota=E9=85=8D=E9=A2=9D?= =?UTF-8?q?=E7=B3=BB=E7=BB=9F=E5=8D=95=E6=B5=8B=E8=A1=A5=E5=85=A8=20+80?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 为 domain/quota.py 新增 80 个单测,纯逻辑 0 外部依赖: - QuotaDimension 枚举:11个 - QuotaTier 数据类:6个 - QUOTA_TIERS 常量:12个(三档套餐核心字段+单调递增验证) - QuotaWarningLevel:2个 - QuotaCheckResult.usage_percent:7个(正常/0/100%/超量/不限量/零限制) - get_warning_level 告警级别:13个(0%/80%/95%/100%/超量/零限制/不限量/负数) - QuotaRegistry 注册/查询:14个 - QuotaChecker 配额检查:13个 - 全局单例:3个 --- tests/unit/domain/test_quota.py | 575 ++++++++++++++++++++++++++++++++ 1 file changed, 575 insertions(+) create mode 100755 tests/unit/domain/test_quota.py diff --git a/tests/unit/domain/test_quota.py b/tests/unit/domain/test_quota.py new file mode 100755 index 000000000..36d578627 --- /dev/null +++ b/tests/unit/domain/test_quota.py @@ -0,0 +1,575 @@ +"""quota 单测. + +domain 层配额系统纯逻辑模块,0 外部依赖。 +覆盖:枚举常量、QuotaTier、QUOTA_TIERS常量、QuotaWarningLevel、 +QuotaCheckResult、QuotaRegistry注册/查询、QuotaChecker检查/告警级别。 +""" + +from __future__ import annotations + +import math + +from packages.domain.quota import ( + QUOTA_TIERS, + QuotaCheckResult, + QuotaChecker, + QuotaDimension, + QuotaRegistry, + QuotaTier, + QuotaWarningLevel, + get_warning_level, + quota_checker, + quota_registry, +) + + +class TestQuotaDimension: + """QuotaDimension 枚举测试.""" + + def test_member_count(self): + """内置维度数量.""" + assert len(QuotaDimension) == 11 + + def test_storage_gb(self): + """存储维度值.""" + assert QuotaDimension.STORAGE_GB.value == "storage_gb" + + def test_videos_per_month(self): + """每月视频数维度.""" + assert QuotaDimension.VIDEOS_PER_MONTH.value == "videos_per_month" + + def test_max_concurrent(self): + """并发数维度.""" + assert QuotaDimension.MAX_CONCURRENT.value == "max_concurrent" + + def test_max_templates(self): + """模板数维度.""" + assert QuotaDimension.MAX_TEMPLATES.value == "max_templates" + + def test_max_titles(self): + """标题库维度.""" + assert QuotaDimension.MAX_TITLES.value == "max_titles" + + def test_max_voiceovers(self): + """配音库维度.""" + assert QuotaDimension.MAX_VOICEOVERS.value == "max_voiceovers" + + def test_ai_voice_enabled(self): + """AI配音开关维度.""" + assert QuotaDimension.AI_VOICE_ENABLED.value == "ai_voice_enabled" + + def test_ai_voice_credits(self): + """AI配音积分维度.""" + assert QuotaDimension.AI_VOICE_CREDITS.value == "ai_voice_credits" + + def test_all_values_are_strings(self): + """所有枚举值都是字符串.""" + for dim in QuotaDimension: + assert isinstance(dim.value, str) + assert len(dim.value) > 0 + + +class TestQuotaTier: + """QuotaTier 数据类测试.""" + + def test_create_empty(self): + """创建空配额等级.""" + tier = QuotaTier(name="test") + assert tier.name == "test" + assert tier.limits == {} + + def test_create_with_limits(self): + """创建带限制的配额等级.""" + tier = QuotaTier(name="pro", limits={"storage": 100, "videos": 50}) + assert tier.name == "pro" + assert tier.get_limit("storage") == 100 + assert tier.get_limit("videos") == 50 + + def test_get_limit_undefined_returns_zero(self): + """未定义的维度返回 0.""" + tier = QuotaTier(name="test") + assert tier.get_limit("nonexistent") == 0 + + def test_is_unlimited_inf(self): + """inf 视为不限量.""" + tier = QuotaTier(name="test", limits={"templates": float("inf")}) + assert tier.is_unlimited("templates") is True + + def test_is_unlimited_finite(self): + """有限值不是不限量.""" + tier = QuotaTier(name="test", limits={"storage": 100}) + assert tier.is_unlimited("storage") is False + + def test_is_unlimited_undefined(self): + """未定义的维度默认不限量.""" + tier = QuotaTier(name="test") + # 未定义的 key 取默认值 inf,因此 is_unlimited 应该返回 True + assert tier.is_unlimited("unknown") is True + + +class TestQuotaTiers: + """QUOTA_TIERS 常量测试.""" + + def test_three_tiers_exist(self): + """三个套餐等级都存在.""" + assert "free" in QUOTA_TIERS + assert "basic" in QUOTA_TIERS + assert "premium" in QUOTA_TIERS + + def test_free_storage(self): + """免费版 2GB 存储.""" + assert QUOTA_TIERS["free"].get_limit(QuotaDimension.STORAGE_GB) == 2 + + def test_free_videos_per_month(self): + """免费版 5个视频/月.""" + assert QUOTA_TIERS["free"].get_limit(QuotaDimension.VIDEOS_PER_MONTH) == 5 + + def test_free_ai_voice_disabled(self): + """免费版无AI配音.""" + assert QUOTA_TIERS["free"].get_limit(QuotaDimension.AI_VOICE_ENABLED) == 0 + + def test_basic_storage(self): + """基础版 20GB 存储.""" + assert QUOTA_TIERS["basic"].get_limit(QuotaDimension.STORAGE_GB) == 20 + + def test_basic_videos(self): + """基础版 30视频/月.""" + assert QUOTA_TIERS["basic"].get_limit(QuotaDimension.VIDEOS_PER_MONTH) == 30 + + def test_basic_ai_voice_enabled(self): + """基础版有AI配音.""" + assert QUOTA_TIERS["basic"].get_limit(QuotaDimension.AI_VOICE_ENABLED) == 1 + + def test_basic_ai_voice_credits(self): + """基础版 100 AI配音积分.""" + assert QUOTA_TIERS["basic"].get_limit(QuotaDimension.AI_VOICE_CREDITS) == 100 + + def test_premium_storage(self): + """高级版 100GB 存储.""" + assert QUOTA_TIERS["premium"].get_limit(QuotaDimension.STORAGE_GB) == 100 + + def test_premium_videos(self): + """高级版 100视频/月.""" + assert QUOTA_TIERS["premium"].get_limit(QuotaDimension.VIDEOS_PER_MONTH) == 100 + + def test_premium_templates_unlimited(self): + """高级版模板不限量.""" + assert QUOTA_TIERS["premium"].is_unlimited(QuotaDimension.MAX_TEMPLATES) + + def test_premium_multi_platform_enabled(self): + """高级版多平台发布.""" + assert QUOTA_TIERS["premium"].get_limit(QuotaDimension.MULTI_PLATFORM_ENABLED) == 1 + + def test_storage_monotonic(self): + """存储量随套餐升级单调递增.""" + free = QUOTA_TIERS["free"].get_limit(QuotaDimension.STORAGE_GB) + basic = QUOTA_TIERS["basic"].get_limit(QuotaDimension.STORAGE_GB) + premium = QUOTA_TIERS["premium"].get_limit(QuotaDimension.STORAGE_GB) + assert free < basic < premium + + def test_videos_monotonic(self): + """视频数随套餐升级单调递增.""" + free = QUOTA_TIERS["free"].get_limit(QuotaDimension.VIDEOS_PER_MONTH) + basic = QUOTA_TIERS["basic"].get_limit(QuotaDimension.VIDEOS_PER_MONTH) + premium = QUOTA_TIERS["premium"].get_limit(QuotaDimension.VIDEOS_PER_MONTH) + assert free < basic < premium + + +class TestQuotaWarningLevel: + """告警级别常量测试.""" + + def test_levels_defined(self): + """四个级别都有定义.""" + assert QuotaWarningLevel.NORMAL == "normal" + assert QuotaWarningLevel.WARNING == "warning" + assert QuotaWarningLevel.CRITICAL == "critical" + assert QuotaWarningLevel.EXCEEDED == "exceeded" + + def test_four_distinct_levels(self): + """四个级别各不相同.""" + levels = { + QuotaWarningLevel.NORMAL, + QuotaWarningLevel.WARNING, + QuotaWarningLevel.CRITICAL, + QuotaWarningLevel.EXCEEDED, + } + assert len(levels) == 4 + + +class TestQuotaCheckResult: + """QuotaCheckResult 测试.""" + + def test_usage_percent_normal(self): + """正常使用百分比.""" + result = QuotaCheckResult( + allowed=True, + dimension="storage", + limit=100, + used=30, + remaining=70, + warning_level=QuotaWarningLevel.NORMAL, + ) + assert result.usage_percent == 30.0 + + def test_usage_percent_zero_used(self): + """使用量为 0.""" + result = QuotaCheckResult( + allowed=True, + dimension="storage", + limit=100, + used=0, + remaining=100, + warning_level=QuotaWarningLevel.NORMAL, + ) + assert result.usage_percent == 0.0 + + def test_usage_percent_exactly_100(self): + """刚好用完.""" + result = QuotaCheckResult( + allowed=False, + dimension="storage", + limit=100, + used=100, + remaining=0, + warning_level=QuotaWarningLevel.EXCEEDED, + ) + assert result.usage_percent == 100.0 + + def test_usage_percent_over_limit_capped(self): + """超出限制时封顶 100%.""" + result = QuotaCheckResult( + allowed=False, + dimension="storage", + limit=100, + used=150, + remaining=0, + warning_level=QuotaWarningLevel.EXCEEDED, + ) + assert result.usage_percent == 100.0 + + def test_usage_percent_unlimited(self): + """不限量时使用率为 0.""" + result = QuotaCheckResult( + allowed=True, + dimension="templates", + limit=float("inf"), + used=1000, + remaining=float("inf"), + warning_level=QuotaWarningLevel.NORMAL, + ) + assert result.usage_percent == 0.0 + + def test_usage_percent_zero_limit_with_usage(self): + """限制为 0 但有使用量,返回 100%.""" + result = QuotaCheckResult( + allowed=False, + dimension="ai_voice", + limit=0, + used=1, + remaining=0, + warning_level=QuotaWarningLevel.EXCEEDED, + ) + assert result.usage_percent == 100.0 + + def test_usage_percent_zero_limit_no_usage(self): + """限制为 0 且无使用量,返回 0%.""" + result = QuotaCheckResult( + allowed=True, + dimension="ai_voice", + limit=0, + used=0, + remaining=0, + warning_level=QuotaWarningLevel.NORMAL, + ) + assert result.usage_percent == 0.0 + + +class TestGetWarningLevel: + """get_warning_level 便捷函数测试.""" + + def test_zero_usage(self): + """0% 使用 - normal.""" + assert get_warning_level(0, 100) == QuotaWarningLevel.NORMAL + + def test_below_80_percent(self): + """低于80% - normal.""" + assert get_warning_level(50, 100) == QuotaWarningLevel.NORMAL + assert get_warning_level(79, 100) == QuotaWarningLevel.NORMAL + + def test_at_80_percent(self): + """刚好80% - warning.""" + assert get_warning_level(80, 100) == QuotaWarningLevel.WARNING + + def test_between_80_and_95(self): + """80%-95%之间 - warning.""" + assert get_warning_level(90, 100) == QuotaWarningLevel.WARNING + + def test_at_95_percent(self): + """刚好95% - critical.""" + assert get_warning_level(95, 100) == QuotaWarningLevel.CRITICAL + + def test_between_95_and_100(self): + """95%-100%之间 - critical.""" + assert get_warning_level(99, 100) == QuotaWarningLevel.CRITICAL + + def test_at_100_percent(self): + """刚好100% - exceeded.""" + assert get_warning_level(100, 100) == QuotaWarningLevel.EXCEEDED + + def test_over_100_percent(self): + """超过100% - exceeded.""" + assert get_warning_level(150, 100) == QuotaWarningLevel.EXCEEDED + + def test_zero_limit_with_usage(self): + """限制为0但有使用 - exceeded.""" + assert get_warning_level(1, 0) == QuotaWarningLevel.EXCEEDED + + def test_zero_limit_no_usage(self): + """限制为0且无使用 - normal.""" + assert get_warning_level(0, 0) == QuotaWarningLevel.NORMAL + + def test_unlimited(self): + """不限量 - 始终 normal.""" + assert get_warning_level(0, float("inf")) == QuotaWarningLevel.NORMAL + assert get_warning_level(9999, float("inf")) == QuotaWarningLevel.NORMAL + + def test_negative_usage(self): + """负使用量 - normal.""" + assert get_warning_level(-10, 100) == QuotaWarningLevel.NORMAL + + +class TestQuotaRegistry: + """QuotaRegistry 测试.""" + + def test_initial_dimensions(self): + """初始化后内置维度都在.""" + reg = QuotaRegistry() + dims = reg.list_dimensions() + for dim in QuotaDimension: + assert dim.value in dims + + def test_initial_dimensions_count(self): + """初始维度数量等于枚举数量.""" + reg = QuotaRegistry() + assert len(reg.list_dimensions()) == len(QuotaDimension) + + def test_list_tiers(self): + """三个套餐等级.""" + reg = QuotaRegistry() + tiers = reg.list_tiers() + assert "free" in tiers + assert "basic" in tiers + assert "premium" in tiers + assert len(tiers) == 3 + + def test_get_tier_existing(self): + """获取已有的套餐.""" + reg = QuotaRegistry() + tier = reg.get_tier("free") + assert tier is not None + assert tier.name == "free" + + def test_get_tier_nonexistent(self): + """获取不存在的套餐返回 None.""" + reg = QuotaRegistry() + assert reg.get_tier("enterprise") is None + + def test_get_limit_existing(self): + """获取已有限制.""" + reg = QuotaRegistry() + assert reg.get_limit("free", QuotaDimension.STORAGE_GB) == 2 + + def test_get_limit_nonexistent_plan(self): + """不存在的套餐返回 0.""" + reg = QuotaRegistry() + assert reg.get_limit("enterprise", QuotaDimension.STORAGE_GB) == 0 + + def test_get_limit_unknown_dimension(self): + """未知维度返回 0.""" + reg = QuotaRegistry() + assert reg.get_limit("free", "unknown_dim") == 0 + + def test_register_dimension_new(self): + """注册新维度.""" + reg = QuotaRegistry() + count_before = len(reg.list_dimensions()) + reg.register_dimension("custom_dim", "自定义维度") + dims = reg.list_dimensions() + assert "custom_dim" in dims + assert len(dims) == count_before + 1 + + def test_register_dimension_with_default_limits(self): + """注册带默认限制的维度.""" + reg = QuotaRegistry() + reg.register_dimension( + "api_calls", + "API调用次数", + default_limits={"free": 100, "basic": 1000, "premium": 10000}, + ) + assert reg.get_limit("free", "api_calls") == 100 + assert reg.get_limit("basic", "api_calls") == 1000 + assert reg.get_limit("premium", "api_calls") == 10000 + + def test_register_dimension_default_zero(self): + """不带默认限制的维度,各套餐默认为 0.""" + reg = QuotaRegistry() + reg.register_dimension("beta_feature", "测试功能") + assert reg.get_limit("free", "beta_feature") == 0 + assert reg.get_limit("basic", "beta_feature") == 0 + assert reg.get_limit("premium", "beta_feature") == 0 + + def test_register_dimension_idempotent(self): + """重复注册幂等.""" + reg = QuotaRegistry() + reg.register_dimension("dup", "重复测试", default_limits={"free": 10}) + count_before = len(reg.list_dimensions()) + # 第二次注册不同的限制,应该不生效 + reg.register_dimension("dup", "重复测试2", default_limits={"free": 999}) + count_after = len(reg.list_dimensions()) + assert count_before == count_after + assert reg.get_limit("free", "dup") == 10 # 仍然是第一次的值 + + def test_list_dimensions_returns_copy(self): + """list_dimensions 返回副本,修改不影响内部.""" + reg = QuotaRegistry() + dims = reg.list_dimensions() + dims["hacked"] = "hack" + assert "hacked" not in reg.list_dimensions() + + def test_register_partial_default_limits(self): + """只给部分套餐设置默认限制.""" + reg = QuotaRegistry() + reg.register_dimension( + "partial", + "部分套餐", + default_limits={"premium": 100}, + ) + assert reg.get_limit("free", "partial") == 0 + assert reg.get_limit("basic", "partial") == 0 + assert reg.get_limit("premium", "partial") == 100 + + +class TestQuotaChecker: + """QuotaChecker 测试.""" + + def test_check_within_limit(self): + """在限制内,allowed=True.""" + checker = QuotaChecker() + result = checker.check("free", QuotaDimension.STORAGE_GB, 1) + assert result.allowed is True + assert result.dimension == QuotaDimension.STORAGE_GB + assert result.limit == 2 + assert result.used == 1 + assert result.remaining == 1 + assert result.warning_level == QuotaWarningLevel.NORMAL + + def test_check_exceeded(self): + """超出限制,allowed=False.""" + checker = QuotaChecker() + result = checker.check("free", QuotaDimension.STORAGE_GB, 3) + assert result.allowed is False + assert result.remaining == 0 + assert result.warning_level == QuotaWarningLevel.EXCEEDED + + def test_check_exactly_at_limit(self): + """刚好等于限制视为超出(used < limit 才允许).""" + checker = QuotaChecker() + result = checker.check("free", QuotaDimension.STORAGE_GB, 2) + assert result.allowed is False + + def test_check_unlimited(self): + """不限量维度.""" + checker = QuotaChecker() + result = checker.check("premium", QuotaDimension.MAX_TEMPLATES, 999) + assert result.allowed is True + assert math.isinf(result.remaining) + assert result.warning_level == QuotaWarningLevel.NORMAL + + def test_check_warning_level_boundaries(self): + """各告警级别的边界值.""" + checker = QuotaChecker() + # 79% - normal + assert checker.check("free", QuotaDimension.STORAGE_GB, 1.58).warning_level == QuotaWarningLevel.NORMAL + # 80% - warning + assert checker.check("free", QuotaDimension.STORAGE_GB, 1.6).warning_level == QuotaWarningLevel.WARNING + # 95% - critical + assert checker.check("free", QuotaDimension.STORAGE_GB, 1.9).warning_level == QuotaWarningLevel.CRITICAL + # 100% - exceeded + assert checker.check("free", QuotaDimension.STORAGE_GB, 2).warning_level == QuotaWarningLevel.EXCEEDED + + def test_check_zero_usage(self): + """0使用量.""" + checker = QuotaChecker() + result = checker.check("basic", QuotaDimension.VIDEOS_PER_MONTH, 0) + assert result.allowed is True + assert result.remaining == 30 + assert result.usage_percent == 0.0 + + def test_check_unknown_plan(self): + """未知套餐,限制为0.""" + checker = QuotaChecker() + result = checker.check("enterprise", QuotaDimension.STORAGE_GB, 0) + assert result.limit == 0 + # used=0, limit=0 → 0 < 0 is False → allowed=False + assert result.allowed is False + assert result.warning_level == QuotaWarningLevel.NORMAL + + def test_check_unknown_dimension(self): + """未知维度,限制为0.""" + checker = QuotaChecker() + result = checker.check("free", "unknown", 0) + assert result.limit == 0 + + def test_check_multiple(self): + """批量检查多个维度.""" + checker = QuotaChecker() + usage = { + QuotaDimension.STORAGE_GB: 1, + QuotaDimension.VIDEOS_PER_MONTH: 10, + } + results = checker.check_multiple("free", usage) + assert len(results) == 2 + dims = {r.dimension for r in results} + assert QuotaDimension.STORAGE_GB in dims + assert QuotaDimension.VIDEOS_PER_MONTH in dims + + def test_check_multiple_empty(self): + """空字典返回空列表.""" + checker = QuotaChecker() + results = checker.check_multiple("free", {}) + assert results == [] + + def test_remaining_never_negative(self): + """剩余量不为负.""" + checker = QuotaChecker() + result = checker.check("free", QuotaDimension.STORAGE_GB, 100) + assert result.remaining >= 0 + + def test_custom_registry(self): + """使用自定义 registry.""" + reg = QuotaRegistry() + reg.register_dimension("custom", "自定义", default_limits={"free": 42}) + checker = QuotaChecker(reg) + result = checker.check("free", "custom", 10) + assert result.limit == 42 + assert result.allowed is True + + +class TestGlobalSingletons: + """全局单例测试.""" + + def test_quota_registry_exists(self): + """全局 registry 单例存在.""" + assert quota_registry is not None + assert isinstance(quota_registry, QuotaRegistry) + + def test_quota_checker_exists(self): + """全局 checker 单例存在.""" + assert quota_checker is not None + assert isinstance(quota_checker, QuotaChecker) + + def test_global_checker_works(self): + """全局 checker 能正常工作.""" + result = quota_checker.check("free", QuotaDimension.STORAGE_GB, 1) + assert result.allowed is True + assert result.limit == 2 -- 2.54.0 From ec92a864db3b6b2a86d4efab63c33610794eea85 Mon Sep 17 00:00:00 2001 From: xiaoxia Date: Tue, 28 Jul 2026 13:35:04 +0800 Subject: [PATCH 2/2] =?UTF-8?q?test(wave152):=20duplication=E6=9F=A5?= =?UTF-8?q?=E9=87=8D=E8=AE=B0=E5=BD=95=E5=8D=95=E6=B5=8B=20+40?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 为 domain/duplication.py 新增 40 个单测,纯逻辑 0 外部依赖: - DuplicateSegment.create 工厂/校验:10个 - DuplicationRecord.create 工厂/校验:11个 - 状态流转 (pending/processing/completed/failed):13个 - can_retry + reset_for_retry:4个 - segments 列表:3个 --- tests/unit/domain/test_duplication.py | 376 ++++++++++++++++++++++++++ 1 file changed, 376 insertions(+) create mode 100644 tests/unit/domain/test_duplication.py diff --git a/tests/unit/domain/test_duplication.py b/tests/unit/domain/test_duplication.py new file mode 100644 index 000000000..884bcbd0b --- /dev/null +++ b/tests/unit/domain/test_duplication.py @@ -0,0 +1,376 @@ +"""duplication 单测. + +domain 层查重记录纯逻辑模块,0 外部依赖。 +覆盖:DuplicateSegment 工厂/校验、DuplicationRecord 创建/状态流转/重试。 +""" + +from __future__ import annotations + +from packages.domain.duplication import DuplicateSegment, DuplicationRecord + + +class TestDuplicateSegmentCreate: + """DuplicateSegment.create 工厂方法测试.""" + + def test_create_valid(self): + """正常创建.""" + seg = DuplicateSegment.create( + source_start=1.0, + source_end=5.0, + matched_video_id="vid123", + matched_video_name="测试视频", + matched_start=10.0, + matched_end=14.0, + similarity=85.5, + ) + assert seg.source_start == 1.0 + assert seg.source_end == 5.0 + assert seg.matched_video_id == "vid123" + assert seg.matched_video_name == "测试视频" + assert seg.matched_start == 10.0 + assert seg.matched_end == 14.0 + assert seg.similarity == 85.5 + assert isinstance(seg.id, str) + assert len(seg.id) > 0 + + def test_create_generates_unique_id(self): + """每次创建生成不同的 id.""" + seg1 = DuplicateSegment.create(0, 1, "v", "n", 0, 1, 50.0) + seg2 = DuplicateSegment.create(0, 1, "v", "n", 0, 1, 50.0) + assert seg1.id != seg2.id + + def test_create_negative_source_start(self): + """source_start 为负抛出 ValueError.""" + try: + DuplicateSegment.create(-1, 5, "v", "n", 0, 1, 50.0) + assert False, "应该抛出 ValueError" + except ValueError as e: + assert "source" in str(e).lower() + + def test_create_source_end_equals_start(self): + """source_end 等于 source_start 无效.""" + try: + DuplicateSegment.create(5, 5, "v", "n", 0, 1, 50.0) + assert False + except ValueError as e: + assert "source" in str(e).lower() + + def test_create_source_end_less_than_start(self): + """source_end 小于 source_start 无效.""" + try: + DuplicateSegment.create(5, 3, "v", "n", 0, 1, 50.0) + assert False + except ValueError as e: + assert "source" in str(e).lower() + + def test_create_negative_matched_start(self): + """matched_start 为负无效.""" + try: + DuplicateSegment.create(0, 5, "v", "n", -1, 1, 50.0) + assert False + except ValueError as e: + assert "matched" in str(e).lower() + + def test_create_matched_end_invalid(self): + """matched_end <= matched_start 无效.""" + try: + DuplicateSegment.create(0, 5, "v", "n", 5, 5, 50.0) + assert False + except ValueError as e: + assert "matched" in str(e).lower() + + def test_create_similarity_zero(self): + """similarity = 0 是合法的.""" + seg = DuplicateSegment.create(0, 1, "v", "n", 0, 1, 0.0) + assert seg.similarity == 0.0 + + def test_create_similarity_100(self): + """similarity = 100 是合法的.""" + seg = DuplicateSegment.create(0, 1, "v", "n", 0, 1, 100.0) + assert seg.similarity == 100.0 + + def test_create_similarity_negative(self): + """similarity < 0 无效.""" + try: + DuplicateSegment.create(0, 1, "v", "n", 0, 1, -1.0) + assert False + except ValueError as e: + assert "similarity" in str(e).lower() + + def test_create_similarity_over_100(self): + """similarity > 100 无效.""" + try: + DuplicateSegment.create(0, 1, "v", "n", 0, 1, 101.0) + assert False + except ValueError as e: + assert "similarity" in str(e).lower() + + +class TestDuplicationRecordCreate: + """DuplicationRecord.create 工厂方法测试.""" + + def test_create_minimal(self): + """最简创建.""" + rec = DuplicationRecord.create( + user_id="user1", + filename="test.mp4", + file_size=1024, + storage_key="oss://bucket/test.mp4", + ) + assert rec.user_id == "user1" + assert rec.filename == "test.mp4" + assert rec.file_size == 1024 + assert rec.storage_key == "oss://bucket/test.mp4" + assert rec.duration_seconds == 0.0 + assert rec.status == "pending" + assert rec.duplicate_rate is None + assert rec.duplicate_count == 0 + assert rec.segments == [] + assert rec.error_message == "" + assert isinstance(rec.id, str) + assert len(rec.id) > 0 + + def test_create_with_duration(self): + """带时长创建.""" + rec = DuplicationRecord.create( + user_id="user1", + filename="test.mp4", + file_size=1024, + storage_key="oss://key", + duration_seconds=120.5, + ) + assert rec.duration_seconds == 120.5 + + def test_create_strips_whitespace(self): + """user_id 和 filename 会 strip.""" + rec = DuplicationRecord.create( + user_id=" user1 ", + filename=" test.mp4 ", + file_size=1024, + storage_key="oss://key", + ) + assert rec.user_id == "user1" + assert rec.filename == "test.mp4" + + def test_create_empty_user_id(self): + """空 user_id 无效.""" + try: + DuplicationRecord.create("", "test.mp4", 1024, "oss://key") + assert False + except ValueError as e: + assert "user_id" in str(e) + + def test_create_whitespace_user_id(self): + """纯空白 user_id 无效.""" + try: + DuplicationRecord.create(" ", "test.mp4", 1024, "oss://key") + assert False + except ValueError as e: + assert "user_id" in str(e) + + def test_create_empty_filename(self): + """空 filename 无效.""" + try: + DuplicationRecord.create("user1", "", 1024, "oss://key") + assert False + except ValueError as e: + assert "filename" in str(e) + + def test_create_whitespace_filename(self): + """纯空白 filename 无效.""" + try: + DuplicationRecord.create("user1", " ", 1024, "oss://key") + assert False + except ValueError as e: + assert "filename" in str(e) + + def test_create_zero_file_size(self): + """file_size = 0 无效.""" + try: + DuplicationRecord.create("user1", "test.mp4", 0, "oss://key") + assert False + except ValueError as e: + assert "file_size" in str(e) + + def test_create_negative_file_size(self): + """file_size 为负无效.""" + try: + DuplicationRecord.create("user1", "test.mp4", -1, "oss://key") + assert False + except ValueError as e: + assert "file_size" in str(e) + + def test_create_unique_id(self): + """不同记录 id 不同.""" + r1 = DuplicationRecord.create("u", "f", 1, "k") + r2 = DuplicationRecord.create("u", "f", 1, "k") + assert r1.id != r2.id + + def test_create_has_timestamps(self): + """有创建和更新时间.""" + rec = DuplicationRecord.create("u", "f", 1, "k") + assert rec.created_at is not None + assert rec.updated_at is not None + # 两者应该很接近(都是 now) + delta = (rec.updated_at - rec.created_at).total_seconds() + assert abs(delta) < 1.0 + + +class TestDuplicationRecordStatusFlow: + """状态流转测试.""" + + def _make_record(self): + return DuplicationRecord.create("user1", "test.mp4", 1024, "oss://key") + + def test_initial_status_pending(self): + """初始状态 pending.""" + rec = self._make_record() + assert rec.status == "pending" + + def test_mark_processing(self): + """标记为处理中.""" + rec = self._make_record() + old_updated = rec.updated_at + rec.mark_processing() + assert rec.status == "processing" + assert rec.updated_at >= old_updated + + def test_mark_completed(self): + """标记为完成.""" + rec = self._make_record() + seg = DuplicateSegment.create(0, 1, "v", "n", 0, 1, 80.0) + rec.mark_completed(duplicate_rate=45.5, duplicate_count=3, segments=[seg]) + assert rec.status == "completed" + assert rec.duplicate_rate == 45.5 + assert rec.duplicate_count == 3 + assert len(rec.segments) == 1 + assert rec.segments[0].similarity == 80.0 + + def test_mark_completed_zero_rate(self): + """重复率为 0 合法.""" + rec = self._make_record() + rec.mark_completed(0.0, 0, []) + assert rec.status == "completed" + assert rec.duplicate_rate == 0.0 + assert rec.duplicate_count == 0 + assert rec.segments == [] + + def test_mark_completed_full_rate(self): + """重复率 100 合法.""" + rec = self._make_record() + rec.mark_completed(100.0, 1, []) + assert rec.duplicate_rate == 100.0 + + def test_mark_completed_negative_rate(self): + """重复率为负无效.""" + rec = self._make_record() + try: + rec.mark_completed(-1, 0, []) + assert False + except ValueError as e: + assert "duplicate_rate" in str(e) + + def test_mark_completed_over_100(self): + """重复率超过 100 无效.""" + rec = self._make_record() + try: + rec.mark_completed(101, 0, []) + assert False + except ValueError as e: + assert "duplicate_rate" in str(e) + + def test_mark_failed(self): + """标记为失败.""" + rec = self._make_record() + rec.mark_failed("网络超时") + assert rec.status == "failed" + assert rec.error_message == "网络超时" + + def test_mark_failed_empty_message(self): + """失败信息可以为空字符串.""" + rec = self._make_record() + rec.mark_failed("") + assert rec.status == "failed" + assert rec.error_message == "" + + def test_can_retry_failed(self): + """failed 状态可以重试.""" + rec = self._make_record() + rec.mark_failed("error") + assert rec.can_retry() is True + + def test_cannot_retry_pending(self): + """pending 状态不可重试.""" + rec = self._make_record() + assert rec.can_retry() is False + + def test_cannot_retry_processing(self): + """processing 状态不可重试.""" + rec = self._make_record() + rec.mark_processing() + assert rec.can_retry() is False + + def test_cannot_retry_completed(self): + """completed 状态不可重试.""" + rec = self._make_record() + rec.mark_completed(50, 1, []) + assert rec.can_retry() is False + + def test_reset_for_retry(self): + """重置回 pending.""" + rec = self._make_record() + rec.mark_failed("error") + seg = DuplicateSegment.create(0, 1, "v", "n", 0, 1, 50.0) + rec.segments = [seg] + rec.video_fingerprint = {"hash": "abc"} + rec.duplicate_rate = 50.0 + rec.duplicate_count = 5 + + rec.reset_for_retry() + assert rec.status == "pending" + assert rec.duplicate_rate is None + assert rec.duplicate_count == 0 + assert rec.error_message == "" + assert rec.segments == [] + assert rec.video_fingerprint is None + + def test_reset_updates_timestamp(self): + """重置更新 updated_at.""" + rec = self._make_record() + rec.mark_failed("error") + old_updated = rec.updated_at + rec.reset_for_retry() + assert rec.updated_at >= old_updated + + +class TestDuplicationRecordSegments: + """segments 列表相关测试.""" + + def _make_record(self): + return DuplicationRecord.create("user1", "test.mp4", 1024, "oss://key") + + def test_segments_default_empty(self): + """初始 segments 为空列表.""" + rec = self._make_record() + assert rec.segments == [] + + def test_segments_independent_list(self): + """不同记录的 segments 是独立列表.""" + r1 = self._make_record() + r2 = self._make_record() + r1.segments.append("fake") + assert len(r2.segments) == 0 + + def test_completed_with_multiple_segments(self): + """完成时带多个片段.""" + rec = self._make_record() + segs = [ + DuplicateSegment.create(0, 1, "v1", "n1", 0, 1, 90.0), + DuplicateSegment.create(2, 3, "v2", "n2", 5, 6, 70.0), + DuplicateSegment.create(4, 5, "v3", "n3", 10, 11, 85.0), + ] + rec.mark_completed(60.0, 3, segs) + assert len(rec.segments) == 3 + assert rec.segments[0].similarity == 90.0 + assert rec.segments[1].matched_video_id == "v2" + assert rec.segments[2].matched_video_name == "n3" -- 2.54.0