"""media_validation 领域模块单元测试。""" import pytest from packages.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 TestSafeParseFpsBasic: def test_integer_fps(self): assert safe_parse_fps("30") == 30.0 def test_decimal_fps(self): assert safe_parse_fps("29.97") == pytest.approx(29.97) def test_fraction_simple(self): assert safe_parse_fps("30/1") == 30.0 def test_fraction_ntsc(self): assert safe_parse_fps("30000/1001") == pytest.approx(29.97002997) def test_fraction_pal(self): assert safe_parse_fps("25/1") == 25.0 def test_fraction_24fps_cine(self): assert safe_parse_fps("24000/1001") == pytest.approx(23.976023976) def test_zero_fps(self): assert safe_parse_fps("0") == 0.0 def test_zero_fraction(self): assert safe_parse_fps("0/1") == 0.0 class TestSafeParseFpsEdgeCases: def test_zero_denominator(self): assert safe_parse_fps("30/0") == 0.0 def test_empty_string(self): assert safe_parse_fps("") == 0.0 def test_garbage_string(self): assert safe_parse_fps("not_a_number") == 0.0 def test_multiple_slashes(self): # split("/", 1) 只切第一个，后面的作为 den 的一部分会解析失败 assert safe_parse_fps("30/1/2") == 0.0 def test_negative_fps(self): assert safe_parse_fps("-30") == -30.0 def test_negative_fraction(self): assert safe_parse_fps("-30/1") == -30.0 def test_very_high_fps(self): assert safe_parse_fps("240/1") == 240.0 def test_fraction_float_num(self): assert safe_parse_fps("29.97/1") == pytest.approx(29.97) def test_fraction_float_den(self): assert safe_parse_fps("30/1.001") == pytest.approx(29.97002997) def test_whitespace_in_string(self): # float(" 30 ") 能解析，所以应该返回 30.0 assert safe_parse_fps(" 30 ") == 30.0 class TestMinFileSizeConstants: def test_min_video_size_is_1kb(self): assert MIN_VIDEO_FILE_SIZE == 1024 def test_min_audio_size(self): assert MIN_AUDIO_FILE_SIZE == 100 def test_min_image_size(self): assert MIN_IMAGE_FILE_SIZE == 100 class TestSupportedVideoCodecs: def test_h264_family_present(self): assert "h264" in SUPPORTED_VIDEO_CODECS assert "avc1" in SUPPORTED_VIDEO_CODECS assert "avc" in SUPPORTED_VIDEO_CODECS def test_h265_family_present(self): assert "hevc" in SUPPORTED_VIDEO_CODECS assert "h265" in SUPPORTED_VIDEO_CODECS assert "hev1" in SUPPORTED_VIDEO_CODECS assert "hvc1" in SUPPORTED_VIDEO_CODECS def test_vp9_av1_present(self): assert "vp9" in SUPPORTED_VIDEO_CODECS assert "vp09" in SUPPORTED_VIDEO_CODECS assert "av1" in SUPPORTED_VIDEO_CODECS assert "av01" in SUPPORTED_VIDEO_CODECS def test_vp8_present(self): assert "vp8" in SUPPORTED_VIDEO_CODECS assert "vp08" in SUPPORTED_VIDEO_CODECS def test_mpeg_family_present(self): assert "mpeg4" in SUPPORTED_VIDEO_CODECS assert "mp4v" in SUPPORTED_VIDEO_CODECS assert "mpeg2video" in SUPPORTED_VIDEO_CODECS def test_prores_family_present(self): assert "prores" in SUPPORTED_VIDEO_CODECS assert "apcn" in SUPPORTED_VIDEO_CODECS assert "apch" in SUPPORTED_VIDEO_CODECS def test_unknown_codec_not_present(self): assert "unknown_codec_xyz" not in SUPPORTED_VIDEO_CODECS def test_codecs_count_reasonable(self): # 白名单应该有足够多的编码格式 assert len(SUPPORTED_VIDEO_CODECS) >= 30 class TestIsValidMediaVideo: def test_valid_video(self): metadata = {"size_bytes": 5000, "duration": 10.0, "codec": "h264"} assert is_valid_media(metadata, "video") is True def test_video_too_small(self): metadata = {"size_bytes": 500, "duration": 10.0, "codec": "h264"} assert is_valid_media(metadata, "video") is False def test_video_exact_min_size(self): metadata = {"size_bytes": 1024, "duration": 10.0, "codec": "h264"} assert is_valid_media(metadata, "video") is True def test_video_zero_duration(self): metadata = {"size_bytes": 5000, "duration": 0, "codec": "h264"} assert is_valid_media(metadata, "video") is False def test_video_negative_duration(self): metadata = {"size_bytes": 5000, "duration": -1.0, "codec": "h264"} assert is_valid_media(metadata, "video") is False def test_video_missing_size_default_zero(self): metadata = {"duration": 10.0, "codec": "h264"} assert is_valid_media(metadata, "video") is False def test_video_missing_duration_default_zero(self): metadata = {"size_bytes": 5000, "codec": "h264"} assert is_valid_media(metadata, "video") is False def test_video_unknown_codec_still_valid(self): # 非白名单编码仍允许通过（渲染层统一转码） metadata = {"size_bytes": 5000, "duration": 10.0, "codec": "some_unknown_codec"} assert is_valid_media(metadata, "video") is True def test_video_missing_codec_still_valid(self): metadata = {"size_bytes": 5000, "duration": 10.0} assert is_valid_media(metadata, "video") is True def test_video_codec_case_insensitive(self): metadata = {"size_bytes": 5000, "duration": 10.0, "codec": "H264"} assert is_valid_media(metadata, "video") is True def test_video_empty_codec(self): metadata = {"size_bytes": 5000, "duration": 10.0, "codec": ""} assert is_valid_media(metadata, "video") is True def test_video_hevc_codec(self): metadata = {"size_bytes": 5000, "duration": 10.0, "codec": "hevc"} assert is_valid_media(metadata, "video") is True def test_video_vp9_codec(self): metadata = {"size_bytes": 5000, "duration": 10.0, "codec": "vp9"} assert is_valid_media(metadata, "video") is True def test_video_av1_codec(self): metadata = {"size_bytes": 5000, "duration": 10.0, "codec": "av1"} assert is_valid_media(metadata, "video") is True def test_video_prores_codec(self): metadata = {"size_bytes": 5000, "duration": 10.0, "codec": "prores"} assert is_valid_media(metadata, "video") is True def test_video_empty_metadata(self): assert is_valid_media({}, "video") is False class TestIsValidMediaAudio: def test_valid_audio(self): metadata = {"size_bytes": 5000, "duration": 30.0} assert is_valid_media(metadata, "audio") is True def test_audio_too_small(self): metadata = {"size_bytes": 50, "duration": 30.0} assert is_valid_media(metadata, "audio") is False def test_audio_exact_min_size(self): metadata = {"size_bytes": 100, "duration": 10.0} assert is_valid_media(metadata, "audio") is True def test_audio_zero_duration(self): metadata = {"size_bytes": 5000, "duration": 0} assert is_valid_media(metadata, "audio") is False def test_audio_negative_duration(self): metadata = {"size_bytes": 5000, "duration": -1.0} assert is_valid_media(metadata, "audio") is False def test_audio_missing_size(self): metadata = {"duration": 10.0} assert is_valid_media(metadata, "audio") is False def test_audio_missing_duration(self): metadata = {"size_bytes": 5000} assert is_valid_media(metadata, "audio") is False def test_audio_with_codec_info(self): metadata = {"size_bytes": 5000, "duration": 30.0, "codec": "aac"} assert is_valid_media(metadata, "audio") is True def test_audio_empty_metadata(self): assert is_valid_media({}, "audio") is False class TestIsValidMediaImage: def test_valid_image(self): metadata = {"size_bytes": 5000, "width": 1920, "height": 1080} assert is_valid_media(metadata, "image") is True def test_image_too_small(self): metadata = {"size_bytes": 50, "width": 1920, "height": 1080} assert is_valid_media(metadata, "image") is False def test_image_exact_min_size(self): metadata = {"size_bytes": 100, "width": 100, "height": 100} assert is_valid_media(metadata, "image") is True def test_image_zero_width(self): metadata = {"size_bytes": 5000, "width": 0, "height": 1080} assert is_valid_media(metadata, "image") is False def test_image_zero_height(self): metadata = {"size_bytes": 5000, "width": 1920, "height": 0} assert is_valid_media(metadata, "image") is False def test_image_negative_dimensions(self): metadata = {"size_bytes": 5000, "width": -1, "height": 1080} assert is_valid_media(metadata, "image") is False def test_image_missing_width(self): metadata = {"size_bytes": 5000, "height": 1080} assert is_valid_media(metadata, "image") is False def test_image_missing_height(self): metadata = {"size_bytes": 5000, "width": 1920} assert is_valid_media(metadata, "image") is False def test_image_missing_size(self): metadata = {"width": 1920, "height": 1080} assert is_valid_media(metadata, "image") is False def test_image_small_but_valid(self): metadata = {"size_bytes": 100, "width": 1, "height": 1} assert is_valid_media(metadata, "image") is True def test_image_empty_metadata(self): assert is_valid_media({}, "image") is False class TestIsValidMediaUnknownType: def test_unknown_type_returns_false(self): metadata = {"size_bytes": 5000, "duration": 10.0} assert is_valid_media(metadata, "unknown") is False def test_empty_type_returns_false(self): metadata = {"size_bytes": 5000, "duration": 10.0} assert is_valid_media(metadata, "") is False def test_text_type_returns_false(self): metadata = {"size_bytes": 5000} assert is_valid_media(metadata, "text") is False class TestIsValidMediaSizeTypes: def test_size_as_string(self): # int("5000") 能解析 metadata = {"size_bytes": "5000", "duration": 10.0, "codec": "h264"} assert is_valid_media(metadata, "video") is True def test_size_as_none(self): # int(None) 会 TypeError，但 metadata.get 返回 0 默认值 metadata = {"size_bytes": None, "duration": 10.0, "codec": "h264"} # int(None) 会抛 TypeError with pytest.raises(TypeError): is_valid_media(metadata, "video")