Example #1
0
 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()
Example #2
0
 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>")
Example #4
0
 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)
Example #5
0
 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_)
Example #6
0
 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>")
Example #7
0
 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)