def test_cancel_accept(self):
self.server = yogi.TcpServer(self.scheduler, self.ADDRESS, self.PORT)
def accept_handler(res, connection):
self.accept_handler_res = res
self.server_connection = connection
self.server.async_accept(None, accept_handler)
self.server.cancel_accept()
while self.accept_handler_res is None:
pass
self.assertEqual(yogi.Canceled(), self.accept_handler_res)
def test_cancel_connect(self):
self.client = yogi.TcpClient(self.scheduler)
def connect_handler(res, connection):
self.connect_handler_res = res
self.client_connection = connection
self.client.async_connect(self.ADDRESS, self.PORT, None,
connect_handler)
self.client.cancel_connect()
while self.connect_handler_res is None:
pass
self.assertEqual(yogi.Canceled(), self.connect_handler_res)
def fn2(res, gathered_msg):
self.assertEqual(yogi.Canceled(), res)
self.assertIsNone(gathered_msg)
def fn(res, request):
self.assertEqual(yogi.Canceled(), res)
self.assertIsNone(request)
def fn(res, msg, cached):
self.assertEqual(yogi.Canceled(), res)
self.assertIsNone(msg)
def fn(res, scattered_msg):
self.assertEqual(yogi.Canceled(), res)
self.assertIsNone(scattered_msg)
def fn(res, info, change):
self.assertEqual(yogi.Canceled(), res)
self.assertIsNone(info)
self.assertIsNone(change)
def fn(res, state):
self.assertEqual(yogi.Canceled(), res)
