daniepusb
diff --git a/‎datastore/cloud-ndb/flask_app_test.py
Copy file name to clipboardExpand all lines: datastore/cloud-ndb/flask_app_test.py
+3-2Lines changed: 3 additions & 2 deletions b/‎datastore/cloud-ndb/flask_app_test.py
Copy file name to clipboardExpand all lines: datastore/cloud-ndb/flask_app_test.py
+3-2Lines changed: 3 additions & 2 deletions
diff --git a/‎datastore/cloud-ndb/quickstart_test.py
Copy file name to clipboardExpand all lines: datastore/cloud-ndb/quickstart_test.py
+2-1Lines changed: 2 additions & 1 deletion b/‎datastore/cloud-ndb/quickstart_test.py
Copy file name to clipboardExpand all lines: datastore/cloud-ndb/quickstart_test.py
+2-1Lines changed: 2 additions & 1 deletion
@@ -43,8 +43,9 @@ def test_index(test_book):
     @backoff.on_exception(backoff.expo, AssertionError, max_time=60)
     def eventually_consistent_test():
         r = client.get('/')
-        assert r.status_code == 200
-        assert test_book.title in r.data.decode('utf-8')
+        with flask_app.client.context():
+            assert r.status_code == 200
+            assert test_book.title in r.data.decode('utf-8')
 
     eventually_consistent_test()
 
 
@@ -39,6 +39,7 @@ def test_quickstart(capsys, test_book):
     def eventually_consistent_test():
         quickstart.list_books()
         out, _ = capsys.readouterr()
-        assert test_book.title in out
+        with quickstart.client.context():
+            assert test_book.title in out
 
     eventually_consistent_test()