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)