def test_module_ping_client(self):
"""
Ensure the function from main module pings clients correctly.
"""
daemon = client.ensure_client_daemon()
response = events.ping_client(daemon.get_port())
self.assertIsNotNone(response)
self.assertEqual(EventResponse.OK, response.status,
'Wrong response status.')
def test_client_replies_ping(self):
"""
Ensure clients reply to a ping.
"""
daemon = client.ensure_client_daemon()
port = daemon.get_port()
request = PingRequest()
service = RpcService(EventsClientService_Stub, port, 'localhost')
response = service.ping(request, timeout=1000)
self.assertEqual(EventResponse.OK, response.status,
'Wrong response status.')
def test_async_ping_client(self):
"""
Test asynchronous pinging of client.
"""
flag = Mock()
# executed after async signal, when response is received from server
def reqcbk(request, response):
flag(response.status)
daemon = client.ensure_client_daemon()
result = events.ping_client(daemon.get_port(), reqcbk=reqcbk)
self.assertIsNone(result)
time.sleep(1) # wait for response to arrive from server.
flag.assert_called_once_with(EventResponse.OK)