def test_opeators(self): self.assertEqual(subject, "1:0:0:1") self.assertFalse(subject == 123) self.assertRaises(TypeError, subject.__lt__, None) self.assertGreaterEqual(subject, HCTL.from_string("1:0:0:0")) self.assertGreater(subject, HCTL.from_string("1:0:0:0")) self.assertEqual([i for i in subject], [1, 0, 0, 1]) self.assertRaises(ValueError, subject.from_string, None) self.assertEqual(repr(subject), "<1:0:0:1>") self.assertEqual(HCT(1, 0, 0)[1], subject) self.assertEqual(HCT(1, 0, 0), HCT(1, 0, 0))
def _append_device_by_type(self, hctl_str, dev_path, scsi_type): if scsi_type == SCSI_TYPE_STORAGE_CONTROLLER: self.controllers.append(SysfsSCSIDevice(dev_path, HCTL.from_string(hctl_str))) elif scsi_type == SCSI_TYPE_DISK: block_dev_names = get_sd_paths(dev_path) if block_dev_names == []: self.sg_disks.append(SysfsSCSIDevice(dev_path, HCTL.from_string(hctl_str))) else: sd_disk = SysfsSDDisk(dev_path, HCTL.from_string(hctl_str), block_dev_names) self.sd_disks.append(sd_disk) self.sg_disks.append(sd_disk) self.block_devices.append(sd_disk) self.block_devno_to_device[sd_disk.get_block_devno()] = sd_disk
def _append_device_by_type(self, hctl_str, dev_path, scsi_type): if scsi_type == SCSI_TYPE_STORAGE_CONTROLLER: self.controllers.append(SysfsSCSIDevice(dev_path, HCTL.from_string(hctl_str))) elif scsi_type == SCSI_TYPE_ENCLOSURE: self.enclosures.append(SysfsEnclosureDevice(dev_path, HCTL.from_string(hctl_str))) elif scsi_type == SCSI_TYPE_DISK: block_dev_names = self._sd_structures.get(hctl_str) if not block_dev_names: self.sg_disks.append(SysfsSCSIDevice(dev_path, HCTL.from_string(hctl_str))) else: sd_disk = SysfsSDDisk(dev_path, HCTL.from_string(hctl_str), block_dev_names) self.sd_disks.append(sd_disk) self.sg_disks.append(sd_disk) self.block_devices.append(sd_disk) self.block_devno_to_device[sd_disk.get_block_devno()] = sd_disk
def _append_device_by_type(self, hctl_str, dev_path, scsi_type): if scsi_type == SCSI_TYPE_STORAGE_CONTROLLER: self.controllers.append( SysfsSCSIDevice(dev_path, HCTL.from_string(hctl_str))) elif scsi_type == SCSI_TYPE_ENCLOSURE: self.enclosures.append( SysfsEnclosureDevice(dev_path, HCTL.from_string(hctl_str))) elif scsi_type == SCSI_TYPE_DISK: block_dev_names = self._sd_structures.get(hctl_str) if not block_dev_names: self.sg_disks.append( SysfsSCSIDevice(dev_path, HCTL.from_string(hctl_str))) else: sd_disk = SysfsSDDisk(dev_path, HCTL.from_string(hctl_str), block_dev_names) self.sd_disks.append(sd_disk) self.sg_disks.append(sd_disk) self.block_devices.append(sd_disk) self.block_devno_to_device[sd_disk.get_block_devno()] = sd_disk
def find_hctl_by_slot(self, slot): dev_path = os.path.join(self._basepath, slot, 'device') if os.path.exists(dev_path): hctl = os.path.basename(os.readlink(dev_path)) return HCTL.from_string(hctl) return None
def test_sysfs(self, open_mock, listdir_mock, exists_mock, islink_mock, get_hctl_for_sd_device_mock): if name == "nt": raise SkipTest listdir_map = { '/sys/class/scsi_device': [ '2:0:0:0', '3:0:0:0', '3:0:1:1', '3:0:1:2', '4:0:0:0', '4:0:1:1', '4:0:1:2', '5:0:0:0' ], # SCSI Disks: '/sys/class/scsi_device/2:0:0:0/device/block': [ 'sda' ], '/sys/class/scsi_device/2:0:0:0/device/scsi_generic': [ 'sg0' ], '/sys/class/scsi_device/3:0:0:0/device/block': [ 'sde' ], '/sys/class/scsi_device/3:0:0:0/device/scsi_generic': [ 'sg4' ], '/sys/class/scsi_device/3:0:1:1/device/block': [ 'sdf' ], '/sys/class/scsi_device/3:0:1:1/device/scsi_generic': [ 'sg2' ], '/sys/class/scsi_device/3:0:1:2/device/block': [ 'sdg' ], '/sys/class/scsi_device/3:0:1:2/device/scsi_generic': [ 'sg5' ], '/sys/class/scsi_device/4:0:0:0/device/block': [ 'sdb' ], '/sys/class/scsi_device/4:0:0:0/device/scsi_generic': [ 'sg1' ], '/sys/class/scsi_device/4:0:1:1/device/block': [ 'sdc' ], '/sys/class/scsi_device/4:0:1:1/device/scsi_generic': [ 'sg7' ], '/sys/class/scsi_device/4:0:1:2/device/block': [ 'sdd' ], '/sys/class/scsi_device/4:0:1:2/device/scsi_generic': [ 'sg6' ], # SCSI Storage Controllers: '/sys/class/scsi_device/5:0:0:0/device/scsi_generic': [ 'sg3' ], # Block Devices: '/sys/block': ['sda','sde','sdf','sdg','sdb','sdc','sdd'], # Device names: '/dev' : ['sg0','sg4','sg2','sg5','sg1','sg7','sg6','sg3','sda','sde','sdf','sdg','sdb','sdc','sdd'] } file_map = { '/sys/class/scsi_device/2:0:0:0/device/type': '0', '/sys/class/scsi_device/2:0:0:0/device/vendor': 'VMware', '/sys/class/scsi_device/2:0:0:0/device/block/sda/size': '16777216', '/sys/class/scsi_device/2:0:0:0/device/block/sda/dev': '8:0', '/sys/class/scsi_device/2:0:0:0/device/queue_depth': '64', '/sys/class/scsi_device/3:0:0:0/device/type': '0', '/sys/class/scsi_device/3:0:0:0/device/vendor': 'NFINIDAT', '/sys/class/scsi_device/3:0:0:0/device/block/sde/size': '2097156', '/sys/class/scsi_device/3:0:0:0/device/block/sde/dev': '8:4', '/sys/class/scsi_device/3:0:0:0/device/queue_depth': '32', '/sys/class/scsi_device/3:0:1:1/device/type': '0', '/sys/class/scsi_device/3:0:1:1/device/vendor': 'NEXSAN', '/sys/class/scsi_device/3:0:1:1/device/block/sdf/size': '1953792', '/sys/class/scsi_device/3:0:1:1/device/block/sdf/dev': '8:5', '/sys/class/scsi_device/3:0:1:1/device/queue_depth': '32', '/sys/class/scsi_device/3:0:1:2/device/type': '0', '/sys/class/scsi_device/3:0:1:2/device/vendor': 'NEXSAN', '/sys/class/scsi_device/3:0:1:2/device/block/sdg/size': '1953792', '/sys/class/scsi_device/3:0:1:2/device/block/sdg/dev': '8:6', '/sys/class/scsi_device/3:0:1:2/device/queue_depth': '32', '/sys/class/scsi_device/4:0:0:0/device/type': '0', '/sys/class/scsi_device/4:0:0:0/device/vendor': 'NFINIDAT', '/sys/class/scsi_device/4:0:0:0/device/block/sdb/size': '2097156', '/sys/class/scsi_device/4:0:0:0/device/block/sdb/dev': '8:1', '/sys/class/scsi_device/4:0:0:0/device/queue_depth': '32', '/sys/class/scsi_device/4:0:1:1/device/type': '0', '/sys/class/scsi_device/4:0:1:1/device/vendor': 'NEXSAN', '/sys/class/scsi_device/4:0:1:1/device/block/sdc/size': '1953792', '/sys/class/scsi_device/4:0:1:1/device/block/sdc/dev': '8:2', '/sys/class/scsi_device/4:0:1:1/device/queue_depth': '32', '/sys/class/scsi_device/4:0:1:2/device/type': '0', '/sys/class/scsi_device/4:0:1:2/device/vendor': 'NEXSAN', '/sys/class/scsi_device/4:0:1:2/device/block/sdd/size': '1953792', '/sys/class/scsi_device/4:0:1:2/device/block/sdd/dev': '8:3', '/sys/class/scsi_device/4:0:1:2/device/queue_depth': '32', '/sys/class/scsi_device/5:0:0:0/device/type': '12', '/sys/class/scsi_device/5:0:0:0/device/vendor': 'NFINIDAT', '/sys/class/scsi_device/5:0:0:0/device/queue_depth': '32', '/sys/block/sda/dev' : '8:0', '/sys/block/sda/size' : '16777216', '/sys/block/sde/dev' : '8:4', '/sys/block/sde/size' : '2097156', '/sys/block/sdf/dev' : '8:5', '/sys/block/sdf/size' : '1953792', '/sys/block/sdg/dev' : '8:6', '/sys/block/sdg/size' : '1953792', '/sys/block/sdb/dev' : '8:1', '/sys/block/sdb/size' : '2097156', '/sys/block/sdc/dev' : '8:2', '/sys/block/sdc/size' : '1953792', '/sys/block/sdd/dev' : '8:3', '/sys/block/sdd/size' : '1953792', } get_sd_hctl_map = { '/dev/sda' : HCTL.from_string('2:0:0:0'), '/dev/sde' : HCTL.from_string('3:0:0:0'), '/dev/sdf' : HCTL.from_string('3:0:1:1'), '/dev/sdg' : HCTL.from_string('3:0:1:2'), '/dev/sdb' : HCTL.from_string('4:0:0:0'), '/dev/sdc' : HCTL.from_string('4:0:1:1'), '/dev/sdd' : HCTL.from_string('4:0:1:2'), } def create_file_context_manager(*args, **kwargs): path = args[0] file_mock = Mock() file_mock.read = Mock(return_value=file_map[path]) cm = Mock() cm.__enter__ = Mock(return_value=file_mock) cm.__exit__ = Mock() return cm disk_properties = { 'sda': dict(queue_depth=64, sysfs_size=16777216, hctl='2:0:0:0', vendor='VMware'), 'sde': dict(queue_depth=32, sysfs_size=2097156, hctl='3:0:0:0', vendor='NFINIDAT'), 'sdf': dict(queue_depth=32, sysfs_size=1953792, hctl='3:0:1:1', vendor='NEXSAN'), 'sdg': dict(queue_depth=32, sysfs_size=1953792, hctl='3:0:1:2', vendor='NEXSAN'), 'sdb': dict(queue_depth=32, sysfs_size=2097156, hctl='4:0:0:0', vendor='NFINIDAT'), 'sdc': dict(queue_depth=32, sysfs_size=1953792, hctl='4:0:1:1', vendor='NEXSAN'), 'sdd': dict(queue_depth=32, sysfs_size=1953792, hctl='4:0:1:2', vendor='NEXSAN'), } listdir_mock.side_effect = listdir_map.get open_mock.side_effect = create_file_context_manager exists_mock.return_value = True islink_mock.return_value = False get_hctl_for_sd_device_mock.side_effect = get_sd_hctl_map.get sysfs = Sysfs() disks = sysfs.get_all_sd_disks() self.assertEquals(7, len(disks)) for disk in disks: block_dev = disk.get_block_device_name() self.assertTrue(block_dev in disk_properties) self.assertEquals(HCTL.from_string(disk_properties[block_dev]['hctl']), disk.get_hctl()) self.assertEquals(disk_properties[block_dev]['queue_depth'], disk.get_queue_depth()) self.assertEquals(disk_properties[block_dev]['sysfs_size'] * 512, disk.get_size_in_bytes()) self.assertEquals(disk_properties[block_dev]['vendor'], disk.get_vendor())
def test_sysfs(self, open_mock, listdir_mock, exists_mock, islink_mock, get_hctl_for_sd_device_mock): if name == "nt": raise SkipTest listdir_map = { '/sys/class/scsi_device': [ '2:0:0:0', '3:0:0:0', '3:0:1:1', '3:0:1:2', '4:0:0:0', '4:0:1:1', '4:0:1:2', '5:0:0:0' ], # SCSI Disks: '/sys/class/scsi_device/2:0:0:0/device/block': ['sda'], '/sys/class/scsi_device/2:0:0:0/device/scsi_generic': ['sg0'], '/sys/class/scsi_device/3:0:0:0/device/block': ['sde'], '/sys/class/scsi_device/3:0:0:0/device/scsi_generic': ['sg4'], '/sys/class/scsi_device/3:0:1:1/device/block': ['sdf'], '/sys/class/scsi_device/3:0:1:1/device/scsi_generic': ['sg2'], '/sys/class/scsi_device/3:0:1:2/device/block': ['sdg'], '/sys/class/scsi_device/3:0:1:2/device/scsi_generic': ['sg5'], '/sys/class/scsi_device/4:0:0:0/device/block': ['sdb'], '/sys/class/scsi_device/4:0:0:0/device/scsi_generic': ['sg1'], '/sys/class/scsi_device/4:0:1:1/device/block': ['sdc'], '/sys/class/scsi_device/4:0:1:1/device/scsi_generic': ['sg7'], '/sys/class/scsi_device/4:0:1:2/device/block': ['sdd'], '/sys/class/scsi_device/4:0:1:2/device/scsi_generic': ['sg6'], # SCSI Storage Controllers: '/sys/class/scsi_device/5:0:0:0/device/scsi_generic': ['sg3'], # Block Devices: '/sys/block': ['sda', 'sde', 'sdf', 'sdg', 'sdb', 'sdc', 'sdd'], # Device names: '/dev': [ 'sg0', 'sg4', 'sg2', 'sg5', 'sg1', 'sg7', 'sg6', 'sg3', 'sda', 'sde', 'sdf', 'sdg', 'sdb', 'sdc', 'sdd' ] } file_map = { '/sys/class/scsi_device/2:0:0:0/device/type': '0', '/sys/class/scsi_device/2:0:0:0/device/vendor': 'VMware', '/sys/class/scsi_device/2:0:0:0/device/block/sda/size': '16777216', '/sys/class/scsi_device/2:0:0:0/device/block/sda/dev': '8:0', '/sys/class/scsi_device/2:0:0:0/device/queue_depth': '64', '/sys/class/scsi_device/3:0:0:0/device/type': '0', '/sys/class/scsi_device/3:0:0:0/device/vendor': 'NFINIDAT', '/sys/class/scsi_device/3:0:0:0/device/block/sde/size': '2097156', '/sys/class/scsi_device/3:0:0:0/device/block/sde/dev': '8:4', '/sys/class/scsi_device/3:0:0:0/device/queue_depth': '32', '/sys/class/scsi_device/3:0:1:1/device/type': '0', '/sys/class/scsi_device/3:0:1:1/device/vendor': 'NEXSAN', '/sys/class/scsi_device/3:0:1:1/device/block/sdf/size': '1953792', '/sys/class/scsi_device/3:0:1:1/device/block/sdf/dev': '8:5', '/sys/class/scsi_device/3:0:1:1/device/queue_depth': '32', '/sys/class/scsi_device/3:0:1:2/device/type': '0', '/sys/class/scsi_device/3:0:1:2/device/vendor': 'NEXSAN', '/sys/class/scsi_device/3:0:1:2/device/block/sdg/size': '1953792', '/sys/class/scsi_device/3:0:1:2/device/block/sdg/dev': '8:6', '/sys/class/scsi_device/3:0:1:2/device/queue_depth': '32', '/sys/class/scsi_device/4:0:0:0/device/type': '0', '/sys/class/scsi_device/4:0:0:0/device/vendor': 'NFINIDAT', '/sys/class/scsi_device/4:0:0:0/device/block/sdb/size': '2097156', '/sys/class/scsi_device/4:0:0:0/device/block/sdb/dev': '8:1', '/sys/class/scsi_device/4:0:0:0/device/queue_depth': '32', '/sys/class/scsi_device/4:0:1:1/device/type': '0', '/sys/class/scsi_device/4:0:1:1/device/vendor': 'NEXSAN', '/sys/class/scsi_device/4:0:1:1/device/block/sdc/size': '1953792', '/sys/class/scsi_device/4:0:1:1/device/block/sdc/dev': '8:2', '/sys/class/scsi_device/4:0:1:1/device/queue_depth': '32', '/sys/class/scsi_device/4:0:1:2/device/type': '0', '/sys/class/scsi_device/4:0:1:2/device/vendor': 'NEXSAN', '/sys/class/scsi_device/4:0:1:2/device/block/sdd/size': '1953792', '/sys/class/scsi_device/4:0:1:2/device/block/sdd/dev': '8:3', '/sys/class/scsi_device/4:0:1:2/device/queue_depth': '32', '/sys/class/scsi_device/5:0:0:0/device/type': '12', '/sys/class/scsi_device/5:0:0:0/device/vendor': 'NFINIDAT', '/sys/class/scsi_device/5:0:0:0/device/queue_depth': '32', '/sys/block/sda/dev': '8:0', '/sys/block/sda/size': '16777216', '/sys/block/sde/dev': '8:4', '/sys/block/sde/size': '2097156', '/sys/block/sdf/dev': '8:5', '/sys/block/sdf/size': '1953792', '/sys/block/sdg/dev': '8:6', '/sys/block/sdg/size': '1953792', '/sys/block/sdb/dev': '8:1', '/sys/block/sdb/size': '2097156', '/sys/block/sdc/dev': '8:2', '/sys/block/sdc/size': '1953792', '/sys/block/sdd/dev': '8:3', '/sys/block/sdd/size': '1953792', } get_sd_hctl_map = { '/dev/sda': HCTL.from_string('2:0:0:0'), '/dev/sde': HCTL.from_string('3:0:0:0'), '/dev/sdf': HCTL.from_string('3:0:1:1'), '/dev/sdg': HCTL.from_string('3:0:1:2'), '/dev/sdb': HCTL.from_string('4:0:0:0'), '/dev/sdc': HCTL.from_string('4:0:1:1'), '/dev/sdd': HCTL.from_string('4:0:1:2'), } def create_file_context_manager(*args, **kwargs): path = args[0] file_mock = Mock() file_mock.read = Mock(return_value=file_map[path]) cm = Mock() cm.__enter__ = Mock(return_value=file_mock) cm.__exit__ = Mock() return cm disk_properties = { 'sda': dict(queue_depth=64, sysfs_size=16777216, hctl='2:0:0:0', vendor='VMware'), 'sde': dict(queue_depth=32, sysfs_size=2097156, hctl='3:0:0:0', vendor='NFINIDAT'), 'sdf': dict(queue_depth=32, sysfs_size=1953792, hctl='3:0:1:1', vendor='NEXSAN'), 'sdg': dict(queue_depth=32, sysfs_size=1953792, hctl='3:0:1:2', vendor='NEXSAN'), 'sdb': dict(queue_depth=32, sysfs_size=2097156, hctl='4:0:0:0', vendor='NFINIDAT'), 'sdc': dict(queue_depth=32, sysfs_size=1953792, hctl='4:0:1:1', vendor='NEXSAN'), 'sdd': dict(queue_depth=32, sysfs_size=1953792, hctl='4:0:1:2', vendor='NEXSAN'), } listdir_mock.side_effect = listdir_map.get open_mock.side_effect = create_file_context_manager exists_mock.return_value = True islink_mock.return_value = False get_hctl_for_sd_device_mock.side_effect = get_sd_hctl_map.get sysfs = Sysfs() disks = sysfs.get_all_sd_disks() self.assertEqual(7, len(disks)) for disk in disks: block_dev = disk.get_block_device_name() self.assertTrue(block_dev in disk_properties) self.assertEqual( HCTL.from_string(disk_properties[block_dev]['hctl']), disk.get_hctl()) self.assertEqual(disk_properties[block_dev]['queue_depth'], disk.get_queue_depth()) self.assertEqual(disk_properties[block_dev]['sysfs_size'] * 512, disk.get_size_in_bytes()) self.assertEqual(disk_properties[block_dev]['vendor'], disk.get_vendor())
def test_fromstring(self): self.assertEqual(subject, HCTL.from_string("1:0:0:1")) self.assertNotEqual(subject, HCTL.from_string("1:0:0:2")) self.assertLessEqual(subject, HCTL.from_string("1:0:0:2"))
def test_sysfs(self, open_mock, listdir_mock, exists_mock, islink_mock): if name == "nt": raise SkipTest listdir_map = { "/sys/class/scsi_device": [ "2:0:0:0", "3:0:0:0", "3:0:1:1", "3:0:1:2", "4:0:0:0", "4:0:1:1", "4:0:1:2", "5:0:0:0", ], # SCSI Disks: "/sys/class/scsi_device/2:0:0:0/device/block": ["sda"], "/sys/class/scsi_device/2:0:0:0/device/scsi_generic": ["sg0"], "/sys/class/scsi_device/3:0:0:0/device/block": ["sde"], "/sys/class/scsi_device/3:0:0:0/device/scsi_generic": ["sg4"], "/sys/class/scsi_device/3:0:1:1/device/block": ["sdf"], "/sys/class/scsi_device/3:0:1:1/device/scsi_generic": ["sg2"], "/sys/class/scsi_device/3:0:1:2/device/block": ["sdg"], "/sys/class/scsi_device/3:0:1:2/device/scsi_generic": ["sg5"], "/sys/class/scsi_device/4:0:0:0/device/block": ["sdb"], "/sys/class/scsi_device/4:0:0:0/device/scsi_generic": ["sg1"], "/sys/class/scsi_device/4:0:1:1/device/block": ["sdc"], "/sys/class/scsi_device/4:0:1:1/device/scsi_generic": ["sg7"], "/sys/class/scsi_device/4:0:1:2/device/block": ["sdd"], "/sys/class/scsi_device/4:0:1:2/device/scsi_generic": ["sg6"], # SCSI Storage Controllers: "/sys/class/scsi_device/5:0:0:0/device/scsi_generic": ["sg3"], # Block Devices: "/sys/block": [], } file_map = { "/sys/class/scsi_device/2:0:0:0/device/type": "0", "/sys/class/scsi_device/2:0:0:0/device/vendor": "VMware", "/sys/class/scsi_device/2:0:0:0/device/block/sda/size": "16777216", "/sys/class/scsi_device/2:0:0:0/device/block/sda/dev": "8:0", "/sys/class/scsi_device/2:0:0:0/device/queue_depth": "64", "/sys/class/scsi_device/3:0:0:0/device/type": "0", "/sys/class/scsi_device/3:0:0:0/device/vendor": "NFINIDAT", "/sys/class/scsi_device/3:0:0:0/device/block/sde/size": "2097156", "/sys/class/scsi_device/3:0:0:0/device/block/sde/dev": "8:4", "/sys/class/scsi_device/3:0:0:0/device/queue_depth": "32", "/sys/class/scsi_device/3:0:1:1/device/type": "0", "/sys/class/scsi_device/3:0:1:1/device/vendor": "NEXSAN", "/sys/class/scsi_device/3:0:1:1/device/block/sdf/size": "1953792", "/sys/class/scsi_device/3:0:1:1/device/block/sdf/dev": "8:5", "/sys/class/scsi_device/3:0:1:1/device/queue_depth": "32", "/sys/class/scsi_device/3:0:1:2/device/type": "0", "/sys/class/scsi_device/3:0:1:2/device/vendor": "NEXSAN", "/sys/class/scsi_device/3:0:1:2/device/block/sdg/size": "1953792", "/sys/class/scsi_device/3:0:1:2/device/block/sdg/dev": "8:6", "/sys/class/scsi_device/3:0:1:2/device/queue_depth": "32", "/sys/class/scsi_device/4:0:0:0/device/type": "0", "/sys/class/scsi_device/4:0:0:0/device/vendor": "NFINIDAT", "/sys/class/scsi_device/4:0:0:0/device/block/sdb/size": "2097156", "/sys/class/scsi_device/4:0:0:0/device/block/sdb/dev": "8:1", "/sys/class/scsi_device/4:0:0:0/device/queue_depth": "32", "/sys/class/scsi_device/4:0:1:1/device/type": "0", "/sys/class/scsi_device/4:0:1:1/device/vendor": "NEXSAN", "/sys/class/scsi_device/4:0:1:1/device/block/sdc/size": "1953792", "/sys/class/scsi_device/4:0:1:1/device/block/sdc/dev": "8:2", "/sys/class/scsi_device/4:0:1:1/device/queue_depth": "32", "/sys/class/scsi_device/4:0:1:2/device/type": "0", "/sys/class/scsi_device/4:0:1:2/device/vendor": "NEXSAN", "/sys/class/scsi_device/4:0:1:2/device/block/sdd/size": "1953792", "/sys/class/scsi_device/4:0:1:2/device/block/sdd/dev": "8:3", "/sys/class/scsi_device/4:0:1:2/device/queue_depth": "32", "/sys/class/scsi_device/5:0:0:0/device/type": "12", "/sys/class/scsi_device/5:0:0:0/device/vendor": "NFINIDAT", "/sys/class/scsi_device/5:0:0:0/device/queue_depth": "32", } def create_file_context_manager(*args, **kwargs): path = args[0] file_mock = Mock() file_mock.read = Mock(return_value=file_map[path]) cm = Mock() cm.__enter__ = Mock(return_value=file_mock) cm.__exit__ = Mock() return cm disk_properties = { "sda": dict(queue_depth=64, sysfs_size=16777216, hctl="2:0:0:0", vendor="VMware"), "sde": dict(queue_depth=32, sysfs_size=2097156, hctl="3:0:0:0", vendor="NFINIDAT"), "sdf": dict(queue_depth=32, sysfs_size=1953792, hctl="3:0:1:1", vendor="NEXSAN"), "sdg": dict(queue_depth=32, sysfs_size=1953792, hctl="3:0:1:2", vendor="NEXSAN"), "sdb": dict(queue_depth=32, sysfs_size=2097156, hctl="4:0:0:0", vendor="NFINIDAT"), "sdc": dict(queue_depth=32, sysfs_size=1953792, hctl="4:0:1:1", vendor="NEXSAN"), "sdd": dict(queue_depth=32, sysfs_size=1953792, hctl="4:0:1:2", vendor="NEXSAN"), } listdir_mock.side_effect = listdir_map.get open_mock.side_effect = create_file_context_manager exists_mock.return_value = True islink_mock.return_value = False sysfs = Sysfs() disks = sysfs.get_all_sd_disks() self.assertEquals(7, len(disks)) for disk in disks: block_dev = disk.get_block_device_name() self.assertTrue(block_dev in disk_properties) self.assertEquals(HCTL.from_string(disk_properties[block_dev]["hctl"]), disk.get_hctl()) self.assertEquals(disk_properties[block_dev]["queue_depth"], disk.get_queue_depth()) self.assertEquals(disk_properties[block_dev]["sysfs_size"] * 512, disk.get_size_in_bytes()) self.assertEquals(disk_properties[block_dev]["vendor"], disk.get_vendor())