client/tests: add unit tests for quicktime videos
This commit is contained in:
parent
8a03015349
commit
42524503b9
3 changed files with 11 additions and 2 deletions
|
@ -39,7 +39,7 @@ def get_mime_type(content: bytes) -> str:
|
|||
if content[4:12] in (b"ftypisom", b"ftypiso5", b"ftypmp42", b"ftypM4V "):
|
||||
return "video/mp4"
|
||||
|
||||
if content[4:12] in (b"ftypqt "):
|
||||
if content[4:12] == b"ftypqt ":
|
||||
return "video/quicktime"
|
||||
|
||||
return "application/octet-stream"
|
||||
|
@ -69,7 +69,12 @@ def is_flash(mime_type: str) -> bool:
|
|||
|
||||
|
||||
def is_video(mime_type: str) -> bool:
|
||||
return mime_type.lower() in ("application/ogg", "video/mp4", "video/quicktime", "video/webm")
|
||||
return mime_type.lower() in (
|
||||
"application/ogg",
|
||||
"video/mp4",
|
||||
"video/quicktime",
|
||||
"video/webm",
|
||||
)
|
||||
|
||||
|
||||
def is_image(mime_type: str) -> bool:
|
||||
|
|
BIN
server/szurubooru/tests/assets/mov.mov
Normal file
BIN
server/szurubooru/tests/assets/mov.mov
Normal file
Binary file not shown.
|
@ -7,6 +7,7 @@ from szurubooru.func import mime
|
|||
"input_path,expected_mime_type",
|
||||
[
|
||||
("mp4.mp4", "video/mp4"),
|
||||
("mov.mov", "video/quicktime"),
|
||||
("webm.webm", "video/webm"),
|
||||
("flash.swf", "application/x-shockwave-flash"),
|
||||
("png.png", "image/png"),
|
||||
|
@ -35,6 +36,7 @@ def test_get_mime_type_for_empty_file():
|
|||
[
|
||||
("video/mp4", "mp4"),
|
||||
("video/webm", "webm"),
|
||||
("video/quicktime", "mov"),
|
||||
("application/x-shockwave-flash", "swf"),
|
||||
("image/png", "png"),
|
||||
("image/jpeg", "jpg"),
|
||||
|
@ -70,6 +72,8 @@ def test_is_flash(input_mime_type, expected_state):
|
|||
("VIDEO/WEBM", True),
|
||||
("video/mp4", True),
|
||||
("VIDEO/MP4", True),
|
||||
("video/quicktime", True),
|
||||
("VIDEO/QUICKTIME", True),
|
||||
("video/anything_else", False),
|
||||
("application/ogg", True),
|
||||
("not a video", False),
|
||||
|
|
Loading…
Reference in a new issue