def test_slice(self): es = FakeXFormES() for i in xrange(0, 1300): es.add_doc(i, {'i': i}) queryset = ESQuerySet(es_client=es, payload={}) qs_slice = list(queryset[3:7]) self.assertEqual(es.queries[0]['from'], 3) self.assertEqual(es.queries[0]['size'], 4) self.assertEqual(len(qs_slice), 4) queryset = ESQuerySet(es_client=es, payload={}) qs_slice = list(queryset[10:20]) self.assertEqual(es.queries[1]['from'], 10) self.assertEqual(es.queries[1]['size'], 10) self.assertEqual(len(qs_slice), 10) queryset = ESQuerySet(es_client=es, payload={}) qs_slice = list(queryset[500:1000]) self.assertEqual(es.queries[2]['from'], 500) self.assertEqual(es.queries[2]['size'], 500) self.assertEqual(len(qs_slice), 500)
def test_order_by(self): es = FakeXFormES() for i in xrange(0, 1300): es.add_doc(i, {'i': i}) queryset = ESQuerySet(es_client=es, payload={}) qs_asc = list(queryset.order_by('foo')) self.assertEqual(es.queries[0]['sort'], [{'foo': 'asc'}]) qs_desc = list(queryset.order_by('-foo')) self.assertEqual(es.queries[1]['sort'], [{'foo': 'desc'}]) qs_overwrite = list(queryset.order_by('bizzle').order_by('-baz')) self.assertEqual(es.queries[2]['sort'], [{'baz': 'desc'}]) qs_multi = list(queryset.order_by('one', '-two', 'three')) self.assertEqual(es.queries[3]['sort'], [{'one': 'asc'}, {'two': 'desc'}, {'three': 'asc'}])
def test_order_by(self): es = FakeXFormES() for i in xrange(0, 1300): es.add_doc(i, {"i": i}) queryset = ESQuerySet(es_client=es, payload={}) qs_asc = list(queryset.order_by("foo")) self.assertEqual(es.queries[0]["sort"], [{"foo": "asc"}]) qs_desc = list(queryset.order_by("-foo")) self.assertEqual(es.queries[1]["sort"], [{"foo": "desc"}]) qs_overwrite = list(queryset.order_by("bizzle").order_by("-baz")) self.assertEqual(es.queries[2]["sort"], [{"baz": "desc"}]) qs_multi = list(queryset.order_by("one", "-two", "three")) self.assertEqual(es.queries[3]["sort"], [{"one": "asc"}, {"two": "desc"}, {"three": "asc"}])