Wait when checking the pool status in tests

Author: beltran

tests/integration/standard/test_authentication.py

@@ -20,7 +20,7 @@
 
 fromtests.integrationimportuse_singledc, get_cluster, remove_cluster, PROTOCOL_VERSION, CASSANDRA_IP, \
 set_default_cass_ip
-fromtests.integration.utilimportassert_quiescent_pool_state
+fromtests.integration.utilimportassert_quiescent_pool_state, assert_quiescent_pool_state_with_wait
 
 try:
 importunittest2asunittest
@@ -93,7 +93,7 @@ def test_auth_connect(self):
 session=cluster.connect()
 try:
 self.assertTrue(session.execute('SELECT release_version FROM system.local'))
-assert_quiescent_pool_state(self, cluster)
+assert_quiescent_pool_state_with_wait(self, cluster, wait=1)
 forpoolinsession.get_pools():
 connection, _=pool.borrow_connection(timeout=0)
 self.assertEqual(connection.authenticator.server_authenticator_class, 'org.apache.cassandra.auth.PasswordAuthenticator')
@@ -102,7 +102,7 @@ def test_auth_connect(self):
 cluster.shutdown()
 finally:
 root_session.execute('DROP USER %s', user)
-assert_quiescent_pool_state(self, root_session.cluster)
+assert_quiescent_pool_state_with_wait(self, root_session.cluster, wait=1)
 root_session.cluster.shutdown()
 
 deftest_connect_wrong_pwd(self):

tests/integration/util.py

@@ -13,6 +13,25 @@
 # limitations under the License.
 
 fromtests.integrationimportPROTOCOL_VERSION
+importtime
+
+
+defassert_quiescent_pool_state_with_wait(test_case, cluster, wait):
+"""
+ This function is useful for waiting before checking the pool state.
+ assert_quiescent_pool_state checks that none of the requests ids have got lost.
+ However the callback corresponding to a request_id is called before the request_id
+ is returned back to the pool, therfore
+
+ session.execute("SELECT * from system.local")
+ assert_quiescent_pool_state(self, session.cluster)
+
+ might fail because when execute comes back the request_id hasn't yet been
+ returned to the pool, therefore the wait.
+ """
+time.sleep(wait)
+assert_quiescent_pool_state(test_case, cluster)
+
 
 defassert_quiescent_pool_state(test_case, cluster):