def test_detect_content_type(self): m = Message() # No extension error self.assertRaises(MessageError, m._detect_content_type, 'xxx') # Blacklisted extension error self.assertRaises(MessageError, m._detect_content_type, 'xxx.bin') # Unknown extension returns default content type ext = 'xcacaswcawc' self.assertEqual(m._detect_content_type('xxx.' + ext), m._default_content_type) # Known extension returns correct mimetype self.assertEqual(m._detect_content_type('xxx.png'), 'image/png')