 fc24142ed4
			
		
	
	fc24142ed4
	
	
	
		
			
			Add pagination parameters, '--limit' and '--marker'. This isn't compatible with our client-side '--project' parameter so we error out for that. Change-Id: I403cf0fb7aabad4a3dfda5adae62d47ecf7faf5c Signed-off-by: Stephen Finucane <sfinucan@redhat.com>
		
			
				
	
	
		
			780 lines
		
	
	
		
			25 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			780 lines
		
	
	
		
			25 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| #   Copyright 2016 IBM
 | |
| #
 | |
| #   Licensed under the Apache License, Version 2.0 (the "License"); you may
 | |
| #   not use this file except in compliance with the License. You may obtain
 | |
| #   a copy of the License at
 | |
| #
 | |
| #        http://www.apache.org/licenses/LICENSE-2.0
 | |
| #
 | |
| #   Unless required by applicable law or agreed to in writing, software
 | |
| #   distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 | |
| #   WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 | |
| #   License for the specific language governing permissions and limitations
 | |
| #   under the License.
 | |
| #
 | |
| 
 | |
| import copy
 | |
| from unittest import mock
 | |
| from unittest.mock import call
 | |
| import uuid
 | |
| 
 | |
| from novaclient import api_versions
 | |
| from openstack import utils as sdk_utils
 | |
| from osc_lib import exceptions
 | |
| 
 | |
| from openstackclient.compute.v2 import keypair
 | |
| from openstackclient.tests.unit.compute.v2 import fakes as compute_fakes
 | |
| from openstackclient.tests.unit import fakes
 | |
| from openstackclient.tests.unit.identity.v2_0 import fakes as identity_fakes
 | |
| from openstackclient.tests.unit import utils as tests_utils
 | |
| 
 | |
| 
 | |
| class TestKeypair(compute_fakes.TestComputev2):
 | |
| 
 | |
|     def setUp(self):
 | |
|         super(TestKeypair, self).setUp()
 | |
| 
 | |
|         # Initialize the user mock
 | |
|         self.users_mock = self.app.client_manager.identity.users
 | |
|         self.users_mock.reset_mock()
 | |
|         self.users_mock.get.return_value = fakes.FakeResource(
 | |
|             None,
 | |
|             copy.deepcopy(identity_fakes.USER),
 | |
|             loaded=True,
 | |
|         )
 | |
| 
 | |
|         self.app.client_manager.sdk_connection = mock.Mock()
 | |
|         self.app.client_manager.sdk_connection.compute = mock.Mock()
 | |
|         self.sdk_client = self.app.client_manager.sdk_connection.compute
 | |
|         self.sdk_client.keypairs = mock.Mock()
 | |
|         self.sdk_client.create_keypair = mock.Mock()
 | |
|         self.sdk_client.delete_keypair = mock.Mock()
 | |
|         self.sdk_client.find_keypair = mock.Mock()
 | |
| 
 | |
| 
 | |
| class TestKeypairCreate(TestKeypair):
 | |
| 
 | |
|     keypair = compute_fakes.FakeKeypair.create_one_keypair()
 | |
| 
 | |
|     def setUp(self):
 | |
|         super(TestKeypairCreate, self).setUp()
 | |
| 
 | |
|         self.columns = (
 | |
|             'fingerprint',
 | |
|             'name',
 | |
|             'type',
 | |
|             'user_id'
 | |
|         )
 | |
|         self.data = (
 | |
|             self.keypair.fingerprint,
 | |
|             self.keypair.name,
 | |
|             self.keypair.type,
 | |
|             self.keypair.user_id
 | |
|         )
 | |
| 
 | |
|         # Get the command object to test
 | |
|         self.cmd = keypair.CreateKeypair(self.app, None)
 | |
| 
 | |
|         self.sdk_client.create_keypair.return_value = self.keypair
 | |
| 
 | |
|     def test_key_pair_create_no_options(self):
 | |
| 
 | |
|         arglist = [
 | |
|             self.keypair.name,
 | |
|         ]
 | |
|         verifylist = [
 | |
|             ('name', self.keypair.name),
 | |
|         ]
 | |
|         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
 | |
