467 lines
21 KiB
Python
467 lines
21 KiB
Python
#
|
|
# 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 lxml.etree
|
|
import mock
|
|
import random
|
|
import requests_mock
|
|
|
|
import dracclient.client
|
|
from dracclient import exceptions
|
|
import dracclient.resources.job
|
|
from dracclient.resources import raid
|
|
from dracclient.resources import uris
|
|
from dracclient.tests import base
|
|
from dracclient.tests import utils as test_utils
|
|
from dracclient import utils
|
|
|
|
|
|
@requests_mock.Mocker()
|
|
class ClientRAIDManagementTestCase(base.BaseTest):
|
|
|
|
def setUp(self):
|
|
super(ClientRAIDManagementTestCase, self).setUp()
|
|
self.drac_client = dracclient.client.DRACClient(
|
|
**test_utils.FAKE_ENDPOINT)
|
|
|
|
def test_list_raid_controllers(self, mock_requests):
|
|
expected_raid_controller = raid.RAIDController(
|
|
id='RAID.Integrated.1-1',
|
|
description='Integrated RAID Controller 1',
|
|
manufacturer='DELL',
|
|
model='PERC H710 Mini',
|
|
primary_status='ok',
|
|
firmware_version='21.3.0-0009')
|
|
|
|
mock_requests.post(
|
|
'https://1.2.3.4:443/wsman',
|
|
text=test_utils.RAIDEnumerations[uris.DCIM_ControllerView]['ok'])
|
|
|
|
self.assertIn(expected_raid_controller,
|
|
self.drac_client.list_raid_controllers())
|
|
|
|
def test_list_virtual_disks(self, mock_requests):
|
|
expected_virtual_disk = raid.VirtualDisk(
|
|
id='Disk.Virtual.0:RAID.Integrated.1-1',
|
|
name='disk 0',
|
|
description='Virtual Disk 0 on Integrated RAID Controller 1',
|
|
controller='RAID.Integrated.1-1',
|
|
raid_level='1',
|
|
size_mb=571776,
|
|
status='ok',
|
|
raid_status='online',
|
|
span_depth=1,
|
|
span_length=2,
|
|
pending_operations=None,
|
|
physical_disks=[
|
|
'Disk.Bay.0:Enclosure.Internal.0-1:RAID.Integrated.1-1',
|
|
'Disk.Bay.1:Enclosure.Internal.0-1:RAID.Integrated.1-1'
|
|
])
|
|
|
|
mock_requests.post(
|
|
'https://1.2.3.4:443/wsman',
|
|
text=test_utils.RAIDEnumerations[uris.DCIM_VirtualDiskView]['ok'])
|
|
|
|
self.assertIn(expected_virtual_disk,
|
|
self.drac_client.list_virtual_disks())
|
|
|
|
def test_list_physical_disks(self, mock_requests):
|
|
expected_physical_disk = raid.PhysicalDisk(
|
|
id='Disk.Bay.1:Enclosure.Internal.0-1:RAID.Integrated.1-1',
|
|
description=('Disk 1 in Backplane 1 of '
|
|
'Integrated RAID Controller 1'),
|
|
controller='RAID.Integrated.1-1',
|
|
manufacturer='SEAGATE',
|
|
model='ST600MM0006',
|
|
media_type='hdd',
|
|
interface_type='sas',
|
|
size_mb=571776,
|
|
free_size_mb=571776,
|
|
serial_number='S0M3EY2Z',
|
|
firmware_version='LS0A',
|
|
status='ok',
|
|
raid_status='ready')
|
|
|
|
mock_requests.post(
|
|
'https://1.2.3.4:443/wsman',
|
|
text=test_utils.RAIDEnumerations[uris.DCIM_PhysicalDiskView]['ok'])
|
|
|
|
self.assertIn(expected_physical_disk,
|
|
self.drac_client.list_physical_disks())
|
|
|
|
def test_list_physical_disks_direct(self, mock_requests):
|
|
expected_physical_disk = raid.PhysicalDisk(
|
|
id='Disk.Direct.2:RAID.Integrated.1-1',
|
|
description=('Disk 2 on '
|
|
'Integrated RAID Controller 1'),
|
|
controller='RAID.Integrated.1-1',
|
|
manufacturer='ATA',
|
|
model='ST600MM0007',
|
|
media_type='ssd',
|
|
interface_type='sata',
|
|
size_mb=571776,
|
|
free_size_mb=571776,
|
|
serial_number='S0M3EY3Z',
|
|
firmware_version='LS0B',
|
|
status='ok',
|
|
raid_status='ready')
|
|
|
|
mock_requests.post(
|
|
'https://1.2.3.4:443/wsman',
|
|
text=test_utils.RAIDEnumerations[uris.DCIM_PhysicalDiskView]['ok'])
|
|
|
|
self.assertIn(expected_physical_disk,
|
|
self.drac_client.list_physical_disks())
|
|
|
|
# Verify that various client convert_physical_disks calls to dracclient
|
|
# result in a WSMan.invoke with appropriate parameters
|
|
def _random_term(self):
|
|
return "".join(random.sample('ABCDEFGHabcdefgh0123456',
|
|
random.randint(4, 12)))
|
|
|
|
def _random_fqdd(self):
|
|
result = self._random_term()
|
|
for i in range(0, random.randint(6, 10)):
|
|
result += random.sample('.:-', 1)[0] + self._random_term()
|
|
return result
|
|
|
|
@mock.patch.object(dracclient.client.WSManClient, 'invoke',
|
|
spec_set=True, autospec=True)
|
|
def test_convert_physical_disks_1(self, mock_requests, mock_invoke):
|
|
'''Convert a single disk to RAID mode'''
|
|
device_fqdd = self._random_fqdd()
|
|
expected_invocation = 'ConvertToRAID'
|
|
expected_selectors = {'SystemCreationClassName': 'DCIM_ComputerSystem',
|
|
'CreationClassName': 'DCIM_RAIDService',
|
|
'SystemName': 'DCIM:ComputerSystem',
|
|
'Name': 'DCIM:RAIDService'}
|
|
expected_properties = {'PDArray': [device_fqdd]}
|
|
|
|
result = self.drac_client.convert_physical_disks(
|
|
raid_controller='controller',
|
|
physical_disks=[device_fqdd],
|
|
raid_enable=True)
|
|
|
|
self.assertEqual({'commit_required': False}, result)
|
|
mock_invoke.assert_called_once_with(
|
|
mock.ANY, uris.DCIM_RAIDService, expected_invocation,
|
|
expected_selectors, expected_properties,
|
|
expected_return_value=utils.RET_SUCCESS)
|
|
|
|
@mock.patch.object(dracclient.client.WSManClient, 'invoke',
|
|
spec_set=True, autospec=True)
|
|
def test_convert_physical_disks_n(self, mock_requests, mock_invoke):
|
|
'''Convert a number of disks to RAID mode'''
|
|
device_list = []
|
|
for i in range(0, random.randint(2, 10)):
|
|
device_list += self._random_fqdd()
|
|
|
|
expected_invocation = 'ConvertToRAID'
|
|
expected_selectors = {'SystemCreationClassName': 'DCIM_ComputerSystem',
|
|
'CreationClassName': 'DCIM_RAIDService',
|
|
'SystemName': 'DCIM:ComputerSystem',
|
|
'Name': 'DCIM:RAIDService'}
|
|
expected_properties = {'PDArray': device_list}
|
|
|
|
result = self.drac_client.convert_physical_disks(
|
|
raid_controller='controller',
|
|
physical_disks=device_list,
|
|
raid_enable=True)
|
|
|
|
self.assertEqual({'commit_required': False}, result)
|
|
mock_invoke.assert_called_once_with(
|
|
mock.ANY, uris.DCIM_RAIDService, expected_invocation,
|
|
expected_selectors, expected_properties,
|
|
expected_return_value=utils.RET_SUCCESS)
|
|
|
|
@mock.patch.object(dracclient.client.WSManClient, 'invoke',
|
|
spec_set=True, autospec=True)
|
|
def test_convert_physical_disks_nonraid_1(self, mock_requests,
|
|
mock_invoke):
|
|
'''Convert a single disk to non-RAID mode'''
|
|
device_fqdd = self._random_fqdd()
|
|
expected_invocation = 'ConvertToNonRAID'
|
|
expected_selectors = {'SystemCreationClassName': 'DCIM_ComputerSystem',
|
|
'CreationClassName': 'DCIM_RAIDService',
|
|
'SystemName': 'DCIM:ComputerSystem',
|
|
'Name': 'DCIM:RAIDService'}
|
|
expected_properties = {'PDArray': [device_fqdd]}
|
|
|
|
result = self.drac_client.convert_physical_disks(
|
|
raid_controller='controller',
|
|
physical_disks=[device_fqdd],
|
|
raid_enable=False)
|
|
|
|
self.assertEqual({'commit_required': False}, result)
|
|
mock_invoke.assert_called_once_with(
|
|
mock.ANY, uris.DCIM_RAIDService, expected_invocation,
|
|
expected_selectors, expected_properties,
|
|
expected_return_value=utils.RET_SUCCESS)
|
|
|
|
@mock.patch.object(dracclient.client.WSManClient, 'invoke',
|
|
spec_set=True, autospec=True)
|
|
def test_convert_physical_disks_nonraid_n(self, mock_requests,
|
|
mock_invoke):
|
|
'''Convert a number of disks to non-RAID mode'''
|
|
device_list = []
|
|
for i in range(0, random.randint(2, 10)):
|
|
device_list += self._random_fqdd()
|
|
|
|
expected_invocation = 'ConvertToNonRAID'
|
|
expected_selectors = {'SystemCreationClassName': 'DCIM_ComputerSystem',
|
|
'CreationClassName': 'DCIM_RAIDService',
|
|
'SystemName': 'DCIM:ComputerSystem',
|
|
'Name': 'DCIM:RAIDService'}
|
|
expected_properties = {'PDArray': device_list}
|
|
|
|
result = self.drac_client.convert_physical_disks(
|
|
raid_controller='controller',
|
|
physical_disks=device_list,
|
|
raid_enable=False)
|
|
|
|
self.assertEqual({'commit_required': False}, result)
|
|
mock_invoke.assert_called_once_with(
|
|
mock.ANY, uris.DCIM_RAIDService, expected_invocation,
|
|
expected_selectors, expected_properties,
|
|
expected_return_value=utils.RET_SUCCESS)
|
|
|
|
@mock.patch.object(dracclient.client.WSManClient, 'invoke',
|
|
spec_set=True, autospec=True)
|
|
def test_convert_physical_disks_ok(self, mock_requests, mock_invoke):
|
|
'''Convert a number of disks to RAID mode and check the return value'''
|
|
device_list = []
|
|
for i in range(0, random.randint(2, 10)):
|
|
device_list += self._random_fqdd()
|
|
|
|
expected_invocation = 'ConvertToRAID'
|
|
expected_selectors = {'SystemCreationClassName': 'DCIM_ComputerSystem',
|
|
'CreationClassName': 'DCIM_RAIDService',
|
|
'SystemName': 'DCIM:ComputerSystem',
|
|
'Name': 'DCIM:RAIDService'}
|
|
expected_properties = {'PDArray': device_list}
|
|
|
|
mock_invoke.return_value = lxml.etree.fromstring(
|
|
test_utils.RAIDInvocations[uris.DCIM_RAIDService][
|
|
expected_invocation]['ok'])
|
|
|
|
result = self.drac_client.convert_physical_disks(
|
|
raid_controller='controller',
|
|
physical_disks=device_list,
|
|
raid_enable=True)
|
|
|
|
self.assertEqual({'commit_required': True}, result)
|
|
mock_invoke.assert_called_once_with(
|
|
mock.ANY, uris.DCIM_RAIDService, expected_invocation,
|
|
expected_selectors, expected_properties,
|
|
expected_return_value=utils.RET_SUCCESS)
|
|
|
|
def test_convert_physical_disks_fail(self, mock_requests):
|
|
mock_requests.post(
|
|
'https://1.2.3.4:443/wsman',
|
|
text=test_utils.RAIDInvocations[
|
|
uris.DCIM_RAIDService]['ConvertToRAID']['error'])
|
|
|
|
self.assertRaises(
|
|
exceptions.DRACOperationFailed,
|
|
self.drac_client.convert_physical_disks,
|
|
raid_controller='controller',
|
|
physical_disks=['Disk0:Enclosure-1:RAID-1',
|
|
'Disk1:Enclosure-1:RAID-1'],
|
|
raid_enable=True)
|
|
|
|
@mock.patch.object(dracclient.client.WSManClient, 'invoke',
|
|
spec_set=True, autospec=True)
|
|
def test_create_virtual_disk(self, mock_requests, mock_invoke):
|
|
expected_selectors = {'SystemCreationClassName': 'DCIM_ComputerSystem',
|
|
'CreationClassName': 'DCIM_RAIDService',
|
|
'SystemName': 'DCIM:ComputerSystem',
|
|
'Name': 'DCIM:RAIDService'}
|
|
expected_properties = {'Target': 'controller',
|
|
'PDArray': ['disk1', 'disk2'],
|
|
'VDPropNameArray': ['Size', 'RAIDLevel'],
|
|
'VDPropValueArray': ['42', '4']}
|
|
mock_invoke.return_value = lxml.etree.fromstring(
|
|
test_utils.RAIDInvocations[uris.DCIM_RAIDService][
|
|
'CreateVirtualDisk']['ok'])
|
|
|
|
result = self.drac_client.create_virtual_disk(
|
|
raid_controller='controller', physical_disks=['disk1', 'disk2'],
|
|
raid_level='1', size_mb=42)
|
|
|
|
self.assertEqual({'commit_required': True}, result)
|
|
mock_invoke.assert_called_once_with(
|
|
mock.ANY, uris.DCIM_RAIDService, 'CreateVirtualDisk',
|
|
expected_selectors, expected_properties,
|
|
expected_return_value=utils.RET_SUCCESS)
|
|
|
|
@mock.patch.object(dracclient.client.WSManClient, 'invoke',
|
|
spec_set=True, autospec=True)
|
|
def test_create_virtual_disk_with_extra_params(self, mock_requests,
|
|
mock_invoke):
|
|
expected_selectors = {'SystemCreationClassName': 'DCIM_ComputerSystem',
|
|
'CreationClassName': 'DCIM_RAIDService',
|
|
'SystemName': 'DCIM:ComputerSystem',
|
|
'Name': 'DCIM:RAIDService'}
|
|
expected_properties = {'Target': 'controller',
|
|
'PDArray': ['disk1', 'disk2'],
|
|
'VDPropNameArray': ['Size', 'RAIDLevel',
|
|
'VirtualDiskName',
|
|
'SpanDepth', 'SpanLength'],
|
|
'VDPropValueArray': ['42', '4', 'name', '3',
|
|
'2']}
|
|
mock_invoke.return_value = lxml.etree.fromstring(
|
|
test_utils.RAIDInvocations[uris.DCIM_RAIDService][
|
|
'CreateVirtualDisk']['ok'])
|
|
|
|
result = self.drac_client.create_virtual_disk(
|
|
raid_controller='controller', physical_disks=['disk1', 'disk2'],
|
|
raid_level='1', size_mb=42, disk_name='name', span_length=2,
|
|
span_depth=3)
|
|
|
|
self.assertEqual({'commit_required': True}, result)
|
|
mock_invoke.assert_called_once_with(
|
|
mock.ANY, uris.DCIM_RAIDService, 'CreateVirtualDisk',
|
|
expected_selectors, expected_properties,
|
|
expected_return_value=utils.RET_SUCCESS)
|
|
|
|
def test_create_virtual_disk_fail(self, mock_requests):
|
|
mock_requests.post(
|
|
'https://1.2.3.4:443/wsman',
|
|
text=test_utils.RAIDInvocations[
|
|
uris.DCIM_RAIDService]['CreateVirtualDisk']['error'])
|
|
|
|
self.assertRaises(
|
|
exceptions.DRACOperationFailed,
|
|
self.drac_client.create_virtual_disk, raid_controller='controller',
|
|
physical_disks=['disk1', 'disk2'], raid_level='1', size_mb=42,
|
|
disk_name='name', span_length=2, span_depth=3)
|
|
|
|
def test_create_virtual_disk_missing_controller(self, mock_requests):
|
|
self.assertRaises(
|
|
exceptions.InvalidParameterValue,
|
|
self.drac_client.create_virtual_disk, raid_controller=None,
|
|
physical_disks=['disk1', 'disk2'], raid_level='1', size_mb=42,
|
|
disk_name='name', span_length=2, span_depth=3)
|
|
|
|
def test_create_virtual_disk_missing_physical_disks(self, mock_requests):
|
|
self.assertRaises(
|
|
exceptions.InvalidParameterValue,
|
|
self.drac_client.create_virtual_disk, raid_controller='controller',
|
|
physical_disks=None, raid_level='1', size_mb=42,
|
|
disk_name='name', span_length=2, span_depth=3)
|
|
|
|
def test_create_virtual_disk_missing_raid_level(self, mock_requests):
|
|
self.assertRaises(
|
|
exceptions.InvalidParameterValue,
|
|
self.drac_client.create_virtual_disk, raid_controller='controller',
|
|
physical_disks=['disk1', 'disk2'], raid_level=None, size_mb=42,
|
|
disk_name='name', span_length=2, span_depth=3)
|
|
|
|
def test_create_virtual_disk_invalid_raid_level(self, mock_requests):
|
|
self.assertRaises(
|
|
exceptions.InvalidParameterValue,
|
|
self.drac_client.create_virtual_disk, raid_controller='controller',
|
|
physical_disks=['disk1', 'disk2'], raid_level='foo', size_mb=42,
|
|
disk_name='name', span_length=2, span_depth=3)
|
|
|
|
def test_create_virtual_disk_missing_size(self, mock_requests):
|
|
self.assertRaises(
|
|
exceptions.InvalidParameterValue,
|
|
self.drac_client.create_virtual_disk, raid_controller='controller',
|
|
physical_disks=['disk1', 'disk2'], raid_level='1', size_mb=None,
|
|
disk_name='name', span_length=2, span_depth=3)
|
|
|
|
def test_create_virtual_disk_invalid_size(self, mock_requests):
|
|
self.assertRaises(
|
|
exceptions.InvalidParameterValue,
|
|
self.drac_client.create_virtual_disk, raid_controller='controller',
|
|
physical_disks=['disk1', 'disk2'], raid_level='1',
|
|
size_mb='foo', disk_name='name', span_length=2, span_depth=3)
|
|
|
|
def test_create_virtual_disk_invalid_span_length(self, mock_requests):
|
|
self.assertRaises(
|
|
exceptions.InvalidParameterValue,
|
|
self.drac_client.create_virtual_disk, raid_controller='controller',
|
|
physical_disks=['disk1', 'disk2'], raid_level='1', size_mb=42,
|
|
disk_name='name', span_length='foo', span_depth=3)
|
|
|
|
def test_create_virtual_disk_invalid_span_depth(self, mock_requests):
|
|
self.assertRaises(
|
|
exceptions.InvalidParameterValue,
|
|
self.drac_client.create_virtual_disk, raid_controller='controller',
|
|
physical_disks=['disk1', 'disk2'], raid_level='1', size_mb=42,
|
|
disk_name='name', span_length=2, span_depth='foo')
|
|
|
|
@mock.patch.object(dracclient.client.WSManClient, 'invoke',
|
|
spec_set=True, autospec=True)
|
|
def test_delete_virtual_disk(self, mock_requests, mock_invoke):
|
|
expected_selectors = {'SystemCreationClassName': 'DCIM_ComputerSystem',
|
|
'CreationClassName': 'DCIM_RAIDService',
|
|
'SystemName': 'DCIM:ComputerSystem',
|
|
'Name': 'DCIM:RAIDService'}
|
|
expected_properties = {'Target': 'disk1'}
|
|
mock_invoke.return_value = lxml.etree.fromstring(
|
|
test_utils.RAIDInvocations[uris.DCIM_RAIDService][
|
|
'DeleteVirtualDisk']['ok'])
|
|
|
|
result = self.drac_client.delete_virtual_disk('disk1')
|
|
|
|
self.assertEqual({'commit_required': True}, result)
|
|
mock_invoke.assert_called_once_with(
|
|
mock.ANY, uris.DCIM_RAIDService, 'DeleteVirtualDisk',
|
|
expected_selectors, expected_properties,
|
|
expected_return_value=utils.RET_SUCCESS)
|
|
|
|
def test_delete_virtual_disk_fail(self, mock_requests):
|
|
mock_requests.post(
|
|
'https://1.2.3.4:443/wsman',
|
|
text=test_utils.RAIDInvocations[
|
|
uris.DCIM_RAIDService]['DeleteVirtualDisk']['error'])
|
|
|
|
self.assertRaises(
|
|
exceptions.DRACOperationFailed,
|
|
self.drac_client.delete_virtual_disk, 'disk1')
|
|
|
|
@mock.patch.object(dracclient.resources.job.JobManagement,
|
|
'create_config_job', spec_set=True, autospec=True)
|
|
def test_commit_pending_raid_changes(self, mock_requests,
|
|
mock_create_config_job):
|
|
self.drac_client.commit_pending_raid_changes('controller')
|
|
|
|
mock_create_config_job.assert_called_once_with(
|
|
mock.ANY, resource_uri=uris.DCIM_RAIDService,
|
|
cim_creation_class_name='DCIM_RAIDService',
|
|
cim_name='DCIM:RAIDService', target='controller', reboot=False)
|
|
|
|
@mock.patch.object(dracclient.resources.job.JobManagement,
|
|
'create_config_job', spec_set=True, autospec=True)
|
|
def test_commit_pending_raid_changes_with_reboot(self, mock_requests,
|
|
mock_create_config_job):
|
|
self.drac_client.commit_pending_raid_changes('controller', reboot=True)
|
|
|
|
mock_create_config_job.assert_called_once_with(
|
|
mock.ANY, resource_uri=uris.DCIM_RAIDService,
|
|
cim_creation_class_name='DCIM_RAIDService',
|
|
cim_name='DCIM:RAIDService', target='controller', reboot=True)
|
|
|
|
@mock.patch.object(dracclient.resources.job.JobManagement,
|
|
'delete_pending_config', spec_set=True, autospec=True)
|
|
def test_abandon_pending_raid_changes(self, mock_requests,
|
|
mock_delete_pending_config):
|
|
self.drac_client.abandon_pending_raid_changes('controller')
|
|
|
|
mock_delete_pending_config.assert_called_once_with(
|
|
mock.ANY, resource_uri=uris.DCIM_RAIDService,
|
|
cim_creation_class_name='DCIM_RAIDService',
|
|
cim_name='DCIM:RAIDService', target='controller')
|