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)