def test_pre_save_update(self, serialize_model_instance):
model = TestModel(
string_field='Test',
integer_field=1,
float_field=1.0,
id=1
)
sender = MagicMock()
sender.objects.get.return_value = model
serialize_model_instance.return_value = model
signals._pre_save(sender, instance=model)
self.assertEqual(signals._CACHE, {id(model): model})
def test_pre_save_new(self):
model = TestModel(
string_field='Test',
integer_field=1,
float_field=1.0,
id=1
)
sender = MagicMock()
sender.objects.get.side_effect = Exception('Test Exception')
signals._pre_save(sender, instance=model)
self.assertEqual(signals._CACHE, {})
def test_pre_save_fail(self, logger):
sender = MagicMock()
signals._pre_save(sender)
self.assertEqual(logger.exception.call_count, 1)