def get_single_host(self):
host_list = []
host = VmHost()
host.set_id('host-01')
host.set_name('host-01')
host.add_virtualMachineIds('vm-01')
host.add_virtualMachineIds('vm-02')
host.add_storageVolumeIds('storage-01')
host.add_storageVolumeIds('storage-02')
host_list.append(host)
return host_list
def get_single_host(self):
host_list = []
host = VmHost()
host.set_id('host-01')
host.set_name('host-01')
host.add_virtualMachineIds('vm-01')
host.add_virtualMachineIds('vm-02')
host.add_storageVolumeIds('storage-01')
host.add_storageVolumeIds('storage-02')
host_list.append(host)
return host_list
def get_limited_list(self, num):
host_list = []
for i in range(1, num + 1):
host = VmHost()
host.set_id('host-' + str(i))
host.set_name('host-' + str(i))
host.add_virtualMachineIds('vm-' + str(i))
host.add_virtualMachineIds('vm-' + str(i))
host.add_storageVolumeIds('storage-' + str(i))
host.add_storageVolumeIds('storage-' + str(i))
host_list.append(host)
return host_list
def get_limited_list(self, num):
host_list = []
for i in range(1, num + 1):
host = VmHost()
host.set_id('host-' + str(i))
host.set_name('host-' + str(i))
host.add_virtualMachineIds('vm-' + str(i))
host.add_virtualMachineIds('vm-' + str(i))
host.add_storageVolumeIds('storage-' + str(i))
host.add_storageVolumeIds('storage-' + str(i))
host_list.append(host)
return host_list
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)
