def test_sends_to_server(self): client = ClientComms("c", Mock()) client.send_to_server = Mock() request = Mock() client.q.get = Mock(side_effect = [request, client.STOP]) client.log_exception = Mock() client.send_loop() client.send_to_server.assert_called_once_with(request) client.log_exception.assert_not_called()
def test_requests_are_stored(self): client = ClientComms("c", Mock()) client._current_id = 1234 request = Mock() client.send_to_server = Mock() client.q.get = Mock(side_effect = [request, CLIENT_STOP]) client.send_loop() expected = OrderedDict({1234 : request}) self.assertEquals(expected, client.requests)
def test_send_logs_error(self): client = ClientComms(Mock()) client.send_to_server = Mock(side_effect=Exception) request = Mock() request.to_dict = Mock(return_value="<to_dict>") client.q.get = Mock(side_effect=[request, client.STOP]) client.log_exception = Mock() client.send_loop() client.log_exception.assert_called_once_with("Exception sending request %s", "<to_dict>")
def test_request_id_provided(self): client = ClientComms("c", Mock()) client._current_id = 1234 client.send_to_server = Mock() request_1 = Mock(id_ = None) request_2 = Mock(id_ = None) client.q.get = Mock(side_effect = [request_1, request_2, client.STOP]) client.send_loop() request_1.set_id.assert_called_once_with(1234) request_2.set_id.assert_called_once_with(1235)
def test_request_id_provided(self): client = ClientComms("c", Mock()) client._current_id = 1234 client.send_to_server = Mock() request_1 = Mock(id_ = None) request_2 = Mock(id_ = None) client.q.get = Mock(side_effect = [request_1, request_2, CLIENT_STOP]) client.send_loop() self.assertEqual(1234, request_1.id_) self.assertEqual(1235, request_2.id_)
def test_send_logs_error(self): client = ClientComms("c", Mock()) client.send_to_server = Mock(side_effect=Exception) request = Mock() request.to_dict = Mock(return_value = "<to_dict>") client.q.get = Mock(side_effect = [request, CLIENT_STOP]) client.log_exception = Mock() client.send_loop() client.log_exception.assert_called_once_with( "Exception sending request %s", "<to_dict>")
def test_request_id_provided(self): client = ClientComms("c", Mock()) client._current_id = 1234 client.send_to_server = Mock() request_1 = Mock(id_=None) request_2 = Mock(id_=None) client.q.get = Mock(side_effect=[request_1, request_2, client.STOP]) client.send_loop() request_1.set_id.assert_called_once_with(1234) request_2.set_id.assert_called_once_with(1235)
def test_requests_are_stored(self): client = ClientComms(Mock()) client._current_id = 1234 request = Mock() def f(id_): request.id = id_ request.set_id.side_effect = f client.send_to_server = Mock() client.q.get = Mock(side_effect=[request, client.STOP]) client.send_loop() expected = OrderedDict({1234: request}) self.assertEquals(expected, client.requests)