def setUp(self): self.photo = BasePhoto() self.photo.image = lambda _: _ self.photo.image.url = 'url atrribute' self.photo.image.height = 'height atrribute' self.photo.image.width = 'width atrribute' self.photo.image_height = 'height field' self.photo.image_width = 'width field' self.photo.backup_image = 'backup image' self.get_safe_image = lambda self: self.photo._get_safe_image( 'image', 'backup_image')
class BasePhotoGetSafeImageTest(TestCase): def setUp(self): self.photo = BasePhoto() self.photo.image = lambda _: _ self.photo.image.url = 'url atrribute' self.photo.image.height = 'height atrribute' self.photo.image.width = 'width atrribute' self.photo.image_height = 'height field' self.photo.image_width = 'width field' self.photo.backup_image = 'backup image' self.get_safe_image = lambda self: self.photo._get_safe_image( 'image', 'backup_image') def test_no_image_returns_backup_image(self): self.photo.image = False assert self.get_safe_image(self) == 'backup image' @override_settings(CANADA_IMAGE_DIMENSION_FIELDS='_') def test_returns_url(self): assert self.get_safe_image(self)['url'] == 'url atrribute' @override_settings(CANADA_IMAGE_DIMENSION_FIELDS=True) def test_returns_dimension_fields(self): assert self.get_safe_image(self)['width'] == 'width field' assert self.get_safe_image(self)['height'] == 'height field' @override_settings(CANADA_IMAGE_DIMENSION_FIELDS=False) def test_no_canada_image_dimension_fields(self): assert self.get_safe_image(self)['width'] == 'width atrribute' assert self.get_safe_image(self)['height'] == 'height atrribute'