def test_prepare(self):
class MockResource(ModelResource):
model_class = mock_orm.Model
fields = [AttributeField('bar.baz', type=int)]
class MockQuerySetResource(QuerySetResource):
resource_class = MockResource
field = RelatedManagerField(attribute='foo',
resource_class=MockQuerySetResource)
related = Related()
field.prepare(mock_context(), related)
self.assertEqual(related._prefetch, {'foo', 'foo__bar'})
def test_prepare(self):
class MockResource(ModelResource):
model_class = mock_orm.Model
fields = [
AttributeField('bar.baz', type=int)
]
class MockQuerySetResource(QuerySetResource):
resource_class = MockResource
field = RelatedManagerField(attribute='foo', resource_class=MockQuerySetResource)
related = Related()
field.prepare(mock_context(), related)
self.assertEqual(related._prefetch, {
'foo',
'foo__bar'
})
