import mimetypes
|
import StringIO
|
import unittest
|
import sys
|
|
from test import test_support
|
|
# Tell it we don't know about external files:
|
mimetypes.knownfiles = []
|
mimetypes.inited = False
|
mimetypes._default_mime_types()
|
|
|
class MimeTypesTestCase(unittest.TestCase):
|
def setUp(self):
|
self.db = mimetypes.MimeTypes()
|
|
def test_default_data(self):
|
eq = self.assertEqual
|
eq(self.db.guess_type("foo.html"), ("text/html", None))
|
eq(self.db.guess_type("foo.tgz"), ("application/x-tar", "gzip"))
|
eq(self.db.guess_type("foo.tar.gz"), ("application/x-tar", "gzip"))
|
eq(self.db.guess_type("foo.tar.Z"), ("application/x-tar", "compress"))
|
eq(self.db.guess_type("foo.tar.bz2"), ("application/x-tar", "bzip2"))
|
eq(self.db.guess_type("foo.tar.xz"), ("application/x-tar", "xz"))
|
|
def test_data_urls(self):
|
eq = self.assertEqual
|
guess_type = self.db.guess_type
|
eq(guess_type("data:,thisIsTextPlain"), ("text/plain", None))
|
eq(guess_type("data:;base64,thisIsTextPlain"), ("text/plain", None))
|
eq(guess_type("data:text/x-foo,thisIsTextXFoo"), ("text/x-foo", None))
|
|
def test_file_parsing(self):
|
eq = self.assertEqual
|
sio = StringIO.StringIO("x-application/x-unittest pyunit\n")
|
self.db.readfp(sio)
|
eq(self.db.guess_type("foo.pyunit"),
|
("x-application/x-unittest", None))
|
eq(self.db.guess_extension("x-application/x-unittest"), ".pyunit")
|
|
def test_non_standard_types(self):
|
eq = self.assertEqual
|
# First try strict
|
eq(self.db.guess_type('foo.xul', strict=True), (None, None))
|
eq(self.db.guess_extension('image/jpg', strict=True), None)
|
# And then non-strict
|
eq(self.db.guess_type('foo.xul', strict=False), ('text/xul', None))
|
eq(self.db.guess_extension('image/jpg', strict=False), '.jpg')
|
|
def test_guess_all_types(self):
|
eq = self.assertEqual
|
unless = self.assertTrue
|
# First try strict. Use a set here for testing the results because if
|
# test_urllib2 is run before test_mimetypes, global state is modified
|
# such that the 'all' set will have more items in it.
|
all = set(self.db.guess_all_extensions('text/plain', strict=True))
|
unless(all >= set(['.bat', '.c', '.h', '.ksh', '.pl', '.txt']))
|
# And now non-strict
|
all = self.db.guess_all_extensions('image/jpg', strict=False)
|
all.sort()
|
eq(all, ['.jpg'])
|
# And now for no hits
|
all = self.db.guess_all_extensions('image/jpg', strict=True)
|
eq(all, [])
|
|
|
@unittest.skipUnless(sys.platform.startswith("win"), "Windows only")
|
class Win32MimeTypesTestCase(unittest.TestCase):
|
def setUp(self):
|
# ensure all entries actually come from the Windows registry
|
self.original_types_map = mimetypes.types_map.copy()
|
mimetypes.types_map.clear()
|
mimetypes.init()
|
self.db = mimetypes.MimeTypes()
|
|
def tearDown(self):
|
# restore default settings
|
mimetypes.types_map.clear()
|
mimetypes.types_map.update(self.original_types_map)
|
|
def test_registry_parsing(self):
|
# the original, minimum contents of the MIME database in the
|
# Windows registry is undocumented AFAIK.
|
# Use file types that should *always* exist:
|
eq = self.assertEqual
|
eq(self.db.guess_type("foo.txt"), ("text/plain", None))
|
|
def test_main():
|
test_support.run_unittest(MimeTypesTestCase,
|
Win32MimeTypesTestCase
|
)
|
|
|
if __name__ == "__main__":
|
test_main()
|
