From 76817ee97eb454d1a7622a6299ea7c067f882a26 Mon Sep 17 00:00:00 2001 From: Chris Wilcox Date: Fri, 3 Apr 2020 15:34:36 -0700 Subject: [PATCH] test: extend scenarios in test --- tests/system/test_system.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/tests/system/test_system.py b/tests/system/test_system.py index e5234d74..f3650c24 100644 --- a/tests/system/test_system.py +++ b/tests/system/test_system.py @@ -72,7 +72,7 @@ def setUpModule(): def tearDownModule(): keys = [entity.key for entity in Config.TO_DELETE] with Config.CLIENT.transaction(): - Config.CLIENT.delete_multi(keys) + Config.CLIENT.delete_multi(keys) class TestDatastore(unittest.TestCase): @@ -430,11 +430,18 @@ def test_query_paginate_with_offset(self): self.assertEqual(entities[2]["name"], "Arya") def test_query_paginate_with_large_offset(self): - page_query = self._base_query() - page_query.order = "appearances" + page_query = self.CLIENT.query(kind="Character") + page_query.add_filter("family", "=", "Stark") + page_query.add_filter("alive", "=", False) offset = 1000 # This is greater than the number of entries. limit = 3 + # Verify that with no offset there are results + iterator = page_query.fetch(limit=limit, offset=0) + page = six.next(iterator.pages) + entities_no_offset = list(page) + self.assertEqual(len(entities_no_offset), 3) + # Offset beyond items. Verify no items found iterator = page_query.fetch(limit=limit, offset=offset) page = six.next(iterator.pages)