コード例 #1
0
ファイル: test_querysets.py プロジェクト: lukaszb/djmongo
    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)
コード例 #2
0
ファイル: test_querysets.py プロジェクト: lukaszb/djmongo
    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})
コード例 #3
0
ファイル: test_querysets.py プロジェクト: lukaszb/djmongo
    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])
コード例 #4
0
ファイル: test_querysets.py プロジェクト: lukaszb/djmongo
    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'])
コード例 #5
0
ファイル: test_querysets.py プロジェクト: lukaszb/djmongo
    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])
コード例 #6
0
ファイル: test_querysets.py プロジェクト: lukaszb/djmongo
 def test_order_by(self):
     queryset = QuerySet(Item)
     self.assertEquals(list(queryset.order_by('id').pluck('id')), range(1, 31))
コード例 #7
0
ファイル: test_querysets.py プロジェクト: lukaszb/djmongo
 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()
コード例 #8
0
ファイル: test_querysets.py プロジェクト: lukaszb/djmongo
    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])
コード例 #9
0
ファイル: test_querysets.py プロジェクト: lukaszb/djmongo
 def test_filter_in(self):
     queryset = QuerySet(Item)
     self.assertItemsEqual(queryset.filter(id__in=[1, 3, 20, 40]).pluck('id'),
         [1, 3, 20])
コード例 #10
0
ファイル: test_querysets.py プロジェクト: lukaszb/djmongo
 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)])
コード例 #11
0
ファイル: test_querysets.py プロジェクト: lukaszb/djmongo
 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'})
コード例 #12
0
ファイル: test_querysets.py プロジェクト: lukaszb/djmongo
 def test_all(self):
     queryset = QuerySet(Item)
     self.assertEquals(queryset.all().count(), 30)
コード例 #13
0
ファイル: test_querysets.py プロジェクト: lukaszb/djmongo
 def test_order_by_reversed(self):
     queryset = QuerySet(Item)
     self.assertEquals(list(queryset.order_by('-id').pluck('id')), range(30, 0, -1))
コード例 #14
0
ファイル: test_querysets.py プロジェクト: lukaszb/djmongo
 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])
コード例 #15
0
ファイル: test_querysets.py プロジェクト: lukaszb/djmongo
 def test_count(self):
     queryset = QuerySet(Item)
     self.assertEqual(queryset.count(), 30)
コード例 #16
0
ファイル: test_querysets.py プロジェクト: lukaszb/djmongo
 def test_get_raises_doesnotexist(self):
     queryset = QuerySet(Item).order_by('id')
     with self.assertRaises(Item.DoesNotExist):
         queryset.get(id=301)
コード例 #17
0
ファイル: test_querysets.py プロジェクト: lukaszb/djmongo
 def test_get_raises_multiple_docuemnts(self):
     queryset = QuerySet(Item).order_by('id')
     with self.assertRaises(MultipleItemsReturnedError):
         queryset.get(number=2)
コード例 #18
0
ファイル: test_querysets.py プロジェクト: lukaszb/djmongo
 def test_get(self):
     queryset = QuerySet(Item).order_by('id')
     self.assertEquals(queryset.get(id=2).data['id'], 2)