tests: Remove unneeded eventually() call for ancestor queries. (#965)

googleapis · Mar 8, 2024 · 98fb176 · 98fb176
1 parent 47db5b9
commit 98fb176
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/system/test_query.py b/tests/system/test_query.py
@@ -156,7 +156,7 @@ class SomeKind(ndb.Model):
  foo = ndb.IntegerProperty()
 
  query = SomeKind.query(ancestor=ndb.Key(KIND, root_id))
- results = eventually(query.fetch, length_equals(6))
+ results = query.fetch()
 
  results = sorted(results, key=operator.attrgetter("foo"))
  assert [entity.foo for entity in results] == [-1, 0, 1, 2, 3, 4]
@@ -180,7 +180,7 @@ class Dummy(ndb.Model):
 
  with client_context.new(namespace=other_namespace).use():
  query = Dummy.query(ancestor=parent_key, namespace="xyz")
- results = eventually(query.fetch, length_equals(2))
+ results = query.fetch()
 
  assert results[0].foo == "bar"
  assert results[1].foo == "child"
@@ -206,7 +206,7 @@ class Dummy(ndb.Model):
 
  with client_context.new(namespace=other_namespace).use():
  query = Dummy.query(ancestor=parent_key, namespace="")
- results = eventually(query.fetch, length_equals(2))
+ results = query.fetch()
 
  assert results[0].foo == "bar"
  assert results[1].foo == "child"