| 
 | |
|         columns, data = self.cmd.take_action(parsed_args)
 | |
| 
 | |
|         self.sdk_client.create_keypair.assert_called_with(
 | |
|             name=self.keypair.name
 | |
|         )
 | |
| 
 | |
|         self.assertEqual({}, columns)
 | |
|         self.assertEqual({}, data)
 | |
| 
 | |
|     def test_keypair_create_public_key(self):
 | |
|         # overwrite the setup one because we want to omit private_key
 | |
|         self.keypair = compute_fakes.FakeKeypair.create_one_keypair(
 | |
|             no_pri=True)
 | |
|         self.sdk_client.create_keypair.return_value = self.keypair
 | |
| 
 | |
|         self.data = (
 | |
|             self.keypair.fingerprint,
 | |
|             self.keypair.name,
 | |
|             self.keypair.type,
 | |
|             self.keypair.user_id
 | |
|         )
 | |
| 
 | |
|         arglist = [
 | |
|             '--public-key', self.keypair.public_key,
 | |
|             self.keypair.name,
 | |
|         ]
 | |
|         verifylist = [
 | |
|             ('public_key', self.keypair.public_key),
 | |
|             ('name', self.keypair.name),
 | |
|         ]
 | |
| 
 | |
|         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
 | |
| 
 | |
|         with mock.patch('io.open') as mock_open:
 | |
|             mock_open.return_value = mock.MagicMock()
 | |
|             m_file = mock_open.return_value.__enter__.return_value
 | |
|             m_file.read.return_value = 'dummy'
 | |
| 
 | |
|             columns, data = self.cmd.take_action(parsed_args)
 | |
| 
 | |
|             self.sdk_client.create_keypair.assert_called_with(
 | |
|                 name=self.keypair.name,
 | |
|                 public_key=self.keypair.public_key,
 | |
|             )
 | |
| 
 | |
|             self.assertEqual(self.columns, columns)
 | |
|             self.assertEqual(self.data, data)
 | |
| 
 | |
|     def test_keypair_create_private_key(self):
 | |
|         tmp_pk_file = '/tmp/kp-file-' + uuid.uuid4().hex
 | |
|         arglist = [
 | |
|             '--private-key', tmp_pk_file,
 | |
|             self.keypair.name,
 | |
|         ]
 | |
|         verifylist = [
 | |
|             ('private_key', tmp_pk_file),
 | |
|             ('name', self.keypair.name),
 | |
|         ]
 | |
| 
 | |
|         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
 | |
| 
 | |
|         with mock.patch('io.open') as mock_open:
 | |
|             mock_open.return_value = mock.MagicMock()
 | |
|             m_file = mock_open.return_value.__enter__.return_value
 | |
| 
 | |
|             columns, data = self.cmd.take_action(parsed_args)
 | |
| 
 | |
|             self.sdk_client.create_keypair.assert_called_with(
 | |
|                 name=self.keypair.name,
 | |
|             )
 | |
| 
 | |
|             mock_open.assert_called_once_with(tmp_pk_file, 'w+')
 | |
|             m_file.write.assert_called_once_with(self.keypair.private_key)
 | |
| 
 | |
|             self.assertEqual(self.columns, columns)
 | |
|             self.assertEqual(self.data, data)
 | |
| 
 | |
|     @mock.patch.object(sdk_utils, 'supports_microversion', return_value=True)
 | |
|     def test_keypair_create_with_key_type(self, sm_mock):
 | |
|         for key_type in ['x509', 'ssh']:
 | |
|             self.keypair = compute_fakes.FakeKeypair.create_one_keypair(
 | |
|                 no_pri=True)
 | |
|             self.sdk_client.create_keypair.return_value = self.keypair
 | |
| 
 | |
|             self.data = (
 | |
|                 self.keypair.fingerprint,
 | |
|                 self.keypair.name,
 | |
|                 self.keypair.type,
 | |
|                 self.keypair.user_id,
 | |
|             )
 | |
|             arglist = [
 | |
|                 '--public-key', self.keypair.public_key,
 | |
|                 self.keypair.name,
 | |
|                 '--type', key_type,
 | |
|             ]
 | |
