def test_get_codec_by_invalid_module(self):
"""Codec: Should raise an error when module doesn't provide a codec class.
"""
operation = lambda: codec.get_codec('djangoflash.models')
self.assertRaises(AttributeError, operation)
def test_get_codec_by_invalid_module_name(self):
"""Codec: Should raise an error when resolving a module name that doesn't exists.
"""
operation = lambda: codec.get_codec('invalid.module.path')
self.assertRaises(ImportError, operation)
def test_get_json_zlib_codec_by_alias(self):
"""Codec: 'json_zlib' should resolve to JSON/zlib-based codec.
"""
codec_impl = codec.get_codec('json_zlib')
self.assertTrue(isinstance(codec_impl, json_impl.CodecClass))
self.assertTrue(isinstance(codec_impl, json_zlib_impl.CodecClass))
def test_get_codec_by_module_name(self):
"""Codec: 'djangoflash.codec.json_impl' should resolve to JSON-based codec.
"""
codec_impl = codec.get_codec('djangoflash.codec.json_impl')
self.assertTrue(isinstance(codec_impl, json_impl.CodecClass))
def test_get_pickle_codec_by_alias(self):
"""Codec: 'pickle' should resolve to Pickle-based codec.
"""
codec_impl = codec.get_codec('pickle')
self.assertTrue(isinstance(codec_impl, pickle_impl.CodecClass))
