wqadded new way to get mime type

This commit is contained in:
2023-12-21 13:36:13 +01:00
parent f8383567fe
commit e6c59507c4
5 changed files with 53 additions and 2 deletions

24
tests/test_mime_types.py Normal file
View File

@@ -0,0 +1,24 @@
import unittest
from src.mime_types import MimeTypes
class TestMimeTypes(unittest.TestCase):
def test_mime_type_image(self):
mime_type = MimeTypes(file_type="image")
assert mime_type.image
assert not mime_type.video
assert not mime_type.unsupported_file_type
def test_mime_type_video(self):
mime_type = MimeTypes(file_type="video")
assert not mime_type.image
assert mime_type.video
assert not mime_type.unsupported_file_type
def test_mime_type_unsupported_file_type(self):
mime_type = MimeTypes(file_type="not_a_valid_file_type")
assert not mime_type.image
assert not mime_type.video
assert mime_type.unsupported_file_type