fix unstable connection test wait proxy ioloop wakeup finish

Author: snower

tests/__init__.py

@@ -3,7 +3,10 @@
 
 import os
 from tormysql import ConnectionPool
-from tornado.testing import AsyncTestCase
+try:
+    from tornado.testing import AsyncTestCase
+except:
+    from unittest import TestCase as AsyncTestCase
 
 class BaseTestCase(AsyncTestCase):
     PARAMS = dict(

tests/test_unstable_connection.py

@@ -82,12 +82,14 @@ def on_connect(server, conn):
         self.PARAMS['port'] = self.pport
         self.PARAMS['host'] = '127.0.0.1'
         sevent.current().wakeup()
+        time.sleep(0.1)
 
     def _close_proxy_sessions(self):
         def do_close():
             for request in TestThroughProxy.proxys:
                 request.conn.end()
         sevent.current().wakeup(do_close)
+        time.sleep(0.1)
 
     def tearDown(self):
         try:
@@ -96,6 +98,7 @@ def do_close():
                     request.conn.end()
                 self.proxy_server.close()
             sevent.current().wakeup(do_close)
+            time.sleep(0.1)
         except:
             pass
         super(BaseTestCase, self).tearDown()
@@ -116,6 +119,7 @@ def _execute_test_connection_closing(self):
             raise AssertionError("Unexpected normal situation")
 
         sevent.current().wakeup(self.proxy_server.close)
+        time.sleep(0.1)
 
     @gen.coroutine
     def _execute_test_connection_closed(self):
@@ -125,6 +129,7 @@ def _execute_test_connection_closed(self):
         yield conn.close()
 
         sevent.current().wakeup(self.proxy_server.close)
+        time.sleep(0.1)
 
         try:
             yield Connection(**self.PARAMS)
@@ -148,6 +153,7 @@ def _execute_test_remote_closing(self):
             yield conn.do_close()
 
             sevent.current().wakeup(self.proxy_server.close)
+            time.sleep(0.1)
 
             yield pool.Connection()
         except OperationalError:
@@ -179,6 +185,7 @@ def _execute_test_pool_closing(self):
             yield pool.close()
 
         sevent.current().wakeup(self.proxy_server.close)
+        time.sleep(0.1)
 
     @gen_test
     def test(self):