Simple-Incremental-Search-Tool/test/test_MimeGuesser.py
2018-04-21 22:43:38 -04:00

27 lines
1.2 KiB
Python

from parsing import ContentMimeGuesser, ExtensionMimeGuesser
from unittest import TestCase
import os
dir_name = os.path.dirname(os.path.abspath(__file__))
class MimeGuesserTest(TestCase):
def test_content_guesser(self):
guesser = ContentMimeGuesser()
self.assertEqual("text/x-shellscript", guesser.guess_mime(dir_name + "/test_folder/test_utf8.sh"))
self.assertEqual("text/plain", guesser.guess_mime(dir_name + "/test_folder/more_books.json"))
self.assertEqual("application/java-archive", guesser.guess_mime(dir_name + "/test_folder/post.jar"))
self.assertEqual("image/jpeg", guesser.guess_mime(dir_name + "/test_folder/sample_1.jpg"))
def test_extension_guesser(self):
guesser = ExtensionMimeGuesser()
self.assertEqual("text/x-sh", guesser.guess_mime(dir_name + "/test_folder/test_utf8.sh"))
self.assertEqual("application/json", guesser.guess_mime(dir_name + "/test_folder/more_books.json"))
self.assertEqual("application/java-archive", guesser.guess_mime(dir_name + "/test_folder/post.jar"))
self.assertEqual("image/jpeg", guesser.guess_mime(dir_name + "/test_folder/sample_1.jpg"))