|             verifylist = [
 | |
|                 ('public_key', self.keypair.public_key),
 | |
|                 ('name', self.keypair.name),
 | |
|                 ('type', key_type),
 | |
|             ]
 | |
|             parsed_args = self.check_parser(self.cmd, arglist, verifylist)
 | |
| 
 | |
|             with mock.patch('io.open') as mock_open:
 | |
|                 mock_open.return_value = mock.MagicMock()
 | |
|                 m_file = mock_open.return_value.__enter__.return_value
 | |
|                 m_file.read.return_value = 'dummy'
 | |
|                 columns, data = self.cmd.take_action(parsed_args)
 | |
| 
 | |
|             self.sdk_client.create_keypair.assert_called_with(
 | |
|                 name=self.keypair.name,
 | |
|                 public_key=self.keypair.public_key,
 | |
|                 key_type=key_type,
 | |
|             )
 | |
| 
 | |
|             self.assertEqual(self.columns, columns)
 | |
|             self.assertEqual(self.data, data)
 | |
| 
 | |
|     @mock.patch.object(sdk_utils, 'supports_microversion', return_value=False)
 | |
|     def test_keypair_create_with_key_type_pre_v22(self, sm_mock):
 | |
|         for key_type in ['x509', 'ssh']:
 | |
|             arglist = [
 | |
|                 '--public-key', self.keypair.public_key,
 | |
|                 self.keypair.name,
 | |
|                 '--type', 'ssh',
 | |
|             ]
 | |
|             verifylist = [
 | |
|                 ('public_key', self.keypair.public_key),
 | |
|                 ('name', self.keypair.name),
 | |
|                 ('type', 'ssh'),
 | |
|             ]
 | |
|             parsed_args = self.check_parser(self.cmd, arglist, verifylist)
 | |
| 
 | |
|             with mock.patch('io.open') as mock_open:
 | |
|                 mock_open.return_value = mock.MagicMock()
 | |
|                 m_file = mock_open.return_value.__enter__.return_value
 | |
|                 m_file.read.return_value = 'dummy'
 | |
| 
 | |
|                 ex = self.assertRaises(
 | |
|                     exceptions.CommandError,
 | |
|                     self.cmd.take_action,
 | |
|                     parsed_args)
 | |
| 
 | |
|             self.assertIn(
 | |
|                 '--os-compute-api-version 2.2 or greater is required',
 | |
|                 str(ex))
 | |
| 
 | |
|     @mock.patch.object(sdk_utils, 'supports_microversion', return_value=True)
 | |
|     def test_key_pair_create_with_user(self, sm_mock):
 | |
|         arglist = [
 | |
|             '--user', identity_fakes.user_name,
 | |
|             self.keypair.name,
 | |
|         ]
 | |
|         verifylist = [
 | |
|             ('user', identity_fakes.user_name),
 | |
|             ('name', self.keypair.name),
 | |
|         ]
 | |
|         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
 | |
| 
 | |
|         columns, data = self.cmd.take_action(parsed_args)
 | |
| 
 | |
|         self.sdk_client.create_keypair.assert_called_with(
 | |
|             name=self.keypair.name,
 | |
|             user_id=identity_fakes.user_id,
 | |
|         )
 | |
| 
 | |
|         self.assertEqual({}, columns)
 | |
|         self.assertEqual({}, data)
 | |
| 
 | |
|     @mock.patch.object(sdk_utils, 'supports_microversion', return_value=False)
 | |
|     def test_key_pair_create_with_user_pre_v210(self, sm_mock):
 | |
|         arglist = [
 | |
|             '--user', identity_fakes.user_name,
 | |
|             self.keypair.name,
 | |
|         ]
 | |
|         verifylist = [
 | |
|             ('user', identity_fakes.user_name),
 | |
|             ('name', self.keypair.name),
 | |
|         ]
 | |
|         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
 | |
| 
 | |
|         ex = self.assertRaises(
 | |
|             exceptions.CommandError,
 | |
|             self.cmd.take_action,
 | |
|             parsed_args)
 | |
|         self.assertIn(
 | |
|             '--os-compute-api-version 2.10 or greater is required', str(ex))
 | |
