def test_prepare_not_top_level(self):
ctx = mock_context()
related = Mock(name='related')
field = AggregateField('name.one.two', django.db.models.Count)
with self.assertRaises(SavoryPieError) as cm:
field.prepare(ctx, related)
self.assertEqual(
'RelatedCountField can only be used on a top level ModelResource',
str(cm.exception))
def test_prepare_not_top_level(self):
ctx = mock_context()
related = Mock(name='related')
field = AggregateField('name.one.two', django.db.models.Count)
with self.assertRaises(SavoryPieError) as cm:
field.prepare(ctx, related)
self.assertEqual(
'RelatedCountField can only be used on a top level ModelResource',
str(cm.exception)
)
def test_prepare(self):
ctx = mock_context()
ctx.peek.side_effect = IndexError
related = Mock(name='related')
field = AggregateField('name.one.two', django.db.models.Count)
field.prepare(ctx, related)
related.annotate.assert_called_with(
django.db.models.Count,
'name__one__two',
distinct=True,
)
def test_prepare(self):
ctx = mock_context()
ctx.peek.side_effect = IndexError
related = Mock(name='related')
field = AggregateField('name.one.two', django.db.models.Count)
field.prepare(ctx, related)
related.annotate.assert_called_with(
django.db.models.Count,
'name__one__two',
distinct=True,
)