def test_string_conversion(self, method):
"""
Test ``EnterpriseCustomerUser`` conversion to string.
"""
file_mock = self._make_file_mock()
customer_branding_config = EnterpriseCustomerBrandingConfiguration(
id=1, logo=file_mock, enterprise_customer=EnterpriseCustomerFactory()
)
expected_str = "<EnterpriseCustomerBrandingConfiguration {ID}>: {enterprise_name}".format(
ID=customer_branding_config.id,
enterprise_name=customer_branding_config.enterprise_customer.name,
)
self.assertEqual(method(customer_branding_config), expected_str)
def test_branding_configuration_editing(self):
"""
Test enterprise customer branding configuration saves changes to existing instance.
"""
configuration = EnterpriseCustomerBrandingConfiguration(
enterprise_customer=EnterpriseCustomerFactory(),
logo="test1.png"
)
configuration.save()
self.assertEqual(configuration.logo.url, '/test1.png') # pylint: disable=no-member
configuration.logo = 'test2.png'
configuration.save()
self.assertEqual(configuration.logo.url, '/test2.png') # pylint: disable=no-member
def test_logo_path(self):
"""
Test path of image file should be enterprise/branding/<model.id>/<model_id>_logo.<ext>.lower().
"""
file_mock = self._make_file_mock()
branding_config = EnterpriseCustomerBrandingConfiguration(
id=1,
enterprise_customer=EnterpriseCustomerFactory(),
logo=file_mock
)
storage_mock = mock.MagicMock(spec=Storage, name="StorageMock")
with mock.patch("django.core.files.storage.default_storage._wrapped", storage_mock):
path = logo_path(branding_config, branding_config.logo.name)
self.assertEqual(path, "enterprise/branding/1/1_logo.png")
def test_logo_path(self, file_exists, delete_called):
"""
Test that the path of image file should beenterprise/branding/<model.id>/<model_id>_logo.<ext>.lower().
Additionally, test that the correct backend actions are taken in regards to deleting existing data.
"""
file_mock = self._make_file_mock()
branding_config = EnterpriseCustomerBrandingConfiguration(
id=1,
enterprise_customer=EnterpriseCustomerFactory(),
logo=file_mock
)
storage_mock = mock.MagicMock(spec=Storage, name="StorageMock")
storage_mock.exists.return_value = file_exists
with mock.patch("django.core.files.storage.default_storage._wrapped", storage_mock):
path = logo_path(branding_config, branding_config.logo.name)
self.assertEqual(path, "enterprise/branding/1/1_logo.png")
assert storage_mock.delete.call_count == (1 if delete_called else 0)
if delete_called:
storage_mock.delete.assert_called_once_with('enterprise/branding/1/1_logo.png')
def test_image_type(self, valid_image, image_extension):
"""
Test image type, currently .png is supported in configuration. see apps.py.
"""
file_mock = mock.MagicMock(spec=File, name="FileMock")
file_mock.name = "test1" + image_extension
file_mock.size = 2 * 1024
branding_configuration = EnterpriseCustomerBrandingConfiguration(
enterprise_customer=EnterpriseCustomerFactory(), logo=file_mock)
if not valid_image:
with self.assertRaises(ValidationError):
branding_configuration.full_clean()
else:
branding_configuration.full_clean(
) # exception here will fail the test
def test_image_size(self, valid_image, image_size):
"""
Test image size, image_size < (4 * 1024) e.g. 4kb. See apps.py.
"""
file_mock = mock.MagicMock(spec=File, name="FileMock")
file_mock.name = "test1.png"
file_mock.size = image_size
branding_configuration = EnterpriseCustomerBrandingConfiguration(
enterprise_customer=EnterpriseCustomerFactory(), logo=file_mock)
if not valid_image:
with self.assertRaises(ValidationError):
branding_configuration.full_clean()
else:
branding_configuration.full_clean(
) # exception here will fail the test
def test_branding_configuration_saving_successfully(self):
"""
Test enterprise customer branding configuration saving successfully.
"""
storage_mock = mock.MagicMock(spec=Storage, name="StorageMock")
branding_config_1 = EnterpriseCustomerBrandingConfiguration(
enterprise_customer=EnterpriseCustomerFactory(),
logo="test1.png"
)
storage_mock.exists.return_value = True
with mock.patch("django.core.files.storage.default_storage._wrapped", storage_mock):
branding_config_1.save()
self.assertEqual(EnterpriseCustomerBrandingConfiguration.objects.count(), 1)
branding_config_2 = EnterpriseCustomerBrandingConfiguration(
enterprise_customer=EnterpriseCustomerFactory(),
logo="test2.png"
)
storage_mock.exists.return_value = False
with mock.patch("django.core.files.storage.default_storage._wrapped", storage_mock):
branding_config_2.save()
self.assertEqual(EnterpriseCustomerBrandingConfiguration.objects.count(), 2)
def test_image_size(self, is_valid_image_size, image_size):
"""
Test image size in KB's, image_size < 512 KB.
Default valid max image size is 512 KB (512 * 1024 bytes).
See config `valid_max_image_size` in apps.py.
"""
file_mock = mock.MagicMock(spec=File, name="FileMock")
file_mock.name = "test1.png"
file_mock.size = image_size * 1024 # image size in bytes
branding_configuration = EnterpriseCustomerBrandingConfiguration(
enterprise_customer=EnterpriseCustomerFactory(),
logo=file_mock
)
if not is_valid_image_size:
with self.assertRaises(ValidationError) as validation_error:
branding_configuration.full_clean()
expected_validation_message = 'The logo image file size must be less than or equal to 512 KB.'
self.assertEqual(validation_error.exception.messages[0], expected_validation_message)
else:
branding_configuration.full_clean() # exception here will fail the test