def test_pluck(self):
queryset = QuerySet(Item)
expected = range(1, 31)
self.assertItemsEqual(queryset.pluck('id'), expected)
expected = range(10) * 3
self.assertItemsEqual(queryset.pluck('number'), expected)
def test_filter_span_another_queryset_with_previouse_filters(self):
queryset = QuerySet(Item)
queryset2 = queryset.filter(ch='a')
self.assertEqual(queryset2.get_filters(), {'ch': 'a'})
queryset3 = queryset2.filter(number=2)
self.assertEqual(queryset3.get_filters(), {'ch': 'a', 'number': 2})
def test_filter_span(self):
queryset = QuerySet(Item)
Item.objects.collection.drop()
Item.objects.create(data=dict(id=101, ch='a', number=1))
Item.objects.create(data=dict(id=102, ch='a', number=1))
Item.objects.create(data=dict(id=103, ch='a', number=2))
Item.objects.create(data=dict(id=104, ch='b', number=1))
Item.objects.create(data=dict(id=105, ch='b', number=2))
self.assertItemsEqual(
queryset.filter(number=1).filter(ch='a').pluck('id'),
[101, 102])
self.assertItemsEqual(
queryset.filter(number=2).filter(ch='b').pluck('id'),
[105])
def test_filter_contains(self):
Item.objects.collection.drop()
Item.objects.create(data={'title': 'Slayer'})
Item.objects.create(data={'title': 'Sabaton'})
Item.objects.create(data={'title': 'Metallica'})
Item.objects.create(data={'title': 'Metal Foobar'})
Item.objects.create(data={'title': 'Tristania'})
queryset = QuerySet(Item)
self.assertItemsEqual(queryset.filter(title__contains='metal').pluck(
'title'), [])
self.assertItemsEqual(queryset.filter(title__contains='Metal').pluck(
'title'), ['Metallica', 'Metal Foobar'])
self.assertItemsEqual(queryset.filter(title__icontains='metal').pluck(
'title'), ['Metallica', 'Metal Foobar'])
self.assertItemsEqual(queryset.filter(title__icontains='ta').pluck(
'title'), ['Metallica', 'Metal Foobar', 'Tristania'])
def test_greater_less_than_equal(self):
Item.objects.collection.drop()
Item.objects.create(data={'id': 1, 'date': '2012-10-12', 'weight': 90})
Item.objects.create(data={'id': 2, 'date': '2012-10-13', 'weight': 88})
Item.objects.create(data={'id': 3, 'date': '2012-10-14', 'weight': 85})
Item.objects.create(data={'id': 4, 'date': '2012-10-15', 'weight': 80})
queryset = QuerySet(Item)
self.assertItemsEqual(queryset.filter(date__gt='2012-10-13').pluck('id'),
[3, 4])
self.assertItemsEqual(queryset.filter(date__gte='2012-10-13').pluck('id'),
[2, 3, 4])
self.assertItemsEqual(queryset.filter(weight__gte=85,
date__lt='2012-10-14').pluck('id'), [1, 2])
self.assertItemsEqual(queryset.filter(weight__lte=88).pluck('id'),
[2, 3, 4])
self.assertItemsEqual(queryset.filter(id__lte=2, weight__gte=88).pluck('id'),
[1, 2])
def test_order_by(self):
queryset = QuerySet(Item)
self.assertEquals(list(queryset.order_by('id').pluck('id')), range(1, 31))
def test_len_calls_count(self):
queryset = QuerySet(Item)
queryset.count = Mock(return_value=101)
self.assertEquals(len(queryset), 101)
queryset.count.assert_called_once_with()
def test_filter(self):
queryset = QuerySet(Item)
self.assertItemsEqual(queryset.filter(id=10).pluck('id'), [10])
self.assertItemsEqual(queryset.filter(number=0, id=10).pluck('id'), [10])
def test_filter_in(self):
queryset = QuerySet(Item)
self.assertItemsEqual(queryset.filter(id__in=[1, 3, 20, 40]).pluck('id'),
[1, 3, 20])
def test_pluck_with_multiple_fields(self):
queryset = QuerySet(Item).order_by('id')[:3]
self.assertItemsEqual(queryset.pluck('id', 'number'), [(1, 1), (2, 2), (3, 3)])
def test_get_filters(self):
queryset = QuerySet(Item).filter(slug__in=['foo', 'bar'], tag='baz')
self.assertDictEqual(queryset.get_filters(),
{'slug': {'$in': ['foo', 'bar']}, 'tag': 'baz'})
def test_all(self):
queryset = QuerySet(Item)
self.assertEquals(queryset.all().count(), 30)
def test_order_by_reversed(self):
queryset = QuerySet(Item)
self.assertEquals(list(queryset.order_by('-id').pluck('id')), range(30, 0, -1))
def test_combined_filter_order_slice(self):
queryset = QuerySet(Item).filter(number=2).order_by('id')[:2]
self.assertEquals(list(queryset.pluck('id')), [2, 12])
def test_count(self):
queryset = QuerySet(Item)
self.assertEqual(queryset.count(), 30)
def test_get_raises_doesnotexist(self):
queryset = QuerySet(Item).order_by('id')
with self.assertRaises(Item.DoesNotExist):
queryset.get(id=301)
def test_get_raises_multiple_docuemnts(self):
queryset = QuerySet(Item).order_by('id')
with self.assertRaises(MultipleItemsReturnedError):
queryset.get(number=2)
def test_get(self):
queryset = QuerySet(Item).order_by('id')
self.assertEquals(queryset.get(id=2).data['id'], 2)
