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)
