def test_iterate_multi(MultiQueryIterator):
query = mock.Mock(
filters=mock.Mock(_multiquery=True, spec=("_multiquery",)),
spec=("filters",),
)
iterator = MultiQueryIterator.return_value
assert _datastore_query.iterate(query) is iterator
MultiQueryIterator.assert_called_once_with(query, raw=False)
def test_iterate_single_w_filters(QueryIterator):
query = mock.Mock(
filters=mock.Mock(
_multiquery=False,
_post_filters=mock.Mock(return_value=None),
spec=("_multiquery", "_post_filters"),
),
spec=("filters", "_post_filters"),
)
iterator = QueryIterator.return_value
assert _datastore_query.iterate(query) is iterator
QueryIterator.assert_called_once_with(query, raw=False)
def test_iterate_single_with_post_filter(QueryIterator):
query = mock.Mock(
filters=mock.Mock(_multiquery=False,
spec=("_multiquery", "_post_filters")),
spec=("filters", "_post_filters"),
)
iterator = QueryIterator.return_value
post_filters = query.filters._post_filters.return_value
predicate = post_filters._to_filter.return_value
assert _datastore_query.iterate(query) is iterator
QueryIterator.assert_called_once_with(query, predicate, raw=False)
post_filters._to_filter.assert_called_once_with(post=True)
def test_iterate_single(QueryIterator):
query = mock.Mock(filters=None, spec=("filters"))
iterator = QueryIterator.return_value
assert _datastore_query.iterate(query) is iterator
QueryIterator.assert_called_once_with(query, raw=False)
