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 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_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_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 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 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 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 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(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_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_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_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'})