| 
 | |
| 
 | |
| class TestKeypairDelete(TestKeypair):
 | |
| 
 | |
|     keypairs = compute_fakes.FakeKeypair.create_keypairs(count=2)
 | |
| 
 | |
|     def setUp(self):
 | |
|         super(TestKeypairDelete, self).setUp()
 | |
| 
 | |
|         self.cmd = keypair.DeleteKeypair(self.app, None)
 | |
| 
 | |
|     def test_keypair_delete(self):
 | |
|         arglist = [
 | |
|             self.keypairs[0].name
 | |
|         ]
 | |
|         verifylist = [
 | |
|             ('name', [self.keypairs[0].name]),
 | |
|         ]
 | |
| 
 | |
|         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
 | |
| 
 | |
|         ret = self.cmd.take_action(parsed_args)
 | |
| 
 | |
|         self.assertIsNone(ret)
 | |
|         self.sdk_client.delete_keypair.assert_called_with(
 | |
|             self.keypairs[0].name, ignore_missing=False)
 | |
| 
 | |
|     def test_delete_multiple_keypairs(self):
 | |
|         arglist = []
 | |
|         for k in self.keypairs:
 | |
|             arglist.append(k.name)
 | |
|         verifylist = [
 | |
|             ('name', arglist),
 | |
|         ]
 | |
| 
 | |
|         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
 | |
|         result = self.cmd.take_action(parsed_args)
 | |
| 
 | |
|         calls = []
 | |
|         for k in self.keypairs:
 | |
|             calls.append(call(k.name, ignore_missing=False))
 | |
|         self.sdk_client.delete_keypair.assert_has_calls(calls)
 | |
|         self.assertIsNone(result)
 | |
| 
 | |
|     def test_delete_multiple_keypairs_with_exception(self):
 | |
|         arglist = [
 | |
|             self.keypairs[0].name,
 | |
|             'unexist_keypair',
 | |
|         ]
 | |
|         verifylist = [
 | |
|             ('name', arglist),
 | |
|         ]
 | |
| 
 | |
|         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
 | |
| 
 | |
|         self.sdk_client.delete_keypair.side_effect = [
 | |
|             None, exceptions.CommandError]
 | |
|         try:
 | |
|             self.cmd.take_action(parsed_args)
 | |
|             self.fail('CommandError should be raised.')
 | |
|         except exceptions.CommandError as e:
 | |
|             self.assertEqual('1 of 2 keys failed to delete.', str(e))
 | |
| 
 | |
|         calls = []
 | |
|         for k in arglist:
 | |
|             calls.append(call(k, ignore_missing=False))
 | |
|         self.sdk_client.delete_keypair.assert_has_calls(calls)
 | |
| 
 | |
|     @mock.patch.object(sdk_utils, 'supports_microversion', return_value=True)
 | |
|     def test_keypair_delete_with_user(self, sm_mock):
 | |
|         arglist = [
 | |
|             '--user', identity_fakes.user_name,
 | |
|             self.keypairs[0].name
 | |
|         ]
 | |
|         verifylist = [
 | |
|             ('user', identity_fakes.user_name),
 | |
|             ('name', [self.keypairs[0].name]),
 | |
|         ]
 | |
|         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
 | |
| 
 | |
|         ret = self.cmd.take_action(parsed_args)
 | |
| 
 | |
|         self.assertIsNone(ret)
 | |
|         self.sdk_client.delete_keypair.assert_called_with(
 | |
|             self.keypairs[0].name,
 | |
|             user_id=identity_fakes.user_id,
 | |
|             ignore_missing=False
 | |
|         )
 | |
| 
 | |
|     @mock.patch.object(sdk_utils, 'supports_microversion', return_value=False)
 | |
|     def test_keypair_delete_with_user_pre_v210(self, sm_mock):
 | |
| 
 | |
|         self.app.client_manager.compute.api_version = \
 | |
|             api_versions.APIVersion('2.9')
 | |
| 
 | |
|         arglist = [
 | |
|             '--user', identity_fakes.user_name,
 | |
|             self.keypairs[0].name
 | |
|         ]
 | |
