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)