style: auto-format with black + isort + prettier [skip ci-format-check]

tests/unit/domain/test_media_validation.py

@@ -0,0 +1,290 @@
+"""media_validation 媒体文件校验单测."""
+
+import pytest
+from domain.media_validation import (
+    MIN_AUDIO_FILE_SIZE,
+    MIN_IMAGE_FILE_SIZE,
+    MIN_VIDEO_FILE_SIZE,
+    SUPPORTED_VIDEO_CODECS,
+    is_valid_media,
+    safe_parse_fps,
+)
+
+# ── 常量测试 ─────────────────────────────────────────────────────────────────
+
+
+class TestConstants:
+    """模块常量"""
+
+    def test_min_sizes(self):
+        assert MIN_VIDEO_FILE_SIZE == 1024
+        assert MIN_AUDIO_FILE_SIZE == 100
+        assert MIN_IMAGE_FILE_SIZE == 100
+
+    def test_supported_codecs_is_frozenset(self):
+        assert isinstance(SUPPORTED_VIDEO_CODECS, frozenset)
+
+    def test_supported_codecs_includes_common(self):
+        assert "h264" in SUPPORTED_VIDEO_CODECS
+        assert "hevc" in SUPPORTED_VIDEO_CODECS
+        assert "vp9" in SUPPORTED_VIDEO_CODECS
+        assert "av1" in SUPPORTED_VIDEO_CODECS
+        assert "mpeg4" in SUPPORTED_VIDEO_CODECS
+        assert "prores" in SUPPORTED_VIDEO_CODECS
+
+    def test_supported_codecs_count(self):
+        assert len(SUPPORTED_VIDEO_CODECS) >= 20
+
+
+# ── safe_parse_fps ───────────────────────────────────────────────────────────
+
+
+class TestSafeParseFps:
+    """safe_parse_fps 函数"""
+
+    def test_simple_decimal(self):
+        assert safe_parse_fps("30.0") == 30.0
+
+    def test_integer_string(self):
+        assert safe_parse_fps("24") == 24.0
+
+    def test_fraction_format(self):
+        assert abs(safe_parse_fps("30000/1001") - 29.97) < 0.01
+
+    def test_simple_fraction(self):
+        assert safe_parse_fps("30/1") == 30.0
+
+    def test_24fps_fraction(self):
+        assert safe_parse_fps("24/1") == 24.0
+
+    def test_60fps_fraction(self):
+        assert safe_parse_fps("60000/1001") == pytest.approx(59.94, abs=0.01)
+
+    def test_zero_denominator_returns_zero(self):
+        assert safe_parse_fps("30/0") == 0.0
+
+    def test_empty_string_returns_zero(self):
+        assert safe_parse_fps("") == 0.0
+
+    def test_invalid_string_returns_zero(self):
+        assert safe_parse_fps("invalid") == 0.0
+
+    def test_none_numerator_fraction(self):
+        assert safe_parse_fps("abc/1001") == 0.0
+
+    def test_negative_fps(self):
+        assert safe_parse_fps("-30") == -30.0
+
+    def test_very_high_fps(self):
+        assert safe_parse_fps("240/1") == 240.0
+
+    def test_multiple_slashes(self):
+        # 只按第一个 / 分割
+        # "30/1/2" → num="30", den="1/2" → float("1/2") 抛异常 → 返回 0
+        assert safe_parse_fps("30/1/2") == 0.0
+
+    def test_float_fraction(self):
+        result = safe_parse_fps("29.97/1")
+        assert result == pytest.approx(29.97)
+
+    def test_zero_fps(self):
+        assert safe_parse_fps("0") == 0.0
+
+    def test_zero_numerator(self):
+        assert safe_parse_fps("0/1000") == 0.0
+
+
+# ── is_valid_media - video ───────────────────────────────────────────────────
+
+
+class TestIsValidMediaVideo:
+    """is_valid_media 视频校验"""
+
+    def test_valid_video(self):
+        metadata = {
+            "size_bytes": 1024 * 1024,  # 1MB
+            "duration": 10.0,
+            "codec": "h264",
+            "width": 1920,
+            "height": 1080,
+        }
+        assert is_valid_media(metadata, "video") is True
+
+    def test_small_file_invalid(self):
+        metadata = {"size_bytes": 100, "duration": 10.0}
+        assert is_valid_media(metadata, "video") is False
+
+    def test_exact_min_size_valid(self):
+        metadata = {"size_bytes": MIN_VIDEO_FILE_SIZE, "duration": 1.0}
+        assert is_valid_media(metadata, "video") is True
+
+    def test_zero_duration_invalid(self):
+        metadata = {"size_bytes": 1024 * 1024, "duration": 0}
+        assert is_valid_media(metadata, "video") is False
+
+    def test_negative_duration_invalid(self):
+        metadata = {"size_bytes": 1024 * 1024, "duration": -1.0}
+        assert is_valid_media(metadata, "video") is False
+
+    def test_unsupported_codec_still_valid(self):
+        # 非白名单编码仍允许通过（不做严格拦截）
+        metadata = {
+            "size_bytes": 1024 * 1024,
+            "duration": 10.0,
+            "codec": "unknown_codec_xyz",
+        }
+        assert is_valid_media(metadata, "video") is True
+
+    def test_empty_codec_valid(self):
+        metadata = {"size_bytes": 1024 * 1024, "duration": 10.0, "codec": ""}
+        assert is_valid_media(metadata, "video") is True
+
+    def test_no_codec_valid(self):
+        metadata = {"size_bytes": 1024 * 1024, "duration": 10.0}
+        assert is_valid_media(metadata, "video") is True
+
+    def test_hevc_codec_valid(self):
+        metadata = {
+            "size_bytes": 1024 * 1024,
+            "duration": 10.0,
+            "codec": "hevc",
+        }
+        assert is_valid_media(metadata, "video") is True
+
+    def test_codec_case_insensitive(self):
+        metadata = {
+            "size_bytes": 1024 * 1024,
+            "duration": 10.0,
+            "codec": "H264",
+        }
+        assert is_valid_media(metadata, "video") is True
+
+    def test_missing_size_invalid(self):
+        metadata = {"duration": 10.0}
+        assert is_valid_media(metadata, "video") is False
+
+    def test_missing_duration_invalid(self):
+        metadata = {"size_bytes": 1024 * 1024}
+        assert is_valid_media(metadata, "video") is False
+
+    def test_empty_metadata_invalid(self):
+        assert is_valid_media({}, "video") is False
+
+
+# ── is_valid_media - audio ───────────────────────────────────────────────────
+
+
+class TestIsValidMediaAudio:
+    """is_valid_media 音频校验"""
+
+    def test_valid_audio(self):
+        metadata = {"size_bytes": 1024, "duration": 30.0, "codec": "aac"}
+        assert is_valid_media(metadata, "audio") is True
+
+    def test_small_audio_invalid(self):
+        metadata = {"size_bytes": 50, "duration": 30.0}
+        assert is_valid_media(metadata, "audio") is False
+
+    def test_exact_min_size_valid(self):
+        metadata = {"size_bytes": MIN_AUDIO_FILE_SIZE, "duration": 1.0}
+        assert is_valid_media(metadata, "audio") is True
+
+    def test_zero_duration_invalid(self):
+        metadata = {"size_bytes": 1024, "duration": 0}
+        assert is_valid_media(metadata, "audio") is False
+
+    def test_negative_duration_invalid(self):
+        metadata = {"size_bytes": 1024, "duration": -5.0}
+        assert is_valid_media(metadata, "audio") is False
+
+    def test_empty_metadata_invalid(self):
+        assert is_valid_media({}, "audio") is False
+
+    def test_very_short_audio_valid(self):
+        metadata = {"size_bytes": 200, "duration": 0.5}
+        assert is_valid_media(metadata, "audio") is True
+
+
+# ── is_valid_media - image ───────────────────────────────────────────────────
+
+
+class TestIsValidMediaImage:
+    """is_valid_media 图片校验"""
+
+    def test_valid_image(self):
+        metadata = {"size_bytes": 1024, "width": 1920, "height": 1080}
+        assert is_valid_media(metadata, "image") is True
+
+    def test_small_image_invalid(self):
+        metadata = {"size_bytes": 50, "width": 1920, "height": 1080}
+        assert is_valid_media(metadata, "image") is False
+
+    def test_exact_min_size_valid(self):
+        metadata = {
+            "size_bytes": MIN_IMAGE_FILE_SIZE,
+            "width": 100,
+            "height": 100,
+        }
+        assert is_valid_media(metadata, "image") is True
+
+    def test_zero_width_invalid(self):
+        metadata = {"size_bytes": 1024, "width": 0, "height": 1080}
+        assert is_valid_media(metadata, "image") is False
+
+    def test_zero_height_invalid(self):
+        metadata = {"size_bytes": 1024, "width": 1920, "height": 0}
+        assert is_valid_media(metadata, "image") is False
+
+    def test_negative_width_invalid(self):
+        metadata = {"size_bytes": 1024, "width": -1, "height": 1080}
+        assert is_valid_media(metadata, "image") is False
+
+    def test_small_image_valid(self):
+        metadata = {"size_bytes": 200, "width": 10, "height": 10}
+        assert is_valid_media(metadata, "image") is True
+
+    def test_missing_width_invalid(self):
+        metadata = {"size_bytes": 1024, "height": 1080}
+        assert is_valid_media(metadata, "image") is False
+
+    def test_missing_height_invalid(self):
+        metadata = {"size_bytes": 1024, "width": 1920}
+        assert is_valid_media(metadata, "image") is False
+
+    def test_empty_metadata_invalid(self):
+        assert is_valid_media({}, "image") is False
+
+
+# ── is_valid_media - edge cases ─────────────────────────────────────────────
+
+
+class TestIsValidMediaEdgeCases:
+    """is_valid_media 边界情况"""
+
+    def test_invalid_media_type(self):
+        metadata = {"size_bytes": 1024, "duration": 10.0}
+        assert is_valid_media(metadata, "document") is False
+
+    def test_empty_media_type(self):
+        metadata = {"size_bytes": 1024}
+        assert is_valid_media(metadata, "") is False
+
+    def test_string_size_converted(self):
+        metadata = {"size_bytes": "2048", "duration": "5.0"}
+        assert is_valid_media(metadata, "video") is True
+
+    def test_video_size_as_string(self):
+        metadata = {"size_bytes": "1000000", "duration": "30"}
+        assert is_valid_media(metadata, "video") is True
+
+    def test_invalid_size_string_raises(self):
+        # int("abc") 会抛 ValueError
+        metadata = {"size_bytes": "abc", "duration": 10.0}
+        with pytest.raises(ValueError):
+            is_valid_media(metadata, "video")
+
+    def test_none_size_raises(self):
+        # int(None) 会抛 TypeError
+        metadata = {"size_bytes": None, "duration": 10.0}
+        with pytest.raises(TypeError):
+            is_valid_media(metadata, "video")