def test_blur_image(image_mock, os_mock, capsys): filename = str(uuid.uuid4()) os_mock.remove = MagicMock() os_mock.path = MagicMock() os_mock.path.basename = MagicMock(side_effect=(lambda x: x)) image_mock.return_value = image_mock image_mock.__enter__.return_value = image_mock blob = UserDict() blob.name = filename blob.bucket = UserDict() blob.bucket.blob = MagicMock(return_value=blob) blob.download_to_filename = MagicMock() blob.upload_from_filename = MagicMock() main.__blur_image(blob) out, _ = capsys.readouterr() assert f'Image {filename} was downloaded to' in out assert f'Image {filename} was blurred.' in out assert f'Blurred image was uploaded to blurred-{filename}.' in out assert os_mock.remove.called assert image_mock.resize.called
def test_blur_image(storage_client, image_mock, os_mock, capsys): filename = str(uuid.uuid4()) blur_bucket = 'blurred-bucket-' + str(uuid.uuid4()) os_mock.remove = MagicMock() os_mock.path = MagicMock() os_mock.path.basename = MagicMock(side_effect=(lambda x: x)) os_mock.getenv = MagicMock(return_value=blur_bucket) image_mock.return_value = image_mock image_mock.__enter__.return_value = image_mock blob = UserDict() blob.name = filename blob.bucket = UserDict() blob.download_to_filename = MagicMock() blob.upload_from_filename = MagicMock() image.__blur_image(blob) out, _ = capsys.readouterr() assert f'Image {filename} was downloaded to' in out assert f'Image {filename} was blurred.' in out assert f'Blurred image uploaded to: gs://{blur_bucket}/{filename}' in out assert os_mock.remove.called assert image_mock.resize.called