def test_marshal_with_error(self): error = 'error message' expected = { 'error': 'error message', 'value': None } response = CommandResponse(error=error) result = response.marshal() self.assertEquals(result, expected)
def test_marshal_with_value(self): value = { 'agent_number': '1000', 'extension': '2000', 'context': 'default' } expected = { 'error': None, 'value': { 'agent_number': '1000', 'extension': '2000', 'context': 'default' } } response = CommandResponse(value=value) result = response.marshal() self.assertEquals(result, expected)