def test_delete_zip_file(self):
        with open(self.download_location, "wb") as zf:
            zf.write(self.zip_file.getvalue())

        delete_zip_file(self.download_location)

        self.assertFalse(os.path.exists(self.download_location))
    def test_delete_zip_file_error(self, mock_remove, mock_logger):
        e = "Error description"
        mock_remove.side_effect = Exception(e)
        error_message = "Failed to remove zip file {}, error {}".format(self.download_location, e)

        delete_zip_file(self.download_location)
        mock_logger.assert_called_with(error_message)