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'})