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)