def test_prepare(self):
field = AttributeField(attribute='foo.bar.baz', type=int)
related = Related()
field.prepare(mock_context(), related)
self.assertEqual(related._select, {'foo__bar'})
def test_prepare_with_use_prefetch(self):
field = AttributeField(attribute='foo.bar.baz', type=int, use_prefetch=True)
related = Related()
field.prepare(mock_context(), related)
self.assertEqual(related._prefetch, {
'foo__bar'
})
def test_prepare(self):
field = AttributeField(attribute='foo.bar.baz', type=int)
related = Related()
field.prepare(mock_context(), related)
self.assertEqual(related._select, {
'foo__bar'
})
def test_prepare_with_use_prefetch(self):
field = AttributeField(attribute='foo.bar.baz',
type=int,
use_prefetch=True)
related = Related()
field.prepare(mock_context(), related)
self.assertEqual(related._prefetch, {'foo__bar'})
