def testStorageVolumePayloadGenerator(self): metadata = \ event_metadata.get_EventMetaData( event_metadata.EVENT_TYPE_STORAGE_ADDED) obj = StorageVolume() obj.name = 'TestStorageVolume' obj.connectionState = 'ACTIVE' obj.size = 200 obj.volumeType = 'DIR' obj.volumeId = 'TestVolumeId' obj.createEpoch = long(time.time() * 1000) obj.lastModifiedEpoch = long(time.time() * 1000) mount_point = HostMountPoint() mount_point.set_path('/root/storage/1') mount_point.set_vmHostId('HOST1') obj.add_mountPoints(mount_point) mount_point = HostMountPoint() mount_point.set_path('/root/storage/2') mount_point.set_vmHostId('HOST2') obj.add_mountPoints(mount_point) payload = payload_generator.generate_payload(metadata, obj) self.assertEquals(payload['entity_type'], obj.__class__.__name__) self.assertEquals(payload['name'], obj.name) self.assertEquals(payload['state'], obj.connectionState) self.assertTrue(obj.mountPoints[0].path in payload['mount_points'])
def test_vm_host_get_all_for_sv(self): host_id = 'VH1' vmhost = VmHost() vmhost.id = host_id healthnmon_db_api.vm_host_save(get_admin_context(), vmhost) mntPnt = HostMountPoint() mntPnt.set_vmHostId(host_id) mntPnt.set_path('/path') volume = StorageVolume() volume.set_id('SV11') volume.add_mountPoints(mntPnt) healthnmon_db_api.storage_volume_save(get_admin_context(), volume) vmhosts = \ healthnmon_db_api.vm_host_get_all(get_admin_context()) self.assertFalse(vmhosts is None, 'Host get by id returned a none list') self.assertTrue( len(vmhosts) > 0, 'Host get by id returned invalid number of list') self.assertTrue(vmhosts[0].id == host_id) svlist = vmhosts[0].get_storageVolumeIds() self.assert_(svlist is not None) self.assert_(len(svlist) == 1) self.assert_(volume.get_id() in svlist) healthnmon_db_api.storage_volume_delete_by_ids(get_admin_context(), [volume.get_id()]) vmhosts = \ healthnmon_db_api.vm_host_get_all(get_admin_context()) self.assertTrue(vmhosts[0].id == host_id) svids = vmhosts[0].get_storageVolumeIds() self.assert_((svids is None) or (len(svids) == 0))
def test_diff_resourcemodel_storagevolume_withupdate(self): """Unit Test to test for resource model comparison with some change """ storagevolume = StorageVolume() storagevolume.set_id('datastore-112') storagevolume.set_name('datastore-112') storagevolume.set_resourceManagerId( '13274325-BFD6-464F-A9D1-61332573B5E2' ) storagevolume.set_size(107105746944) storagevolume.set_free(32256294912) storagevolume.set_vmfsVolume(False) storagevolume.set_shared(False) storagevolume.set_assignedServerCount(1) storagevolume.set_volumeType('VMFS') storagevolume.set_volumeId( '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db' ) hostMountPoint = \ HostMountPoint( '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db', 'host-9') storagevolume.add_mountPoints(hostMountPoint) storagevolume1 = StorageVolume() storagevolume1.set_id('datastore-112') storagevolume1.set_name('datastore-112') storagevolume1.set_resourceManagerId( '13274325-BFD6-464F-A9D1-61332573B5E2' ) storagevolume1.set_size(107105746944) storagevolume1.set_free(32256294912) storagevolume1.set_vmfsVolume(False) storagevolume1.set_shared(False) storagevolume1.set_assignedServerCount(1) storagevolume1.set_volumeType('VMFS') storagevolume1.set_volumeId( '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db' ) hostMountPoint1 = \ HostMountPoint( '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db', 'host-19') storagevolume1.add_mountPoints(hostMountPoint1) diff = ResourceModelDiff(storagevolume, storagevolume1) diff_res = diff.diff_resourcemodel() self.assertTrue(len(diff_res) > 0) self.assertTrue(self.update in diff_res) mountPoints = 'mountPoints' vmHostId = 'vmHostId' self.assertTrue(mountPoints in diff_res[self.update]) self.assertTrue(self.update in diff_res[self.update][mountPoints]) key = diff_res[self.update][mountPoints][self.update].keys()[0] self.assertTrue(self.update in diff_res[self.update][mountPoints][ self.update][key]) self.assertTrue(vmHostId in diff_res[self.update][mountPoints][ self.update][key][self.update]) self.assertEquals(diff_res[self.update][mountPoints][self.update][ key][self.update][vmHostId], 'host-19')
def test_vm_host_get_by_id(self): host_id = 'VH1' vmhost = VmHost() vmhost.id = host_id healthnmon_db_api.vm_host_save(get_admin_context(), vmhost) vm = Vm() vm.id = 'VM11' vm.set_vmHostId(host_id) healthnmon_db_api.vm_save(get_admin_context(), vm) mntPnt = HostMountPoint() mntPnt.set_vmHostId(host_id) mntPnt.set_path('/path') volume = StorageVolume() volume.set_id('SV11') volume.add_mountPoints(mntPnt) healthnmon_db_api.storage_volume_save(get_admin_context(), volume) vmhosts = \ healthnmon_db_api.vm_host_get_by_ids(get_admin_context(), [host_id]) self.assertFalse(vmhosts is None, 'Host get by id returned a none list') self.assertTrue(len(vmhosts) > 0, 'Host get by id returned invalid number of list' ) self.assertTrue(vmhosts[0].id == host_id)
def test_vm_host_get_all_for_sv(self): host_id = 'VH1' vmhost = VmHost() vmhost.id = host_id healthnmon_db_api.vm_host_save(get_admin_context(), vmhost) mntPnt = HostMountPoint() mntPnt.set_vmHostId(host_id) mntPnt.set_path('/path') volume = StorageVolume() volume.set_id('SV11') volume.add_mountPoints(mntPnt) healthnmon_db_api.storage_volume_save(get_admin_context(), volume) vmhosts = \ healthnmon_db_api.vm_host_get_all(get_admin_context()) self.assertFalse(vmhosts is None, 'Host get by id returned a none list') self.assertTrue(len(vmhosts) > 0, 'Host get by id returned invalid number of list' ) self.assertTrue(vmhosts[0].id == host_id) svlist = vmhosts[0].get_storageVolumeIds() self.assert_(svlist is not None) self.assert_(len(svlist) == 1) self.assert_(volume.get_id() in svlist) healthnmon_db_api.storage_volume_delete_by_ids( get_admin_context(), [volume.get_id()]) vmhosts = \ healthnmon_db_api.vm_host_get_all(get_admin_context()) self.assertTrue(vmhosts[0].id == host_id) svids = vmhosts[0].get_storageVolumeIds() self.assert_((svids is None) or (len(svids) == 0))
def test_vm_host_get_by_id(self): host_id = 'VH1' vmhost = VmHost() vmhost.id = host_id healthnmon_db_api.vm_host_save(get_admin_context(), vmhost) vm = Vm() vm.id = 'VM11' vm.set_vmHostId(host_id) healthnmon_db_api.vm_save(get_admin_context(), vm) mntPnt = HostMountPoint() mntPnt.set_vmHostId(host_id) mntPnt.set_path('/path') volume = StorageVolume() volume.set_id('SV11') volume.add_mountPoints(mntPnt) healthnmon_db_api.storage_volume_save(get_admin_context(), volume) vmhosts = \ healthnmon_db_api.vm_host_get_by_ids(get_admin_context(), [host_id]) self.assertFalse(vmhosts is None, 'Host get by id returned a none list') self.assertTrue( len(vmhosts) > 0, 'Host get by id returned invalid number of list') self.assertTrue(vmhosts[0].id == host_id)
def test_diff_resourcemodel_storagevolume_withdelete(self): """Unit Test to test for resource model comparison with some change """ storagevolume = StorageVolume() storagevolume.set_id('datastore-112') storagevolume.set_name('datastore-112') storagevolume.set_resourceManagerId( '13274325-BFD6-464F-A9D1-61332573B5E2') storagevolume.set_size(107105746944) storagevolume.set_free(32256294912) storagevolume.set_vmfsVolume(False) storagevolume.set_shared(False) storagevolume.set_assignedServerCount(1) storagevolume.set_volumeType('VMFS') storagevolume.set_volumeId( '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db') hostMountPoint = \ HostMountPoint( '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db', 'host-9') storagevolume.add_mountPoints(hostMountPoint) storagevolume1 = StorageVolume() storagevolume1.set_id('datastore-114') storagevolume1.set_name('datastore-114') storagevolume1.set_resourceManagerId( '13274325-BFD6-464F-A9D1-61332573B5E2') storagevolume1.set_size(107105746944) storagevolume1.set_free(32256294912) storagevolume1.set_vmfsVolume(False) storagevolume1.set_shared(False) storagevolume1.set_assignedServerCount(1) storagevolume1.set_volumeType('VMFS') storagevolume1.set_volumeId( '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db') diff = ResourceModelDiff(storagevolume, storagevolume1) diff_res = diff.diff_resourcemodel() self.assertTrue(len(diff_res) > 0) self.assertTrue(self.update in diff_res) mountPoints = 'mountPoints' self.assertTrue(mountPoints in diff_res[self.update]) self.assertTrue(self.delete in diff_res[self.update][mountPoints]) key = diff_res[self.update][mountPoints][self.delete].keys()[0] self.assertTrue( isinstance(diff_res[self.update][mountPoints][self.delete][key], HostMountPoint)) delMount = diff_res[self.update][mountPoints][self.delete][key] self.assertEquals(delMount.vmHostId, 'host-9') self.assertEquals(delMount.path, '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db') self.assertTrue('id' in diff_res[self.update]) self.assertEquals(diff_res[self.update]['id'], 'datastore-114') self.assertTrue('name' in diff_res[self.update]) self.assertEquals(diff_res[self.update]['name'], 'datastore-114')
def test_diff_dict_emptyDict(self): storagevolume = StorageVolume() hostMountPoint = HostMountPoint({}) storagevolume.add_mountPoints(hostMountPoint) storagevolume1 = StorageVolume() hostMountPoint1 = HostMountPoint({}) storagevolume1.add_mountPoints(hostMountPoint1) diff = ResourceModelDiff(storagevolume, storagevolume1) diff_res = diff.diff_resourcemodel() self.assertTrue(len(diff_res) == 0)
def test_diff_resourcemodel_storagevolume_nochange(self): """Unit Test to test for resource model comparison with no change """ storagevolume = StorageVolume() storagevolume.set_id('datastore-112') storagevolume.set_name('datastore-112') storagevolume.set_resourceManagerId( '13274325-BFD6-464F-A9D1-61332573B5E2' ) storagevolume.set_size(107105746944) storagevolume.set_free(32256294912) storagevolume.set_vmfsVolume(False) storagevolume.set_shared(False) storagevolume.set_assignedServerCount(1) storagevolume.set_volumeType('VMFS') storagevolume.set_volumeId( '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db' ) hostMountPoint = \ HostMountPoint( '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db', 'host-9') storagevolume.add_mountPoints(hostMountPoint) storagevolume1 = StorageVolume() storagevolume1.set_id('datastore-112') storagevolume1.set_name('datastore-112') storagevolume1.set_resourceManagerId( '13274325-BFD6-464F-A9D1-61332573B5E2' ) storagevolume1.set_size(107105746944) storagevolume1.set_free(32256294912) storagevolume1.set_vmfsVolume(False) storagevolume1.set_shared(False) storagevolume1.set_assignedServerCount(1) storagevolume1.set_volumeType('VMFS') storagevolume1.set_volumeId( '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db' ) hostMountPoint1 = \ HostMountPoint( '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db', 'host-9') storagevolume1.add_mountPoints(hostMountPoint1) diff = ResourceModelDiff(storagevolume, storagevolume1) diff_res = diff.diff_resourcemodel() self.assertTrue(len(diff_res) == 0)
def get_storagevolume_list(self, storageId=None): storagevolume_dict = {} storagevolume_list = [] storagevolume = StorageVolume() storagevolume.set_id('datastore-111') storagevolume.set_name('datastore-111') storagevolume.set_resourceManagerId( '13274325-BFD6-464F-A9D1-61332573B5E2') storagevolume.set_size(107105746944) storagevolume.set_free(32256294912) storagevolume.set_vmfsVolume(True) storagevolume.set_shared(True) storagevolume.set_assignedServerCount(2) storagevolume.set_volumeType('VMFS') storagevolume.set_volumeId( '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5da') hostMountPoint = \ HostMountPoint( '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5da', 'host-9') storagevolume.add_mountPoints(hostMountPoint) storagevolume_list.append(storagevolume) storagevolume_dict[storagevolume.get_id()] = storagevolume storagevolume = StorageVolume() storagevolume.set_id('datastore-112') storagevolume.set_name('datastore-112') storagevolume.set_resourceManagerId( '13274325-BFD6-464F-A9D1-61332573B5E2') storagevolume.set_size(107105746944) storagevolume.set_free(32256294912) storagevolume.set_vmfsVolume(False) storagevolume.set_shared(False) storagevolume.set_assignedServerCount(1) storagevolume.set_volumeType('VMFS') storagevolume.set_volumeId( '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db') hostMountPoint = \ HostMountPoint( '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db', 'host-9') storagevolume.add_mountPoints(hostMountPoint) storagevolume_list.append(storagevolume) storagevolume_dict[storagevolume.get_id()] = storagevolume if storageId is not None: return [storagevolume_dict[storageId]] return storagevolume_list
def test_diff_resourcemodel_storagevolume_nochange(self): """Unit Test to test for resource model comparison with no change """ storagevolume = StorageVolume() storagevolume.set_id('datastore-112') storagevolume.set_name('datastore-112') storagevolume.set_resourceManagerId( '13274325-BFD6-464F-A9D1-61332573B5E2') storagevolume.set_size(107105746944) storagevolume.set_free(32256294912) storagevolume.set_vmfsVolume(False) storagevolume.set_shared(False) storagevolume.set_assignedServerCount(1) storagevolume.set_volumeType('VMFS') storagevolume.set_volumeId( '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db') hostMountPoint = \ HostMountPoint( '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db', 'host-9') storagevolume.add_mountPoints(hostMountPoint) storagevolume1 = StorageVolume() storagevolume1.set_id('datastore-112') storagevolume1.set_name('datastore-112') storagevolume1.set_resourceManagerId( '13274325-BFD6-464F-A9D1-61332573B5E2') storagevolume1.set_size(107105746944) storagevolume1.set_free(32256294912) storagevolume1.set_vmfsVolume(False) storagevolume1.set_shared(False) storagevolume1.set_assignedServerCount(1) storagevolume1.set_volumeType('VMFS') storagevolume1.set_volumeId( '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db') hostMountPoint1 = \ HostMountPoint( '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db', 'host-9') storagevolume1.add_mountPoints(hostMountPoint1) diff = ResourceModelDiff(storagevolume, storagevolume1) diff_res = diff.diff_resourcemodel() self.assertTrue(len(diff_res) == 0)
def testVmHostPayload_with_storage_size(self): self.flags(instances_path="/var/lib/nova/instances") metadata = \ event_metadata.get_EventMetaData( event_metadata.EVENT_TYPE_HOST_ADDED) obj = VmHost() obj.name = 'TestVmHost' ipProfile = IpProfile() ipProfile.ipAddress = '10.10.10.1' obj.add_ipAddresses(ipProfile) storage_obj = StorageVolume() storage_obj.id = "storage_id" storage_obj.name = 'TestStorageVolume' storage_obj.connectionState = 'ACTIVE' storage_obj.size = 200 storage_obj.free = 100 storage_obj.volumeType = 'DIR' storage_obj.volumeId = 'TestVolumeId' storage_obj.createEpoch = long(time.time() * 1000) storage_obj.lastModifiedEpoch = long(time.time() * 1000) mount_point = HostMountPoint() mount_point.set_path('/var/lib/nova/instances') mount_point.set_vmHostId('TestVmHost') storage_obj.add_mountPoints(mount_point) obj.add_storageVolumeIds("storage_id") self.mox.StubOutWithMock( InventoryCacheManager, 'get_object_from_cache') InventoryCacheManager.get_object_from_cache( storage_obj.id, Constants.StorageVolume).AndReturn(storage_obj) self.mox.ReplayAll() payload = payload_generator.generate_payload(metadata, obj) self.assertEquals(payload['entity_type'], obj.__class__.__name__) self.assertEquals(payload['name'], obj.name) self.assertEquals(payload['ipAddresses'], ipProfile.ipAddress) self.assertEquals(payload['totalStorageSize'], storage_obj.size) self.assertEquals(payload['storageUsed'], storage_obj.free)
def testVmHostPayload_with_storage_size(self): self.flags(instances_path="/var/lib/nova/instances") metadata = \ event_metadata.get_EventMetaData( event_metadata.EVENT_TYPE_HOST_ADDED) obj = VmHost() obj.name = 'TestVmHost' ipProfile = IpProfile() ipProfile.ipAddress = '10.10.10.1' obj.add_ipAddresses(ipProfile) storage_obj = StorageVolume() storage_obj.id = "storage_id" storage_obj.name = 'TestStorageVolume' storage_obj.connectionState = 'ACTIVE' storage_obj.size = 200 storage_obj.free = 100 storage_obj.volumeType = 'DIR' storage_obj.volumeId = 'TestVolumeId' storage_obj.createEpoch = long(time.time() * 1000) storage_obj.lastModifiedEpoch = long(time.time() * 1000) mount_point = HostMountPoint() mount_point.set_path('/var/lib/nova/instances') mount_point.set_vmHostId('TestVmHost') storage_obj.add_mountPoints(mount_point) obj.add_storageVolumeIds("storage_id") self.mox.StubOutWithMock(InventoryCacheManager, 'get_object_from_cache') InventoryCacheManager.get_object_from_cache( storage_obj.id, Constants.StorageVolume).AndReturn(storage_obj) self.mox.ReplayAll() payload = payload_generator.generate_payload(metadata, obj) self.assertEquals(payload['entity_type'], obj.__class__.__name__) self.assertEquals(payload['name'], obj.name) self.assertEquals(payload['ipAddresses'], ipProfile.ipAddress) self.assertEquals(payload['totalStorageSize'], storage_obj.size) self.assertEquals(payload['storageUsed'], storage_obj.free)
def test_vm_host_delete(self): vmhost_id = 'VH1' vmhost = VmHost() vmhost.id = vmhost_id vSwitch = VirtualSwitch() vSwitch.set_id('vSwitch-01') vSwitch.set_name('vSwitch-01') vSwitch.set_resourceManagerId('rmId') vSwitch.set_switchType('vSwitch') cost1 = Cost() cost1.set_value(100) cost1.set_units('USD') vSwitch.set_cost(cost1) portGroup = PortGroup() portGroup.set_id('pg-01') portGroup.set_name('pg-01') portGroup.set_resourceManagerId('rmId') portGroup.set_type('portgroup_type') portGroup.set_cost(cost1) vSwitch.add_portGroups(portGroup) vmhost.add_virtualSwitches(vSwitch) vmhost.add_portGroups(portGroup) healthnmon_db_api.vm_host_save(get_admin_context(), vmhost) vmhost2 = VmHost() vmhost2.set_id('VH2') healthnmon_db_api.vm_host_save(get_admin_context(), vmhost2) storage = StorageVolume() storage.set_id('sv-01') storage.set_name('storage-01') storage.set_resourceManagerId('rmId') storage.set_size(1234) storage.set_free(2345) storage.set_vmfsVolume(True) storage.set_shared(True) storage.set_assignedServerCount(1) storage.set_volumeType('VMFS') storage.set_volumeId('101') hostMount1 = HostMountPoint() hostMount1.set_path('test_path1') hostMount1.set_vmHostId('VH1') storage.add_mountPoints(hostMount1) hostMount2 = HostMountPoint() hostMount2.set_path('test_path2') hostMount2.set_vmHostId('VH2') storage.add_mountPoints(hostMount2) healthnmon_db_api.storage_volume_save(get_admin_context(), storage) vm = Vm() vm.set_id('vm-01') vm.set_name('vm-01') vm.set_vmHostId('VH1') healthnmon_db_api.vm_save(get_admin_context(), vm) vmhosts = \ healthnmon_db_api.vm_host_get_by_ids(get_admin_context(), [vmhost_id]) self.assertFalse(vmhosts is None, 'host get by id returned a none list') self.assertTrue(len(vmhosts) > 0, 'host get by id returned invalid number of list' ) healthnmon_db_api.vm_host_delete_by_ids(get_admin_context(), [vmhost_id]) vmhosts = \ healthnmon_db_api.vm_host_get_by_ids(get_admin_context(), [vmhost_id]) self.assertTrue(vmhosts is None or len(vmhosts) == 0, 'host not deleted')
def test_diff_differentDict(self): storagevolume = StorageVolume() storagevolume.set_id('datastore-112') storagevolume.set_name('datastore-112') storagevolume.set_resourceManagerId( '13274325-BFD6-464F-A9D1-61332573B5E2') storagevolume.set_size(107105746944) storagevolume.set_free(32256294912) storagevolume.set_vmfsVolume(False) storagevolume.set_shared(False) storagevolume.set_assignedServerCount(1) storagevolume.set_volumeType('VMFS') storagevolume.set_volumeId( '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db') hostMountPoint = \ HostMountPoint( {'1': '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db', '2': 'host-9'}) storagevolume.add_mountPoints(hostMountPoint) storagevolume1 = StorageVolume() storagevolume1.set_id('datastore-112') storagevolume1.set_name('datastore-112') storagevolume1.set_resourceManagerId( '13274325-BFD6-464F-A9D1-61332573B5E2') storagevolume1.set_size(107105746944) storagevolume1.set_free(32256294912) storagevolume1.set_vmfsVolume(False) storagevolume1.set_shared(False) storagevolume1.set_assignedServerCount(1) storagevolume1.set_volumeType('VMFS') storagevolume1.set_volumeId( '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db') hostMountPoint1 = \ HostMountPoint( {'3': '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db', '4': 'host-19'}) storagevolume1.add_mountPoints(hostMountPoint1) diff = ResourceModelDiff(storagevolume, storagevolume1) diff_res = diff.diff_resourcemodel() self.assertTrue(len(diff_res) > 0) self.assertTrue(self.update in diff_res) mountPoints = 'mountPoints' self.assertTrue(mountPoints in diff_res[self.update]) self.assertTrue(self.update in diff_res[self.update][mountPoints]) key = diff_res[self.update][mountPoints][self.update].keys()[0] self.assertTrue(self.update in diff_res[self.update][mountPoints][ self.update][key]) self.assertTrue('path' in diff_res[self.update][mountPoints][ self.update][key][self.update]) self.assertTrue(self.add in diff_res[self.update][mountPoints][ self.update][key][self.update]['path']) self.assertTrue('3' in diff_res[self.update][mountPoints][self.update] [key][self.update]['path'][self.add]) self.assertEquals( diff_res[self.update][mountPoints][self.update][key][ self.update]['path'][self.add]['3'], '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db') self.assertTrue('4' in diff_res[self.update][mountPoints][self.update] [key][self.update]['path'][self.add]) self.assertEquals( diff_res[self.update][mountPoints][self.update][key][self.update] ['path'][self.add]['4'], 'host-19') self.assertTrue(self.delete in diff_res[self.update][mountPoints][ self.update][key][self.update]['path']) self.assertTrue('1' in diff_res[self.update][mountPoints][self.update] [key][self.update]['path'][self.delete]) self.assertEquals( diff_res[self.update][mountPoints][self.update][key][ self.update]['path'][self.delete]['1'], '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db') self.assertTrue('2' in diff_res[self.update][mountPoints][self.update] [key][self.update]['path'][self.delete]) self.assertEquals( diff_res[self.update][mountPoints][self.update][key][self.update] ['path'][self.delete]['2'], 'host-9')
def test_vm_host_get_all(self): ''' Inserts more than one host with vms and storage volumes. Also validates the data retrieved from the vmhost, vm, storage volumes. ''' vmhost = VmHost() vmhost.id = 'VH1-id' healthnmon_db_api.vm_host_save(get_admin_context(), vmhost) vmhost = VmHost() vmhost.id = 'VH2-id' healthnmon_db_api.vm_host_save(get_admin_context(), vmhost) vm = Vm() vm.id = 'VM1-id' vm.set_vmHostId('VH1-id') vmGlobalSettings = VmGlobalSettings() vmGlobalSettings.set_id(vm.id) vmGlobalSettings.set_autoStartAction('autoStartAction') vmGlobalSettings.set_autoStopAction('autoStopAction') vm.set_vmGlobalSettings(vmGlobalSettings) healthnmon_db_api.vm_save(get_admin_context(), vm) mntPnt = HostMountPoint() mntPnt.set_vmHostId('VH1-id') mntPnt.set_path('/path') volume = StorageVolume() sv_id = 'SV1-id' volume.set_id(sv_id) volume.add_mountPoints(mntPnt) healthnmon_db_api.storage_volume_save(get_admin_context(), volume) vmhosts = healthnmon_db_api.vm_host_get_all(get_admin_context()) self.assertFalse(vmhosts is None, 'vm_host_get_all returned a None') self.assertTrue( len(vmhosts) == 2, 'vm_host_get_all does not returned expected number of hosts') self.assertEqual(vmhosts[0].get_id(), 'VH1-id', "VMHost id is not same") self.assertEqual(vmhosts[1].get_id(), 'VH2-id', "VMHost id is not same") vmlist = vmhosts[0].get_virtualMachineIds() self.assertFalse(vmlist is None, "virtual machines from the host returned None") self.assertTrue( len(vmlist) == 1, "length of virtual machines list is not returned as expected") self.assertTrue(vm.id in vmlist, "VmId is not in host") vms = healthnmon_db_api.vm_get_by_ids(get_admin_context(), ['VM1-id']) self.assertTrue(vms is not None) self.assertTrue(len(vms) == 1) vm = vms[0] self.assertEqual(vm.get_id(), 'VM1-id', "VM id is not same") vmGlobalSets = vm.get_vmGlobalSettings() self.assertTrue(vmGlobalSets is not None) self.assertEqual(vmGlobalSets.get_id(), 'VM1-id', "VM id is not same") self.assertEqual(vmGlobalSets.get_autoStartAction(), 'autoStartAction', "autoStartAction is not same") self.assertEqual(vmGlobalSets.get_autoStopAction(), 'autoStopAction', "autoStopAction is not same") svlist = vmhosts[0].get_storageVolumeIds() self.assertFalse(svlist is None, "Storage Volumes from the host returned None") self.assertTrue( len(svlist) >= 1, "length of storage volumes list is not returned as expected") self.assertTrue(sv_id in svlist, "Storage Volume Id is not host") storagevolumes = \ healthnmon_db_api.storage_volume_get_by_ids(get_admin_context(), ['SV1-id']) self.assertFalse(storagevolumes is None, 'Storage volume get by id returned a none list') self.assertTrue( len(storagevolumes) > 0, 'Storage volume get by id returned invalid number of list') self.assertEquals(storagevolumes[0].id, 'SV1-id', "Storage volume id is not same") hostMountPoints = storagevolumes[0].get_mountPoints() self.assertEquals(hostMountPoints[0].get_path(), '/path', "Host mount point path is not same") self.assertEquals( hostMountPoints[0].get_vmHostId(), 'VH1-id', "VmHost id is not same for storage volumes")
def test_diff_differentDict(self): storagevolume = StorageVolume() storagevolume.set_id('datastore-112') storagevolume.set_name('datastore-112') storagevolume.set_resourceManagerId( '13274325-BFD6-464F-A9D1-61332573B5E2' ) storagevolume.set_size(107105746944) storagevolume.set_free(32256294912) storagevolume.set_vmfsVolume(False) storagevolume.set_shared(False) storagevolume.set_assignedServerCount(1) storagevolume.set_volumeType('VMFS') storagevolume.set_volumeId( '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db' ) hostMountPoint = \ HostMountPoint( {'1': '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db', '2': 'host-9'}) storagevolume.add_mountPoints(hostMountPoint) storagevolume1 = StorageVolume() storagevolume1.set_id('datastore-112') storagevolume1.set_name('datastore-112') storagevolume1.set_resourceManagerId( '13274325-BFD6-464F-A9D1-61332573B5E2' ) storagevolume1.set_size(107105746944) storagevolume1.set_free(32256294912) storagevolume1.set_vmfsVolume(False) storagevolume1.set_shared(False) storagevolume1.set_assignedServerCount(1) storagevolume1.set_volumeType('VMFS') storagevolume1.set_volumeId( '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db' ) hostMountPoint1 = \ HostMountPoint( {'3': '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db', '4': 'host-19'}) storagevolume1.add_mountPoints(hostMountPoint1) diff = ResourceModelDiff(storagevolume, storagevolume1) diff_res = diff.diff_resourcemodel() self.assertTrue(len(diff_res) > 0) self.assertTrue(self.update in diff_res) mountPoints = 'mountPoints' self.assertTrue(mountPoints in diff_res[self.update]) self.assertTrue(self.update in diff_res[self.update][mountPoints]) key = diff_res[self.update][mountPoints][self.update].keys()[0] self.assertTrue(self.update in diff_res[self.update][mountPoints][ self.update][key]) self.assertTrue('path' in diff_res[self.update][mountPoints][ self.update][key][self.update]) self.assertTrue(self.add in diff_res[self.update][mountPoints][ self.update][key][self.update]['path']) self.assertTrue('3' in diff_res[self.update][mountPoints][ self.update][key][self.update]['path'][self.add]) self.assertEquals(diff_res[self.update][mountPoints][self.update][ key][self.update]['path'][self.add]['3'], '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db') self.assertTrue('4' in diff_res[self.update][mountPoints][ self.update][key][self.update]['path'][self.add]) self.assertEquals(diff_res[self.update][mountPoints][self.update][ key][self.update]['path'][self.add]['4'], 'host-19') self.assertTrue(self.delete in diff_res[self.update][mountPoints][ self.update][key][self.update]['path']) self.assertTrue('1' in diff_res[self.update][mountPoints][ self.update][key][self.update]['path'][self.delete]) self.assertEquals(diff_res[self.update][mountPoints][self.update][ key][self.update]['path'][self.delete]['1'], '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db') self.assertTrue('2' in diff_res[self.update][mountPoints][ self.update][key][self.update]['path'][self.delete]) self.assertEquals(diff_res[self.update][mountPoints][self.update][ key][self.update]['path'][self.delete]['2'], 'host-9')
def test_diff_resourcemodel_storagevolume_withdelete_withDict(self): """Unit Test to test for resource model comparison with some change """ storagevolume = StorageVolume() storagevolume.set_id('datastore-112') storagevolume.set_name('datastore-112') storagevolume.set_resourceManagerId( '13274325-BFD6-464F-A9D1-61332573B5E2' ) storagevolume.set_size(107105746944) storagevolume.set_free(32256294912) storagevolume.set_vmfsVolume(False) storagevolume.set_shared(False) storagevolume.set_assignedServerCount(1) storagevolume.set_volumeType('VMFS') storagevolume.set_volumeId( '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db' ) hostMountPoint = \ HostMountPoint( {'1': '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db', '2': 'host-9'}) storagevolume.add_mountPoints(hostMountPoint) storagevolume1 = StorageVolume() storagevolume1.set_id('datastore-114') storagevolume1.set_name('datastore-114') storagevolume1.set_resourceManagerId( '13274325-BFD6-464F-A9D1-61332573B5E2' ) storagevolume1.set_size(107105746944) storagevolume1.set_free(32256294912) storagevolume1.set_vmfsVolume(False) storagevolume1.set_shared(False) storagevolume1.set_assignedServerCount(1) storagevolume1.set_volumeType('VMFS') storagevolume1.set_volumeId( {'1': '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db'}) diff = ResourceModelDiff(storagevolume, storagevolume1) diff_res = diff.diff_resourcemodel() self.assertTrue(len(diff_res) > 0) self.assertTrue(self.update in diff_res) mountPoints = 'mountPoints' self.assertTrue(mountPoints in diff_res[self.update]) self.assertTrue(self.delete in diff_res[self.update][mountPoints]) key = diff_res[self.update][mountPoints][self.delete].keys()[0] self.assertTrue(isinstance(diff_res[self.update][mountPoints][ self.delete][key], HostMountPoint)) delMount = diff_res[self.update][mountPoints][self.delete][key] self.assertEquals( delMount.pathProp, {'1': '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db', '2': 'host-9'}) self.assertTrue('id' in diff_res[self.update]) self.assertEquals(diff_res[self.update]['id'], 'datastore-114') self.assertTrue('name' in diff_res[self.update]) self.assertEquals(diff_res[self.update]['name'], 'datastore-114') self.assertTrue('volumeId' in diff_res[self.update]) self.assertEquals( diff_res[self.update]['volumeId'], {'1': '/vmfs/volumes/4e374cf3-328f8064-aa2c-78acc0fcb5db'})
def test_vm_host_delete(self): vmhost_id = 'VH1' vmhost = VmHost() vmhost.id = vmhost_id vSwitch = VirtualSwitch() vSwitch.set_id('vSwitch-01') vSwitch.set_name('vSwitch-01') vSwitch.set_resourceManagerId('rmId') vSwitch.set_switchType('vSwitch') cost1 = Cost() cost1.set_value(100) cost1.set_units('USD') vSwitch.set_cost(cost1) portGroup = PortGroup() portGroup.set_id('pg-01') portGroup.set_name('pg-01') portGroup.set_resourceManagerId('rmId') portGroup.set_type('portgroup_type') portGroup.set_cost(cost1) vSwitch.add_portGroups(portGroup) vmhost.add_virtualSwitches(vSwitch) vmhost.add_portGroups(portGroup) healthnmon_db_api.vm_host_save(get_admin_context(), vmhost) vmhost2 = VmHost() vmhost2.set_id('VH2') healthnmon_db_api.vm_host_save(get_admin_context(), vmhost2) storage = StorageVolume() storage.set_id('sv-01') storage.set_name('storage-01') storage.set_resourceManagerId('rmId') storage.set_size(1234) storage.set_free(2345) storage.set_vmfsVolume(True) storage.set_shared(True) storage.set_assignedServerCount(1) storage.set_volumeType('VMFS') storage.set_volumeId('101') hostMount1 = HostMountPoint() hostMount1.set_path('test_path1') hostMount1.set_vmHostId('VH1') storage.add_mountPoints(hostMount1) hostMount2 = HostMountPoint() hostMount2.set_path('test_path2') hostMount2.set_vmHostId('VH2') storage.add_mountPoints(hostMount2) healthnmon_db_api.storage_volume_save(get_admin_context(), storage) vm = Vm() vm.set_id('vm-01') vm.set_name('vm-01') vm.set_vmHostId('VH1') healthnmon_db_api.vm_save(get_admin_context(), vm) vmhosts = \ healthnmon_db_api.vm_host_get_by_ids(get_admin_context(), [vmhost_id]) self.assertFalse(vmhosts is None, 'host get by id returned a none list') self.assertTrue( len(vmhosts) > 0, 'host get by id returned invalid number of list') # self.assertRaises(Exception, healthnmon_db_api.vm_host_delete_by_ids,([vmhost_id])) healthnmon_db_api.vm_host_delete_by_ids(get_admin_context(), [vmhost_id]) vmhosts = \ healthnmon_db_api.vm_host_get_by_ids(get_admin_context(), [vmhost_id]) self.assertTrue(vmhosts is None or len(vmhosts) == 0, 'host not deleted')
def test_vm_host_get_all(self): ''' Inserts more than one host with vms and storage volumes. Also validates the data retrieved from the vmhost, vm, storage volumes. ''' vmhost = VmHost() vmhost.id = 'VH1-id' healthnmon_db_api.vm_host_save(get_admin_context(), vmhost) vmhost = VmHost() vmhost.id = 'VH2-id' healthnmon_db_api.vm_host_save(get_admin_context(), vmhost) vm = Vm() vm.id = 'VM1-id' vm.set_vmHostId('VH1-id') vmGlobalSettings = VmGlobalSettings() vmGlobalSettings.set_id(vm.id) vmGlobalSettings.set_autoStartAction('autoStartAction') vmGlobalSettings.set_autoStopAction('autoStopAction') vm.set_vmGlobalSettings(vmGlobalSettings) healthnmon_db_api.vm_save(get_admin_context(), vm) mntPnt = HostMountPoint() mntPnt.set_vmHostId('VH1-id') mntPnt.set_path('/path') volume = StorageVolume() sv_id = 'SV1-id' volume.set_id(sv_id) volume.add_mountPoints(mntPnt) healthnmon_db_api.storage_volume_save(get_admin_context(), volume) vmhosts = healthnmon_db_api.vm_host_get_all(get_admin_context()) self.assertFalse(vmhosts is None, 'vm_host_get_all returned a None') self.assertTrue( len(vmhosts) == 2, 'vm_host_get_all does not returned expected number of hosts') self.assertEqual(vmhosts[0].get_id(), 'VH1-id', "VMHost id is not same") self.assertEqual(vmhosts[1].get_id(), 'VH2-id', "VMHost id is not same") vmlist = vmhosts[0].get_virtualMachineIds() self.assertFalse(vmlist is None, "virtual machines from the host returned None") self.assertTrue( len(vmlist) == 1, "length of virtual machines list is not returned as expected") self.assertTrue(vm.id in vmlist, "VmId is not in host") vms = healthnmon_db_api.vm_get_by_ids(get_admin_context(), ['VM1-id']) self.assertTrue(vms is not None) self.assertTrue(len(vms) == 1) vm = vms[0] self.assertEqual(vm.get_id(), 'VM1-id', "VM id is not same") vmGlobalSets = vm.get_vmGlobalSettings() self.assertTrue(vmGlobalSets is not None) self.assertEqual(vmGlobalSets.get_id(), 'VM1-id', "VM id is not same") self.assertEqual(vmGlobalSets.get_autoStartAction(), 'autoStartAction', "autoStartAction is not same") self.assertEqual(vmGlobalSets.get_autoStopAction(), 'autoStopAction', "autoStopAction is not same") svlist = vmhosts[0].get_storageVolumeIds() self.assertFalse(svlist is None, "Storage Volumes from the host returned None") self.assertTrue( len(svlist) >= 1, "length of storage volumes list is not returned as expected") self.assertTrue(sv_id in svlist, "Storage Volume Id is not host") storagevolumes = \ healthnmon_db_api.storage_volume_get_by_ids(get_admin_context(), ['SV1-id']) self.assertFalse(storagevolumes is None, 'Storage volume get by id returned a none list') self.assertTrue( len(storagevolumes) > 0, 'Storage volume get by id returned invalid number of list') self.assertEquals(storagevolumes[0].id, 'SV1-id', "Storage volume id is not same") hostMountPoints = storagevolumes[0].get_mountPoints() self.assertEquals(hostMountPoints[0].get_path(), '/path', "Host mount point path is not same") self.assertEquals(hostMountPoints[0].get_vmHostId(), 'VH1-id', "VmHost id is not same for storage volumes")