def test_pre_save(self, filename, is_existing_file, mock_validate_size, mock_validate_type): """ Ensure that image validation and resizing take place only when a new file is being stored. """ # pylint: disable=too-many-function-args field_value = ResizingImageFieldFile(self.model_instance, self.field, filename) self.model_instance.resized_image = field_value self.field.attname = 'resized_image' self.field.name = 'resized_image' with mock.patch(PATCH_MODULE + '.ResizingImageFieldFile.create_resized_copies' ) as mock_resize: # actual file data is needed for this test to work with make_uploaded_file('image/jpeg', (1000, 1000)) as image_file: if filename: field_value.file = image_file field_value._committed = is_existing_file # pylint: disable=protected-access self.field.pre_save(self.model_instance, False) expected_called = bool(filename) and not is_existing_file for actual_called in (mock_validate_size.called, mock_validate_type.called, mock_resize.called): self.assertEqual(actual_called, expected_called)
def test_create_resized_copies(self): """ Ensure the create_resized_copies function produces and stores copies with the correct sizes and data. """ field_value = ResizingImageFieldFile(self.model_instance, self.field, 'test_name') with mock.patch.object(field_value, 'storage') as mock_storage: with make_image_file((300, 300)) as image_file: mock_storage.listdir = mock.Mock(return_value=([], [])) field_value.file = image_file field_value.create_resized_copies() self.assertEqual(mock_storage.save.call_count, len(TEST_SIZES)) actual_calls = dict((v[1] for v in mock_storage.save.mock_calls)) for width, height in TEST_SIZES: expected_name = 'test_name__{}x{}.jpg'.format(width, height) actual_data = actual_calls[expected_name] image_object = Image.open(actual_data) self.assertEqual(image_object.size, (width, height))
def test_pre_save(self, filename, is_existing_file, mock_validate_size, mock_validate_type): """ Ensure that image validation and resizing take place only when a new file is being stored. """ # pylint: disable=too-many-function-args field_value = ResizingImageFieldFile(self.model_instance, self.field, filename) self.model_instance.resized_image = field_value self.field.attname = 'resized_image' self.field.name = 'resized_image' with mock.patch(PATCH_MODULE + '.ResizingImageFieldFile.create_resized_copies') as mock_resize: # actual file data is needed for this test to work with make_uploaded_file('image/jpeg', (1000, 1000)) as image_file: if filename: field_value.file = image_file field_value._committed = is_existing_file # pylint: disable=protected-access self.field.pre_save(self.model_instance, False) expected_called = bool(filename) and not is_existing_file for actual_called in (mock_validate_size.called, mock_validate_type.called, mock_resize.called): self.assertEqual(actual_called, expected_called)