diff --git a/datastore/tests/integration/query_test.py b/datastore/tests/integration/query_test.py index 5bd2665..cb43fdd 100644 --- 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