#46 MIME Type check

* adds unit tests for new methods in konova/tests/test_documents.py
* refactors _accepted_mime_types of AbstractDocument into _valid_mime_types
This commit is contained in:
2021-12-09 13:19:52 +01:00
parent ce747255bc
commit 2a8669ded2
2 changed files with 75 additions and 2 deletions

View File

@@ -63,7 +63,7 @@ class AbstractDocument(BaseResource):
file = models.FileField()
comment = models.TextField()
_accepted_mime_types = {
_valid_mime_types = {
mimetypes.types_map[".pdf"],
mimetypes.types_map[".jpg"],
mimetypes.types_map[".jpeg"],
@@ -95,7 +95,7 @@ class AbstractDocument(BaseResource):
@classmethod
def is_mime_type_valid(cls, _file: str):
mime_type = _file.content_type
return mime_type in cls._accepted_mime_types
return mime_type in cls._valid_mime_types
@classmethod
def is_file_size_valid(cls, _file):