From d22e6de9f42b1851352fa0fd314c01ac36104c98 Mon Sep 17 00:00:00 2001 From: Kevin James Date: Tue, 9 Jul 2019 11:40:30 -0700 Subject: [PATCH] tests(datastore): fix flaky tests --- datastore/tests/integration/query_test.py | 10 ++++++++++ 1 file changed, 10 insertions(+) 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