def show_vmss_encryption_status(cmd, resource_group_name, vmss_name): encryption_ext_names = [v['name'] for v in vmss_extension_info.values()] views = get_vmss_instance_view(cmd, resource_group_name, vmss_name) if not views.extensions or not [ e for e in views.extensions if e.name in encryption_ext_names ]: from knack.util import CLIError raise CLIError("'{}' is not encrypted yet".format(vmss_name)) views = get_vmss_instance_view(cmd, resource_group_name, vmss_name, instance_id='*') result = [{'disks': v.disks, 'extensions': v.extensions} for v in views] # get rid of unrelaed disk status for r in result: for d in r['disks']: d.statuses = [ s for s in d.statuses if s.code.startswith('EncryptionState') ] r['encryption-extension'] = next((e for e in (r['extensions'] or []) if e.name in encryption_ext_names), None) r.pop('extensions') # we don't need this array any more return result
def test_show_vmss_instance_view(self, factory_mock): vm_client = mock.MagicMock() factory_mock.return_value = vm_client # execute get_vmss_instance_view('rg1', 'vmss1', '*') # assert vm_client.virtual_machine_scale_set_vms.list.assert_called_once_with( 'rg1', 'vmss1', expand='instanceView', select='instanceView')
def test_show_vmss_instance_view(self, factory_mock): vm_client = mock.MagicMock() factory_mock.return_value = vm_client # execute get_vmss_instance_view('rg1', 'vmss1', '*') # assert vm_client.virtual_machine_scale_set_vms.list.assert_called_once_with('rg1', 'vmss1', expand='instanceView', select='instanceView')