def test_module_ping_server(self):
"""
Ensure the function from main module pings server correctly.
"""
response = events.ping_server()
self.assertIsNotNone(response)
self.assertEqual(EventResponse.OK, response.status,
'Wrong response status.')
def test_async_ping_server(self):
"""
Test asynchronous pinging of server.
"""
flag = Mock()
# executed after async signal, when response is received from server
def reqcbk(request, response):
flag(response.status)
result = events.ping_server(reqcbk=reqcbk)
self.assertIsNone(result)
time.sleep(1) # wait for response to arrive from server.
flag.assert_called_once_with(EventResponse.OK)
