def test_prepare(self):
class MockResource(ModelResource):
model_class = Mock()
fields = [
AttributeField(attribute='bar.baz', type=int),
]
field = SubModelResourceField(attribute='foo',
resource_class=MockResource)
related = Related()
field.prepare(mock_context(), related)
self.assertEqual(related._select, {'foo', 'foo__bar'})
def test_prepare(self):
class MockResource(ModelResource):
model_class = Mock()
fields = [
AttributeField(attribute='bar.baz', type=int),
]
field = SubModelResourceField(attribute='foo', resource_class=MockResource)
related = Related()
field.prepare(mock_context(), related)
self.assertEqual(related._select, {
'foo',
'foo__bar'
})
