wqadded new way to get mime type
This commit is contained in:
24
tests/test_mime_types.py
Normal file
24
tests/test_mime_types.py
Normal 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
|
||||
|
||||
Reference in New Issue
Block a user