Example #1
0
    def test_empty_prepare(self):
        related = Related()

        queryset = related.prepare(mock_orm.QuerySet())

        self.assertEqual(queryset._selected, set())
        self.assertEqual(queryset._prefetched, set())
Example #2
0
    def test_prepare(self):
        related = Related()

        related.select('foo')
        related.prefetch('bar')

        queryset = related.prepare(mock_orm.QuerySet())

        self.assertEqual(queryset._selected, {
            'foo'
        })
        self.assertEqual(queryset._prefetched, {
            'bar'
        })
Example #3
0
 def prepare_queryset(self, ctx, queryset):
     related = Related()
     self.prepare(ctx, related)
     return related.prepare(queryset)
Example #4
0
 def prepare_queryset(self, ctx, queryset):
     related = Related()
     self.prepare(ctx, related)
     return related.prepare(queryset)
Example #5
0
 def _prefetch_related(self, qs):
     related = Related()
     ctx = APIContext('', None, None)
     self.resource_class.prepare(ctx, related)
     self.prefetch_related(related)
     return related.prepare(qs)
Example #6
0
 def _prefetch_related(self, qs):
     related = Related()
     ctx = APIContext('', None, None)
     self.resource_class.prepare(ctx, related)
     self.prefetch_related(related)
     return related.prepare(qs)