def test_npr_apply_no_parent():
qs = NestedPrefetchRelated('pages').apply(default_qs)
assert qs._prefetch_related_lookups == ('pages',)
def test_nested_apply_annotation_parent():
qs = NestedPrefetchRelated('pages').rebuild(Annotation(**{'a': 1})).apply(default_qs)
assert qs._prefetch_related_lookups == ('pages',)
def test_nested_prefetch_object_in_parent():
with pytest.raises(AssertionError) as e:
NestedPrefetchRelated('pages').rebuild(PrefetchRelated(Prefetch('pages')))
assert str(e.value) == 'Only simple parent relations are supported.'
def test_npr_apply_parent(parent):
qs = NestedPrefetchRelated('publisher').rebuild(parent).apply(default_qs)
assert qs._prefetch_related_lookups == ('author__publisher',)