def test_json_codec(self):
mime_codec = MimeCodec()
mime = 'application/json'
value = dict(key='value')
data = '{"key": "value"}'
self.assertEqual(data, mime_codec.marshal(mime, value))
self.assertEqual(value, mime_codec.unmarshal(mime, data))
def app():
_codec = MimeCodec()
_codec.register('application/json', JsonCodec())
# TODO: how to load .yaml?
app = App.load('http://petstore.swagger.io/v2/swagger.json',
mime_codec=_codec)
app.prepare(strict=True)
return app
def test_register_unregister(self):
mime_codec = MimeCodec()
mime = 'test'
dummy_codec = {}
self.assertEqual(None, mime_codec.codec(mime))
mime_codec.register(mime, dummy_codec)
self.assertEqual(dummy_codec, mime_codec.codec(mime))
mime_codec.unregister(mime)
self.assertEqual(None, mime_codec.codec(mime))
def test_plain_codec(self):
mime_codec = MimeCodec()
mime = 'text/plain'
text = 'plain text'
self.assertEqual(text, mime_codec.marshal(mime, text))
self.assertEqual(text, mime_codec.unmarshal(mime, text))