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_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 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_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 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_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_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")
