openstack/python-glanceclient
Code Issues Proposed changes

Remove incorrect validation for glance-download import method

Browse Source 
Since REMOTE SERVICE INTERFACE has default value to 'public',
it doesn't make sense to have validation for the same.

Removing this validation and added few missing tests for
glance-download import method

Closes-Bug: #2051761
Change-Id: I89adf23aac5db4f2c46379546def2f71d7c8e163
This commit is contained in:
Pranali Deore 2024-01-31 06:54:20 +00:00
parent 2952978676
commit 3c5dd2381f
2 changed files with 68 additions and 3 deletions
glanceclient
tests/unit/v2
test_shell_v2.py
v2
shell.py

68
glanceclient/tests/unit/v2/test_shell_v2.py

@ -2262,6 +2262,74 @@ class ShellV2Test(testtools.TestCase):
all_stores=None, allow_failure=True,
backend=None, stores=None)
@mock.patch('glanceclient.common.utils.exit')
def test_image_import_neg_no_glance_download_with_remote_region(
self, mock_utils_exit):
expected_msg = ("Import method should be 'glance-download' if "
"REMOTE REGION is provided.")
my_args = self.base_args.copy()
my_args['id'] = 'IMG-01'
my_args['remote_region'] = 'REGION2'
my_args['import_method'] = 'web-download'
my_args['uri'] = 'https://example.com/some/stuff'
args = self._make_args(my_args)
mock_utils_exit.side_effect = self._mock_utils_exit
with mock.patch.object(self.gc.images,
'get_import_info') as mocked_info:
mocked_info.return_value = self.import_info_response
try:
test_shell.do_image_import(self.gc, args)
self.fail("utils.exit should have been called")
except SystemExit:
pass
mock_utils_exit.assert_called_once_with(expected_msg)
@mock.patch('glanceclient.common.utils.exit')
def test_image_import_neg_no_glance_download_with_remote_id(
self, mock_utils_exit):
expected_msg = ("Import method should be 'glance-download' if "
"REMOTE IMAGE ID is provided.")
my_args = self.base_args.copy()
my_args['id'] = 'IMG-01'
my_args['remote_image_id'] = 'IMG-02'
my_args['import_method'] = 'web-download'
my_args['uri'] = 'https://example.com/some/stuff'
args = self._make_args(my_args)
mock_utils_exit.side_effect = self._mock_utils_exit
with mock.patch.object(self.gc.images,
'get_import_info') as mocked_info:
mocked_info.return_value = self.import_info_response
try:
test_shell.do_image_import(self.gc, args)
self.fail("utils.exit should have been called")
except SystemExit:
pass
mock_utils_exit.assert_called_once_with(expected_msg)
def test_image_import_glance_download_without_remote_service_interface(
self):
args = self._make_args(
{'id': 'IMG-01', 'uri': None, 'remote_region': 'REGION2',
'remote_image_id': 'IMG-02',
'import_method': 'glance-download'})
with mock.patch.object(self.gc.images, 'image_import') as mock_import:
with mock.patch.object(self.gc.images, 'get') as mocked_get:
with mock.patch.object(self.gc.images,
'get_import_info') as mocked_info:
mocked_get.return_value = {'status': 'queued',
'container_format': 'bare',
'disk_format': 'raw'}
mocked_info.return_value = self.import_info_response
mock_import.return_value = None
test_shell.do_image_import(self.gc, args)
mock_import.assert_called_once_with(
'IMG-01', 'glance-download',
uri=None, remote_region='REGION2',
remote_image_id='IMG-02',
remote_service_interface=None,
all_stores=None, allow_failure=True,
backend=None, stores=None)
@mock.patch('glanceclient.common.utils.print_image')
def test_image_import_no_print_image(self, mocked_utils_print_image):
args = self._make_args(

3
glanceclient/v2/shell.py

@ -838,9 +838,6 @@ def do_image_import(gc, args):
if remote_image_id and args.import_method != 'glance-download':
utils.exit("Import method should be 'glance-download' if "
"REMOTE IMAGE ID is provided.")
if remote_service_interface and args.import_method != 'glance-download':
utils.exit("Import method should be 'glance-download' if "
"REMOTE SERVICE INTERFACE is provided.")
if args.import_method == 'copy-image' and not (stores or all_stores):
utils.exit("Provide either --stores or --all-stores for "