def test_raise_unknown_method(self, mapper):
ctx = mock_context()
mapper.get.side_effect = AttributeError
stub_model = self._get_stub([{'name': 23}, {'name': 10}])
mock_sum = Mock(spec='name')
mock_sum.name = 'sum'
target_dict = {}
field = AggregateField('name', mock_sum)
with self.assertRaises(AttributeError):
field.handle_outgoing(ctx, stub_model, target_dict)
def test_raise_unknown_method(self, mapper):
ctx = mock_context()
mapper.get.side_effect = AttributeError
stub_model = self._get_stub([{'name': 23}, {'name': 10}])
mock_sum = Mock(spec='name')
mock_sum.name = 'sum'
target_dict = {}
field = AggregateField('name', mock_sum)
with self.assertRaises(AttributeError):
field.handle_outgoing(ctx, stub_model, target_dict)
def test_handle_outgoing(self):
ctx = mock_context()
mock_model = Mock(name='model')
mock_model.name__count = 14
target_dict = {}
field = AggregateField('name', django.db.models.Count)
field.handle_outgoing(ctx, mock_model, target_dict)
self.assertEqual(
14,
target_dict['name'],
)
def test_handle_outgoing(self):
ctx = mock_context()
mock_model = Mock(name='model')
mock_model.name__count = 14
target_dict = {}
field = AggregateField('name', django.db.models.Count)
field.handle_outgoing(ctx, mock_model, target_dict)
self.assertEqual(
14,
target_dict['name'],
)
def test_handel_outgoing_sum_non(self, mapper):
ctx = mock_context()
target_dict = {}
mock_sum = Mock(spec='name')
mock_sum.name = 'sum'
mapper.get.return_value = sum
field = AggregateField('name', mock_sum)
stub_model = self._get_stub([{'name': None}])
field.handle_outgoing(ctx, stub_model, target_dict)
self.assertEqual(
0,
target_dict['name'],
)
def test_handel_outgoing_sum_non(self, mapper):
ctx = mock_context()
target_dict = {}
mock_sum = Mock(spec='name')
mock_sum.name = 'sum'
mapper.get.return_value = sum
field = AggregateField('name', mock_sum)
stub_model = self._get_stub([{'name': None}])
field.handle_outgoing(ctx, stub_model, target_dict)
self.assertEqual(
0,
target_dict['name'],
)
def test_handel_outgoing_sum_attr_error(self, mapper):
ctx = mock_context()
target_dict = {}
mock_sum = Mock(spec='name')
mock_sum.name = 'sum'
mapper.get.return_value = sum
field = AggregateField('name', mock_sum)
stub_model = self._get_stub([{'name': 23}, {'name': 10}])
field.handle_outgoing(ctx, stub_model, target_dict)
self.assertEqual(
33,
target_dict['name'],
)
StubModel.objects.filter.assert_called_with(pk=2)
def test_handel_outgoing_sum_attr_error(self, mapper):
ctx = mock_context()
target_dict = {}
mock_sum = Mock(spec='name')
mock_sum.name = 'sum'
mapper.get.return_value = sum
field = AggregateField('name', mock_sum)
stub_model = self._get_stub([{'name': 23}, {'name': 10}])
field.handle_outgoing(ctx, stub_model, target_dict)
self.assertEqual(
33,
target_dict['name'],
)
StubModel.objects.filter.assert_called_with(pk=2)
