class TestRequest(unittest.TestCase):
def setUp(self):
self.callback = MagicMock()
self.o = Request(32, self.callback)
def test_init(self):
assert self.o.id == 32
assert self.o.callback == self.callback
def test_respond_with_return(self):
cb, response = self.o.return_response(value=5)
assert cb == self.callback
assert response == Return(id=32, value=5)
def test_respond_with_error(self):
cb, response = self.o.error_response(
exception=ValueError("Test Error"))
assert cb == self.callback
assert response == Error(id=32, message="ValueError: Test Error")
def test_setters(self):
self.o.set_id(123)
assert 123 == self.o.id
self.o.set_callback(None)
self.o.callback(888)
self.callback.assert_not_called()
class TestRequest(unittest.TestCase):
def setUp(self):
self.context = MagicMock()
self.response_queue = MagicMock()
self.request = Request(self.context, self.response_queue)
def test_init(self):
self.assertEqual(self.context, self.request.context)
self.assertEqual(self.response_queue, self.request.response_queue)
def test_repr(self):
r = Request(MagicMock(), MagicMock())
s = r.__repr__()
self.assertTrue(isinstance(s, str))
self.assertIn('id', s)
def test_respond_with_return(self):
self.request.respond_with_return(value=5)
call_arg = self.response_queue.put.call_args_list[0][0][0].to_dict()
expected_response = Return(self.request.id,
self.request.context,
value=5).to_dict()
self.assertEqual(call_arg, expected_response)
def test_respond_with_error(self):
self.request.respond_with_error(message="Test Error")
call_arg = self.response_queue.put.call_args_list[0][0][0].to_dict()
expected_response = Error(self.request.id,
self.request.context,
message="Test Error").to_dict()
self.assertEqual(call_arg, expected_response)
def test_setters(self):
self.request.set_id(123)
self.assertEquals(123, self.request.id)
class TestRequest(unittest.TestCase):
def setUp(self):
self.context = MagicMock()
self.response_queue = MagicMock()
self.response_queue.qsize.return_value = 0
self.request = Request(self.context, self.response_queue)
def test_init(self):
self.assertEqual(self.context, self.request.context)
self.assertEqual(self.response_queue, self.request.response_queue)
def test_repr(self):
r = Request(MagicMock(), MagicMock())
s = r.__repr__()
self.assertTrue(isinstance(s, str))
self.assertIn('id', s)
def test_respond_with_return(self):
self.request.respond_with_return(value=5)
call_arg = self.response_queue.put.call_args_list[0][0][0].to_dict()
expected_response = Return(self.request.id, self.request.context, value=5).to_dict()
self.assertEqual(call_arg, expected_response)
def test_respond_with_error(self):
self.request.respond_with_error(message="Test Error")
call_arg = self.response_queue.put.call_args_list[0][0][0].to_dict()
expected_response = Error(self.request.id, self.request.context,
message="Test Error").to_dict()
self.assertEqual(call_arg, expected_response)
def test_setters(self):
self.request.set_id(123)
self.assertEquals(123, self.request.id)
