def test_sends_to_server(self): client = ClientComms(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_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_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_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_loop_starts(self): process = Mock(spawn = lambda x: x()) client = ClientComms("c", process) client.send_loop = Mock() client.start_recv_loop = Mock() client.log_exception = Mock() client.start() client.send_loop.assert_called_once_with() client.start_recv_loop.assert_called_once_with() client.log_exception.assert_not_called()
def test_start_stop(self): sync_factory = SyncFactory("s") process = Mock() process.spawn = sync_factory.spawn process.create_queue = sync_factory.create_queue client = ClientComms("c", process) client.send_loop = Mock(side_effect = client.send_loop) client.start_recv_loop = Mock() client.stop_recv_loop = Mock() client.log_exception = Mock() client.start() self.assertFalse(client._send_spawned.ready()) client.start_recv_loop.assert_called_once_with() client.stop(0.1) self.assertTrue(client._send_spawned.ready()) client.send_loop.assert_called_once_with() client.stop_recv_loop.assert_called_once_with() client.log_exception.assert_not_called()