|         verifylist = [
 | |
|             ('user', identity_fakes.user_name),
 | |
|             ('name', [self.keypairs[0].name]),
 | |
|         ]
 | |
|         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
 | |
| 
 | |
|         ex = self.assertRaises(
 | |
|             exceptions.CommandError,
 | |
|             self.cmd.take_action,
 | |
|             parsed_args)
 | |
|         self.assertIn(
 | |
|             '--os-compute-api-version 2.10 or greater is required', str(ex))
 | |
| 
 | |
| 
 | |
| class TestKeypairList(TestKeypair):
 | |
| 
 | |
|     # Return value of self.sdk_client.keypairs().
 | |
|     keypairs = compute_fakes.FakeKeypair.create_keypairs(count=1)
 | |
| 
 | |
|     def setUp(self):
 | |
|         super(TestKeypairList, self).setUp()
 | |
| 
 | |
|         self.sdk_client.keypairs.return_value = self.keypairs
 | |
| 
 | |
|         # Get the command object to test
 | |
|         self.cmd = keypair.ListKeypair(self.app, None)
 | |
| 
 | |
|     @mock.patch.object(sdk_utils, 'supports_microversion', return_value=False)
 | |
|     def test_keypair_list_no_options(self, sm_mock):
 | |
|         arglist = []
 | |
|         verifylist = []
 | |
| 
 | |
|         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
 | |
| 
 | |
|         # In base command class Lister in cliff, abstract method take_action()
 | |
|         # returns a tuple containing the column names and an iterable
 | |
|         # containing the data to be listed.
 | |
|         columns, data = self.cmd.take_action(parsed_args)
 | |
| 
 | |
|         # Set expected values
 | |
| 
 | |
|         self.sdk_client.keypairs.assert_called_with()
 | |
| 
 | |
|         self.assertEqual(('Name', 'Fingerprint'), columns)
 | |
|         self.assertEqual(
 | |
|             ((self.keypairs[0].name, self.keypairs[0].fingerprint), ),
 | |
|             tuple(data)
 | |
|         )
 | |
| 
 | |
|     @mock.patch.object(sdk_utils, 'supports_microversion', return_value=True)
 | |
|     def test_keypair_list_v22(self, sm_mock):
 | |
|         arglist = []
 | |
|         verifylist = []
 | |
| 
 | |
|         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
 | |
| 
 | |
|         # In base command class Lister in cliff, abstract method take_action()
 | |
|         # returns a tuple containing the column names and an iterable
 | |
|         # containing the data to be listed.
 | |
|         columns, data = self.cmd.take_action(parsed_args)
 | |
| 
 | |
|         # Set expected values
 | |
| 
 | |
|         self.sdk_client.keypairs.assert_called_with()
 | |
| 
 | |
|         self.assertEqual(('Name', 'Fingerprint', 'Type'), columns)
 | |
|         self.assertEqual(
 | |
|             ((
 | |
|                 self.keypairs[0].name,
 | |
|                 self.keypairs[0].fingerprint,
 | |
|                 self.keypairs[0].type,
 | |
|             ), ),
 | |
|             tuple(data)
 | |
|         )
 | |
| 
 | |
|     @mock.patch.object(sdk_utils, 'supports_microversion', return_value=True)
 | |
|     def test_keypair_list_with_user(self, sm_mock):
 | |
| 
 | |
|         users_mock = self.app.client_manager.identity.users
 | |
|         users_mock.reset_mock()
 | |
|         users_mock.get.return_value = fakes.FakeResource(
 | |
|             None,
 | |
|             copy.deepcopy(identity_fakes.USER),
 | |
|             loaded=True,
 | |
|         )
 | |
| 
 | |
|         arglist = [
 | |
|             '--user', identity_fakes.user_name,
 | |
|         ]
 | |
|         verifylist = [
 | |
|             ('user', identity_fakes.user_name),
 | |
|         ]
 | |
|         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
 | |
| 
 | |
|         columns, data = self.cmd.take_action(parsed_args)
 | |
| 
 | |
|         users_mock.get.assert_called_with(identity_fakes.user_name)
 | |
|         self.sdk_client.keypairs.assert_called_with(
 | |
|             user_id=identity_fakes.user_id,
 | |
|         )
 | |
