def test_empty_prepare(self): related = Related() queryset = related.prepare(mock_orm.QuerySet()) self.assertEqual(queryset._selected, set()) self.assertEqual(queryset._prefetched, set())
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' })
def prepare_queryset(self, ctx, queryset): related = Related() self.prepare(ctx, related) return related.prepare(queryset)
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)