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()