| 
 | |
|         self.assertEqual(('Name', 'Fingerprint', 'Type'), columns)
 | |
|         self.assertEqual(
 | |
|             ((
 | |
|                 self.keypairs[0].name,
 | |
|                 self.keypairs[0].fingerprint,
 | |
|                 self.keypairs[0].type,
 | |
|             ), ),
 | |
|             tuple(data)
 | |
|         )
 | |
| 
 | |
|     @mock.patch.object(sdk_utils, 'supports_microversion', return_value=False)
 | |
|     def test_keypair_list_with_user_pre_v210(self, sm_mock):
 | |
| 
 | |
|         arglist = [
 | |
|             '--user', identity_fakes.user_name,
 | |
|         ]
 | |
|         verifylist = [
 | |
|             ('user', identity_fakes.user_name),
 | |
|         ]
 | |
|         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
 | |
| 
 | |
|         ex = self.assertRaises(
 | |
|             exceptions.CommandError,
 | |
|             self.cmd.take_action,
 | |
|             parsed_args)
 | |
|         self.assertIn(
 | |
|             '--os-compute-api-version 2.10 or greater is required', str(ex))
 | |
| 
 | |
|     @mock.patch.object(sdk_utils, 'supports_microversion', return_value=True)
 | |
|     def test_keypair_list_with_project(self, sm_mock):
 | |
| 
 | |
|         projects_mock = self.app.client_manager.identity.tenants
 | |
|         projects_mock.reset_mock()
 | |
|         projects_mock.get.return_value = fakes.FakeResource(
 | |
|             None,
 | |
|             copy.deepcopy(identity_fakes.PROJECT),
 | |
|             loaded=True,
 | |
|         )
 | |
| 
 | |
|         users_mock = self.app.client_manager.identity.users
 | |
|         users_mock.reset_mock()
 | |
|         users_mock.list.return_value = [
 | |
|             fakes.FakeResource(
 | |
|                 None,
 | |
|                 copy.deepcopy(identity_fakes.USER),
 | |
|                 loaded=True,
 | |
|             ),
 | |
|         ]
 | |
| 
 | |
|         arglist = ['--project', identity_fakes.project_name]
 | |
|         verifylist = [('project', identity_fakes.project_name)]
 | |
|         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
 | |
| 
 | |
|         columns, data = self.cmd.take_action(parsed_args)
 | |
| 
 | |
|         projects_mock.get.assert_called_with(identity_fakes.project_name)
 | |
|         users_mock.list.assert_called_with(tenant_id=identity_fakes.project_id)
 | |
|         self.sdk_client.keypairs.assert_called_with(
 | |
|             user_id=identity_fakes.user_id,
 | |
|         )
 | |
| 
 | |
|         self.assertEqual(('Name', 'Fingerprint', 'Type'), columns)
 | |
|         self.assertEqual(
 | |
|             ((
 | |
|                 self.keypairs[0].name,
 | |
|                 self.keypairs[0].fingerprint,
 | |
|                 self.keypairs[0].type,
 | |
|             ), ),
 | |
|             tuple(data)
 | |
|         )
 | |
| 
 | |
|     @mock.patch.object(sdk_utils, 'supports_microversion', return_value=False)
 | |
|     def test_keypair_list_with_project_pre_v210(self, sm_mock):
 | |
| 
 | |
|         arglist = ['--project', identity_fakes.project_name]
 | |
|         verifylist = [('project', identity_fakes.project_name)]
 | |
| 
 | |
|         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
 | |
|         ex = self.assertRaises(
 | |
|             exceptions.CommandError,
 | |
|             self.cmd.take_action,
 | |
|             parsed_args)
 | |
|         self.assertIn(
 | |
|             '--os-compute-api-version 2.10 or greater is required', str(ex))
 | |
| 
 | |
|     def test_keypair_list_conflicting_user_options(self):
 | |
| 
 | |
|         arglist = [
 | |
|             '--user', identity_fakes.user_name,
 | |
|             '--project', identity_fakes.project_name,
 | |
|         ]
 | |
| 
 | |
