def test_send_to_client_called(self): request = Mock() dummy_queue = Mock() dummy_queue.get = Mock(side_effect = [request, Spawnable.STOP]) self.process.create_queue = Mock(return_value = dummy_queue) server = ServerComms(self.process) server.send_to_client = Mock() server.send_loop() server.send_to_client.assert_called_once_with(request)
def test_send_to_client(self): request = Mock() dummy_queue = Mock() dummy_queue.get = Mock(side_effect = [request, SERVER_STOP]) self.process.create_queue = Mock(return_value = dummy_queue) self.process.spawn = Mock(side_effect = lambda x: x()) server = ServerComms("server", self.process) server.send_to_client = Mock( side_effect = server.send_to_client) server.start_recv_loop = Mock() server.start() server.send_to_client.assert_called_once_with(request)