tests(datastore): fix flaky tests

talkiq · Jul 9, 2019 · d22e6de · d22e6de
1 parent db4994e
commit d22e6de
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/datastore/tests/integration/query_test.py b/datastore/tests/integration/query_test.py
@@ -1,3 +1,5 @@
+import time
+
 import requests
 
 from gcloud.rest.datastore import Datastore
@@ -36,6 +38,10 @@ def test_query(creds, kind, project):
         ]
         ds.commit(mutations, transaction=transaction, session=s)
 
+        # TODO: figure out why this is flaky without the sleep
+        # Seems to only be flaky intermittently in py2.
+        time.sleep(2)
+
         after = ds.runQuery(query, session=s)
         assert len(after.entity_results) == num_results + 2
 
@@ -65,5 +71,9 @@ def test_gql_query(creds, kind, project):
         ]
         ds.commit(mutations, transaction=transaction, session=s)
 
+        # TODO: figure out why this is flaky without the sleep
+        # Seems to only be flaky intermittently in py2.
+        time.sleep(2)
+
         after = ds.runQuery(query, session=s)
         assert len(after.entity_results) == num_results + 3