|         self.assertRaises(
 | |
|             tests_utils.ParserException,
 | |
|             self.check_parser, self.cmd, arglist, None)
 | |
| 
 | |
|     @mock.patch.object(
 | |
|         sdk_utils, 'supports_microversion', new=mock.Mock(return_value=True))
 | |
|     def test_keypair_list_with_limit(self):
 | |
|         arglist = [
 | |
|             '--limit', '1',
 | |
|         ]
 | |
|         verifylist = [
 | |
|             ('limit', 1),
 | |
|         ]
 | |
| 
 | |
|         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
 | |
|         self.cmd.take_action(parsed_args)
 | |
| 
 | |
|         self.sdk_client.keypairs.assert_called_with(limit=1)
 | |
| 
 | |
|     @mock.patch.object(
 | |
|         sdk_utils, 'supports_microversion', new=mock.Mock(return_value=False))
 | |
|     def test_keypair_list_with_limit_pre_v235(self):
 | |
|         arglist = [
 | |
|             '--limit', '1',
 | |
|         ]
 | |
|         verifylist = [
 | |
|             ('limit', 1),
 | |
|         ]
 | |
| 
 | |
|         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
 | |
|         ex = self.assertRaises(
 | |
|             exceptions.CommandError,
 | |
|             self.cmd.take_action,
 | |
|             parsed_args)
 | |
| 
 | |
|         self.assertIn(
 | |
|             '--os-compute-api-version 2.35 or greater is required', str(ex))
 | |
| 
 | |
|     @mock.patch.object(
 | |
|         sdk_utils, 'supports_microversion', new=mock.Mock(return_value=True))
 | |
|     def test_keypair_list_with_marker(self):
 | |
|         arglist = [
 | |
|             '--marker', 'test_kp',
 | |
|         ]
 | |
|         verifylist = [
 | |
|             ('marker', 'test_kp'),
 | |
|         ]
 | |
| 
 | |
|         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
 | |
|         self.cmd.take_action(parsed_args)
 | |
| 
 | |
|         self.sdk_client.keypairs.assert_called_with(marker='test_kp')
 | |
| 
 | |
|     @mock.patch.object(
 | |
|         sdk_utils, 'supports_microversion', new=mock.Mock(return_value=False))
 | |
|     def test_keypair_list_with_marker_pre_v235(self):
 | |
|         arglist = [
 | |
|             '--marker', 'test_kp',
 | |
|         ]
 | |
|         verifylist = [
 | |
|             ('marker', 'test_kp'),
 | |
|         ]
 | |
| 
 | |
|         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
 | |
|         ex = self.assertRaises(
 | |
|             exceptions.CommandError,
 | |
|             self.cmd.take_action,
 | |
|             parsed_args)
 | |
| 
 | |
|         self.assertIn(
 | |
|             '--os-compute-api-version 2.35 or greater is required', str(ex))
 | |
| 
 | |
| 
 | |
| class TestKeypairShow(TestKeypair):
 | |
| 
 | |
|     keypair = compute_fakes.FakeKeypair.create_one_keypair()
 | |
| 
 | |
|     def setUp(self):
 | |
|         super(TestKeypairShow, self).setUp()
 | |
| 
 | |
|         self.sdk_client.find_keypair.return_value = self.keypair
 | |
| 
 | |
|         self.cmd = keypair.ShowKeypair(self.app, None)
 | |
| 
 | |
|         self.columns = (
 | |
|             "fingerprint",
 | |
|             "name",
 | |
|             "type",
 | |
|             "user_id"
 | |
|         )
 | |
| 
 | |
|         self.data = (
 | |
|             self.keypair.fingerprint,
 | |
|             self.keypair.name,
 | |
|             self.keypair.type,
 | |
|             self.keypair.user_id
 | |
|         )
 | |
| 
 | |
|     def test_keypair_show_no_options(self):
 | |
| 
 | |
|         arglist = []
 | |
|         verifylist = []
 | |
| 
 | |
|         # Missing required args should boil here
 | |
|         self.assertRaises(tests_utils.ParserException, self.check_parser,
 | |
|                           self.cmd, arglist, verifylist)
 | |
| 
 | |
|     def test_keypair_show(self):
 | |
