def test_save_model_create_link(self):
mock_get_or_create_link = Mock(return_value=(self.mock_link, True,))
booking_type = BookingTypeF.build()
booking_type.get_or_create_link = mock_get_or_create_link
self.bookingtypeadmin.save_model(self.mock_request, booking_type,
self.mock_form, True)
mock_get_or_create_link.assert_is_called_once()
self.mock_message.assert_called_once_with(
self.mock_request,
'Link to "{0}" successfully created'.format(self.mock_link.slug),
level=messages.SUCCESS)
def test_save_model_raise_iomproperly_configured(self):
msg = 'ImproperlyConfigured message'
mock_get_or_create_link = Mock(side_effect=ImproperlyConfigured(msg))
self.bookingtypeadmin.message_user = self.mock_message
booking_type = BookingTypeF.build()
booking_type.get_or_create_link = mock_get_or_create_link
self.bookingtypeadmin.save_model(self.mock_request, booking_type,
self.mock_form, True)
mock_get_or_create_link.assert_is_called_once()
self.mock_message.assert_called_once_with(
self.mock_request,
'Error on Link creation: {0}'.format(msg),
level=messages.ERROR)
