def test_unmarshal_message(self):
json = '{"error": null, "value": "foobar"}'
expected = {'value': 'foobar', 'error': None}
marshal = Marshaler({})
result = marshal.unmarshal_message(json)
self.assertEquals(result, expected)
def test_unmarshal_response(self):
json = '{"error": null, "value": "foobar"}'
marshal = Marshaler({})
result = marshal.unmarshal_response(json)
self.assertTrue(isinstance(result, CommandResponse))
self.assertEquals(result.value, 'foobar')
self.assertEquals(result.error, None)
def test_unmarshal_command(self):
json = '{"name": "foobar", "data": {"a":1}}'
command = Mock()
registry = {'foobar': command}
marshal = Marshaler(registry)
marshal.unmarshal_command(json)
command.unmarshal.assert_called_once_with({'a': 1})
def test_marshal_response(self):
response = Mock()
response.marshal.return_value = {'value': 'success', 'error': None}
marshal = Marshaler({})
result = marshal.marshal_response(response)
response.marshal.assert_called_once_with()
self.assertEquals(result, '{"value": "success", "error": null}')
def test_marshal_command(self):
command = Mock()
command.name = 'foobar'
command.marshal.return_value = {'a': 1}
marshal = Marshaler({})
result = marshal.marshal_command(command)
command.marshal.assert_called_once_with()
self.assertEquals(result, ('{"data": {"a": 1}, "name": "foobar"}'))
def __init__(self):
self._transport = None
self._marshaler = Marshaler()
def run(self):
self._marshaler = Marshaler(self._commands_registry)
self._transport.run()
def __init__(self):
self._marshaler = Marshaler()