|         # overwrite the setup one because we want to omit private_key
 | |
|         self.keypair = compute_fakes.FakeKeypair.create_one_keypair(
 | |
|             no_pri=True)
 | |
|         self.sdk_client.find_keypair.return_value = self.keypair
 | |
| 
 | |
|         self.data = (
 | |
|             self.keypair.fingerprint,
 | |
|             self.keypair.name,
 | |
|             self.keypair.type,
 | |
|             self.keypair.user_id
 | |
|         )
 | |
| 
 | |
|         arglist = [
 | |
|             self.keypair.name
 | |
|         ]
 | |
|         verifylist = [
 | |
|             ('name', self.keypair.name)
 | |
|         ]
 | |
|         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
 | |
| 
 | |
|         columns, data = self.cmd.take_action(parsed_args)
 | |
| 
 | |
|         self.sdk_client.find_keypair.assert_called_with(
 | |
|             self.keypair.name,
 | |
|             ignore_missing=False
 | |
|         )
 | |
| 
 | |
|         self.assertEqual(self.columns, columns)
 | |
|         self.assertEqual(self.data, data)
 | |
| 
 | |
|     def test_keypair_show_public(self):
 | |
| 
 | |
|         arglist = [
 | |
|             '--public-key',
 | |
|             self.keypair.name
 | |
|         ]
 | |
|         verifylist = [
 | |
|             ('public_key', True),
 | |
|             ('name', self.keypair.name)
 | |
|         ]
 | |
| 
 | |
|         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
 | |
| 
 | |
|         columns, data = self.cmd.take_action(parsed_args)
 | |
| 
 | |
|         self.assertEqual({}, columns)
 | |
|         self.assertEqual({}, data)
 | |
| 
 | |
|     @mock.patch.object(sdk_utils, 'supports_microversion', return_value=True)
 | |
|     def test_keypair_show_with_user(self, sm_mock):
 | |
| 
 | |
|         # overwrite the setup one because we want to omit private_key
 | |
|         self.keypair = compute_fakes.FakeKeypair.create_one_keypair(
 | |
|             no_pri=True)
 | |
|         self.sdk_client.find_keypair.return_value = self.keypair
 | |
| 
 | |
|         self.data = (
 | |
|             self.keypair.fingerprint,
 | |
|             self.keypair.name,
 | |
|             self.keypair.type,
 | |
|             self.keypair.user_id
 | |
|         )
 | |
| 
 | |
|         arglist = [
 | |
|             '--user', identity_fakes.user_name,
 | |
|             self.keypair.name,
 | |
|         ]
 | |
|         verifylist = [
 | |
|             ('user', identity_fakes.user_name),
 | |
|             ('name', self.keypair.name)
 | |
|         ]
 | |
|         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
 | |
| 
 | |
|         columns, data = self.cmd.take_action(parsed_args)
 | |
| 
 | |
|         self.users_mock.get.assert_called_with(identity_fakes.user_name)
 | |
|         self.sdk_client.find_keypair.assert_called_with(
 | |
|             self.keypair.name,
 | |
|             ignore_missing=False,
 | |
|             user_id=identity_fakes.user_id
 | |
|         )
 | |
| 
 | |
|         self.assertEqual(self.columns, columns)
 | |
|         self.assertEqual(self.data, data)
 | |
| 
 | |
|     @mock.patch.object(sdk_utils, 'supports_microversion', return_value=False)
 | |
|     def test_keypair_show_with_user_pre_v210(self, sm_mock):
 | |
| 
 | |
|         arglist = [
 | |
|             '--user', identity_fakes.user_name,
 | |
|             self.keypair.name,
 | |
|         ]
 | |
|         verifylist = [
 | |
|             ('user', identity_fakes.user_name),
 | |
|             ('name', self.keypair.name)
 | |
|         ]
 | |
|         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
 | |
| 
 | |
|         ex = self.assertRaises(
 | |
|             exceptions.CommandError,
 | |
|             self.cmd.take_action,
 | |
|             parsed_args)
 | |
|         self.assertIn(
 | |
|             '--os-compute-api-version 2.10 or greater is required', str(ex))
 |