def test_vm_host_save_update_with_new_vSwitch(self):
host_id = 'VH1'
vmhost = VmHost()
vmhost.id = host_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)
vSwitch_new = VirtualSwitch()
vSwitch_new.set_id('vSwitch-02')
vSwitch_new.set_name('vSwitch-02')
vSwitch_new.set_resourceManagerId('rmId')
vSwitch_new.set_switchType('vSwitch')
portGroup_new = PortGroup()
portGroup_new.set_id('pg-02')
portGroup_new.set_name('pg-02')
portGroup_new.set_resourceManagerId('rmId')
portGroup_new.set_type('portgroup_type')
vSwitch.add_portGroups(portGroup_new)
vmhost.add_virtualSwitches(vSwitch_new)
vmhost.add_portGroups(portGroup_new)
healthnmon_db_api.vm_host_save(get_admin_context(), vmhost)
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(
len(vmhosts[0].get_virtualSwitches()) > 0,
'Host get by virtual switch returned invalid number of list')
self.assertTrue(
len(vmhosts[0].get_portGroups()) > 0,
'Host get by port group returned invalid number of list')
self.assertTrue(vmhosts[0].id == host_id)
def test_vm_host_save_update_with_new_vSwitch(self):
host_id = 'VH1'
vmhost = VmHost()
vmhost.id = host_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)
vSwitch_new = VirtualSwitch()
vSwitch_new.set_id('vSwitch-02')
vSwitch_new.set_name('vSwitch-02')
vSwitch_new.set_resourceManagerId('rmId')
vSwitch_new.set_switchType('vSwitch')
portGroup_new = PortGroup()
portGroup_new.set_id('pg-02')
portGroup_new.set_name('pg-02')
portGroup_new.set_resourceManagerId('rmId')
portGroup_new.set_type('portgroup_type')
vSwitch.add_portGroups(portGroup_new)
vmhost.add_virtualSwitches(vSwitch_new)
vmhost.add_portGroups(portGroup_new)
healthnmon_db_api.vm_host_save(get_admin_context(), vmhost)
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(
len(vmhosts[0].get_virtualSwitches()) > 0,
'Host get by virtual switch returned invalid number of list')
self.assertTrue(
len(vmhosts[0].get_portGroups()) > 0,
'Host get by port group returned invalid number of list')
self.assertTrue(vmhosts[0].id == host_id)
def test_vmhost_save_modify_delete_with_vSwitch_pGroup(self):
"""Test case for filter deleted virtual switch and port group
1. Create host with 2 virtual switch and port groups.
2. Assert for the above point.
3. Save VmHost by removing one virtualwitch and one port group.
4. Assert for deleted virtual switch and port group.
5. Delete the host.
6. Use filter_by api to assert for deleted host,
virtual switch and port group.
"""
"Test for points 1 and 2"
host_id = 'VH1'
vmhost = VmHost()
vmhost.id = host_id
cost = self._create_cost()
vSwitch1 = self._create_switch(host_id + '_vSwitch-01')
vSwitch1.set_cost(cost)
portGroup1 = self._create_port_group(host_id + '_pGroup-01')
portGroup1.set_cost(cost)
vSwitch1.add_portGroups(portGroup1)
vmhost.add_virtualSwitches(vSwitch1)
vmhost.add_portGroups(portGroup1)
"Add the second vswitch and portgroup"
vSwitch2 = self._create_switch(host_id + '_vSwitch-02')
vSwitch2.set_cost(cost)
portGroup2 = self._create_port_group(host_id + '_pGroup-02')
portGroup2.set_cost(cost)
vSwitch2.add_portGroups(portGroup2)
vmhost.add_virtualSwitches(vSwitch2)
vmhost.add_portGroups(portGroup2)
healthnmon_db_api.vm_host_save(get_admin_context(), vmhost)
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].get_virtualSwitches()) > 0,
'Host get by virtual switch returned invalid number of list')
vss = vmhosts[0].get_virtualSwitches()
vs_ids = []
for vs in vss:
vs_ids.append(vs.get_id())
self.assertTrue(
vSwitch1.get_id() in vs_ids,
"Added virtual switch1 does not appears in the host api")
self.assertTrue(
vSwitch2.get_id() in vs_ids,
"Added virtual switch2 does not appears in the host api")
pgs = vmhosts[0].get_portGroups()
pg_ids = []
for pg in pgs:
pg_ids.append(pg.get_id())
self.assertTrue(portGroup1.get_id(
) in pg_ids, "Added port group1 does not appears in the host api")
self.assertTrue(portGroup2.get_id(
) in pg_ids, "Added port group2 does not appears in the host api")
# Points 3 and 4 - Remove the second vswitch and
# portgroup from the vmhost and the save the vmhost"
vmhost = VmHost()
vmhost.id = host_id
cost = self._create_cost()
vSwitch = self._create_switch(host_id + '_vSwitch-01')
vSwitch.set_cost(cost)
portGroup = self._create_port_group(host_id + '_pGroup-01')
portGroup.set_cost(cost)
vSwitch.add_portGroups(portGroup)
vmhost.add_virtualSwitches(vSwitch)
vmhost.add_portGroups(portGroup)
healthnmon_db_api.vm_host_save(get_admin_context(), vmhost)
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')
vss = vmhosts[0].get_virtualSwitches()
vs_ids = []
for vs in vss:
vs_ids.append(vs.get_id())
self.assertTrue(
vSwitch.get_id() in vs_ids,
"Modified virtual switch1 not appearing in the host api")
self.assertTrue(vSwitch2.get_id() not in vs_ids,
"Deleted virtual switch2 appears in the host api")
pgs = vmhosts[0].get_portGroups()
pg_ids = []
for pg in pgs:
pg_ids.append(pg.get_id())
self.assertTrue(portGroup.get_id(
) in pg_ids, "Modified port group1 not appearing in the host api")
self.assertTrue(portGroup2.get_id(
) not in pg_ids, "Deleted port group2 appears in the host api")
# Points 5 and 6 - Delete the host and
# assert for deletion using filter-by api"
filters = {'id': host_id, 'deleted': 'true'}
healthnmon_db_api.vm_host_delete_by_ids(get_admin_context(), [host_id])
del_vmhosts = healthnmon_db_api.vm_host_get_all_by_filters(
get_admin_context(), filters, 'id', 'asc')
self.assertFalse(
del_vmhosts is None, 'Host get by filters returned a none list')
vss = del_vmhosts[0].get_virtualSwitches()
vs_ids = []
for vs in vss:
vs_ids.append(vs.get_id())
self.assertTrue(
vSwitch1.get_id() in vs_ids,
"Deleted virtual switch1 not appearing in the host filter api")
self.assertTrue(
vSwitch2.get_id() in vs_ids,
"Deleted virtual switch2 not appears in the host filter api")
pgs = del_vmhosts[0].get_portGroups()
pg_ids = []
for pg in pgs:
pg_ids.append(pg.get_id())
self.assertTrue(
portGroup1.get_id() in pg_ids,
"Deleted port group1 not appearing in the host filter api")
self.assertTrue(
portGroup2.get_id() in pg_ids,
"Deleted port group2 not appears in the host filter api")
def test_timestamp_columns(self):
"""
Test the time stamp columns createEpoch,
modifiedEpoch and deletedEpoch
"""
vmhost = VmHost()
vmhost.set_id('VH1')
virSw1 = VirtualSwitch()
virSw1.set_id('VS1_VH1')
portGrp1 = PortGroup()
portGrp1.set_id('PG1_VH1')
virSw1.add_portGroups(portGrp1)
vmhost.add_virtualSwitches(virSw1)
vmhost.add_portGroups(portGrp1)
# Check for createEpoch
epoch_before = utils.get_current_epoch_ms()
healthnmon_db_api.vm_host_save(self.admin_context, vmhost)
epoch_after = utils.get_current_epoch_ms()
vmhost_queried = healthnmon_db_api.vm_host_get_by_ids(
self.admin_context, [vmhost.get_id()])[0]
self.assert_(test_utils.is_timestamp_between(
epoch_before, epoch_after, vmhost_queried.get_createEpoch()))
for virSw in vmhost_queried.get_virtualSwitches():
self.assert_(test_utils.is_timestamp_between(
epoch_before, epoch_after, virSw.get_createEpoch()))
for pg in virSw.get_portGroups():
self.assert_(test_utils.is_timestamp_between(
epoch_before, epoch_after, pg.get_createEpoch()))
# Check for lastModifiedEpoch after modifying host
vmhost_modified = vmhost_queried
test_utils.unset_timestamp_fields(vmhost_modified)
vmhost_modified.set_name('changed_name')
epoch_before = utils.get_current_epoch_ms()
healthnmon_db_api.vm_host_save(self.admin_context, vmhost_modified)
epoch_after = utils.get_current_epoch_ms()
vmhost_queried = healthnmon_db_api.vm_host_get_by_ids(
self.admin_context, [vmhost.get_id()])[0]
self.assert_(vmhost_modified.get_createEpoch(
) == vmhost_queried.get_createEpoch())
self.assert_(test_utils.is_timestamp_between(
epoch_before, epoch_after, vmhost_queried.get_lastModifiedEpoch()))
for virSw in vmhost_queried.get_virtualSwitches():
self.assert_(test_utils.is_timestamp_between(
epoch_before, epoch_after, virSw.get_lastModifiedEpoch()))
for pg in virSw.get_portGroups():
self.assert_(test_utils.is_timestamp_between(
epoch_before, epoch_after, pg.get_lastModifiedEpoch()))
# Check for createdEpoch after adding switch and portgroup to host
vmhost_modified = vmhost_queried
test_utils.unset_timestamp_fields(vmhost_modified)
virSw2 = VirtualSwitch()
virSw2.set_id('VS2_VH1')
portGrp2 = PortGroup()
portGrp2.set_id('PG2_VH1')
virSw2.add_portGroups(portGrp2)
vmhost_modified.add_virtualSwitches(virSw2)
vmhost_modified.add_portGroups(portGrp2)
epoch_before = utils.get_current_epoch_ms()
healthnmon_db_api.vm_host_save(self.admin_context, vmhost_modified)
epoch_after = utils.get_current_epoch_ms()
vmhost_queried = healthnmon_db_api.vm_host_get_by_ids(
self.admin_context, [vmhost.get_id()])[0]
self.assert_(vmhost_modified.get_createEpoch(
) == vmhost_queried.get_createEpoch())
self.assert_(test_utils.is_timestamp_between(
epoch_before, epoch_after, vmhost_queried.get_lastModifiedEpoch()))
for virSw in vmhost_queried.get_virtualSwitches():
if virSw.get_id() == virSw2.get_id():
self.assert_(test_utils.is_timestamp_between(
epoch_before, epoch_after, virSw.get_createEpoch()))
else:
self.assert_(test_utils.is_timestamp_between(
epoch_before, epoch_after, virSw.get_lastModifiedEpoch()))
for pg in virSw.get_portGroups():
if pg.get_id() == portGrp2.get_id():
self.assert_(test_utils.is_timestamp_between(
epoch_before, epoch_after, pg.get_createEpoch()))
else:
self.assert_(test_utils.is_timestamp_between(
epoch_before, epoch_after, pg.get_lastModifiedEpoch()))
# Check for deletedEpoch
epoch_before = utils.get_current_epoch_ms()
healthnmon_db_api.vm_host_delete_by_ids(
self.admin_context, [vmhost_queried.get_id()])
epoch_after = utils.get_current_epoch_ms()
deleted_host = healthnmon_db_api.vm_host_get_all_by_filters(
self.admin_context,
{"id": vmhost_queried.get_id()}, None, None)[0]
self.assertTrue(deleted_host.get_deleted())
self.assert_(test_utils.is_timestamp_between(
epoch_before, epoch_after, deleted_host.get_deletedEpoch()))
deleted_switches = healthnmon_db_api.\
virtual_switch_get_all_by_filters(self.admin_context,
{"id": (virSw1.get_id(),
virSw2.get_id())},
None, None)
for deleted_switch in deleted_switches:
self.assertTrue(deleted_switch.get_deleted())
self.assert_(test_utils.is_timestamp_between(
epoch_before, epoch_after, deleted_switch.get_deletedEpoch()))
for deleted_portgrp in deleted_switch.get_portGroups():
self.assertTrue(deleted_portgrp.get_deleted())
self.assert_(test_utils.is_timestamp_between(
epoch_before, epoch_after,
deleted_portgrp.get_deletedEpoch()))
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_vmhost_save_modify_delete_with_vSwitch_pGroup(self):
"""Test case for filter deleted virtual switch and port group
1. Create host with 2 virtual switch and port groups.
2. Assert for the above point.
3. Save VmHost by removing one virtualwitch and one port group.
4. Assert for deleted virtual switch and port group.
5. Delete the host.
6. Use filter_by api to assert for deleted host, virtual switch and port group.
"""
"Test for points 1 and 2"
host_id = 'VH1'
vmhost = VmHost()
vmhost.id = host_id
cost = self._create_cost()
vSwitch1 = self._create_switch(host_id + '_vSwitch-01')
vSwitch1.set_cost(cost)
portGroup1 = self._create_port_group(host_id + '_pGroup-01')
portGroup1.set_cost(cost)
vSwitch1.add_portGroups(portGroup1)
vmhost.add_virtualSwitches(vSwitch1)
vmhost.add_portGroups(portGroup1)
"Add the second vswitch and portgroup"
vSwitch2 = self._create_switch(host_id + '_vSwitch-02')
vSwitch2.set_cost(cost)
portGroup2 = self._create_port_group(host_id + '_pGroup-02')
portGroup2.set_cost(cost)
vSwitch2.add_portGroups(portGroup2)
vmhost.add_virtualSwitches(vSwitch2)
vmhost.add_portGroups(portGroup2)
healthnmon_db_api.vm_host_save(get_admin_context(), vmhost)
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].get_virtualSwitches()) > 0,
'Host get by virtual switch returned invalid number of list')
vss = vmhosts[0].get_virtualSwitches()
vs_ids = []
for vs in vss:
vs_ids.append(vs.get_id())
self.assertTrue(
vSwitch1.get_id() in vs_ids,
"Added virtual switch1 does not appears in the host api")
self.assertTrue(
vSwitch2.get_id() in vs_ids,
"Added virtual switch2 does not appears in the host api")
pgs = vmhosts[0].get_portGroups()
pg_ids = []
for pg in pgs:
pg_ids.append(pg.get_id())
self.assertTrue(portGroup1.get_id() in pg_ids,
"Added port group1 does not appears in the host api")
self.assertTrue(portGroup2.get_id() in pg_ids,
"Added port group2 does not appears in the host api")
"Points 3 and 4 - Remove the second vswitch and portgroup from the vmhost and the save the vmhost"
vmhost = VmHost()
vmhost.id = host_id
cost = self._create_cost()
vSwitch = self._create_switch(host_id + '_vSwitch-01')
vSwitch.set_cost(cost)
portGroup = self._create_port_group(host_id + '_pGroup-01')
portGroup.set_cost(cost)
vSwitch.add_portGroups(portGroup)
vmhost.add_virtualSwitches(vSwitch)
vmhost.add_portGroups(portGroup)
healthnmon_db_api.vm_host_save(get_admin_context(), vmhost)
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')
vss = vmhosts[0].get_virtualSwitches()
vs_ids = []
for vs in vss:
vs_ids.append(vs.get_id())
self.assertTrue(
vSwitch.get_id() in vs_ids,
"Modified virtual switch1 not appearing in the host api")
self.assertTrue(vSwitch2.get_id() not in vs_ids,
"Deleted virtual switch2 appears in the host api")
pgs = vmhosts[0].get_portGroups()
pg_ids = []
for pg in pgs:
pg_ids.append(pg.get_id())
self.assertTrue(portGroup.get_id() in pg_ids,
"Modified port group1 not appearing in the host api")
self.assertTrue(portGroup2.get_id() not in pg_ids,
"Deleted port group2 appears in the host api")
"Points 5 and 6 - Delete the host and assert for deletion using filter-by api"
filters = {'id': host_id, 'deleted': 'true'}
healthnmon_db_api.vm_host_delete_by_ids(get_admin_context(), [host_id])
del_vmhosts = healthnmon_db_api.vm_host_get_all_by_filters(
get_admin_context(), filters, 'id', 'asc')
self.assertFalse(del_vmhosts is None,
'Host get by filters returned a none list')
vss = del_vmhosts[0].get_virtualSwitches()
vs_ids = []
for vs in vss:
vs_ids.append(vs.get_id())
self.assertTrue(
vSwitch1.get_id() in vs_ids,
"Deleted virtual switch1 not appearing in the host filter api")
self.assertTrue(
vSwitch2.get_id() in vs_ids,
"Deleted virtual switch2 not appears in the host filter api")
pgs = del_vmhosts[0].get_portGroups()
pg_ids = []
for pg in pgs:
pg_ids.append(pg.get_id())
self.assertTrue(
portGroup1.get_id() in pg_ids,
"Deleted port group1 not appearing in the host filter api")
self.assertTrue(
portGroup2.get_id() in pg_ids,
"Deleted port group2 not appears in the host filter api")
def test_timestamp_columns(self):
"""
Test the time stamp columns createEpoch, modifiedEpoch and deletedEpoch
"""
vmhost = VmHost()
vmhost.set_id('VH1')
virSw1 = VirtualSwitch()
virSw1.set_id('VS1_VH1')
portGrp1 = PortGroup()
portGrp1.set_id('PG1_VH1')
virSw1.add_portGroups(portGrp1)
vmhost.add_virtualSwitches(virSw1)
vmhost.add_portGroups(portGrp1)
# Check for createEpoch
epoch_before = utils.get_current_epoch_ms()
healthnmon_db_api.vm_host_save(self.admin_context, vmhost)
epoch_after = utils.get_current_epoch_ms()
vmhost_queried = healthnmon_db_api.vm_host_get_by_ids(
self.admin_context, [vmhost.get_id()])[0]
self.assert_(
test_utils.is_timestamp_between(epoch_before, epoch_after,
vmhost_queried.get_createEpoch()))
for virSw in vmhost_queried.get_virtualSwitches():
self.assert_(
test_utils.is_timestamp_between(epoch_before, epoch_after,
virSw.get_createEpoch()))
for pg in virSw.get_portGroups():
self.assert_(
test_utils.is_timestamp_between(epoch_before, epoch_after,
pg.get_createEpoch()))
# Check for lastModifiedEpoch after modifying host
vmhost_modified = vmhost_queried
test_utils.unset_timestamp_fields(vmhost_modified)
vmhost_modified.set_name('changed_name')
epoch_before = utils.get_current_epoch_ms()
healthnmon_db_api.vm_host_save(self.admin_context, vmhost_modified)
epoch_after = utils.get_current_epoch_ms()
vmhost_queried = healthnmon_db_api.vm_host_get_by_ids(
self.admin_context, [vmhost.get_id()])[0]
self.assert_(vmhost_modified.get_createEpoch() ==
vmhost_queried.get_createEpoch())
self.assert_(
test_utils.is_timestamp_between(
epoch_before, epoch_after,
vmhost_queried.get_lastModifiedEpoch()))
for virSw in vmhost_queried.get_virtualSwitches():
self.assert_(
test_utils.is_timestamp_between(epoch_before, epoch_after,
virSw.get_lastModifiedEpoch()))
for pg in virSw.get_portGroups():
self.assert_(
test_utils.is_timestamp_between(
epoch_before, epoch_after, pg.get_lastModifiedEpoch()))
# Check for createdEpoch after adding switch and portgroup to host
vmhost_modified = vmhost_queried
test_utils.unset_timestamp_fields(vmhost_modified)
virSw2 = VirtualSwitch()
virSw2.set_id('VS2_VH1')
portGrp2 = PortGroup()
portGrp2.set_id('PG2_VH1')
virSw2.add_portGroups(portGrp2)
vmhost_modified.add_virtualSwitches(virSw2)
vmhost_modified.add_portGroups(portGrp2)
epoch_before = utils.get_current_epoch_ms()
healthnmon_db_api.vm_host_save(self.admin_context, vmhost_modified)
epoch_after = utils.get_current_epoch_ms()
vmhost_queried = healthnmon_db_api.vm_host_get_by_ids(
self.admin_context, [vmhost.get_id()])[0]
self.assert_(vmhost_modified.get_createEpoch() ==
vmhost_queried.get_createEpoch())
self.assert_(
test_utils.is_timestamp_between(
epoch_before, epoch_after,
vmhost_queried.get_lastModifiedEpoch()))
for virSw in vmhost_queried.get_virtualSwitches():
if virSw.get_id() == virSw2.get_id():
self.assert_(
test_utils.is_timestamp_between(epoch_before, epoch_after,
virSw.get_createEpoch()))
else:
self.assert_(
test_utils.is_timestamp_between(
epoch_before, epoch_after,
virSw.get_lastModifiedEpoch()))
for pg in virSw.get_portGroups():
if pg.get_id() == portGrp2.get_id():
self.assert_(
test_utils.is_timestamp_between(
epoch_before, epoch_after, pg.get_createEpoch()))
else:
self.assert_(
test_utils.is_timestamp_between(
epoch_before, epoch_after,
pg.get_lastModifiedEpoch()))
# Check for deletedEpoch
epoch_before = utils.get_current_epoch_ms()
healthnmon_db_api.vm_host_delete_by_ids(self.admin_context,
[vmhost_queried.get_id()])
epoch_after = utils.get_current_epoch_ms()
deleted_host = healthnmon_db_api.vm_host_get_all_by_filters(
self.admin_context, {"id": vmhost_queried.get_id()}, None, None)[0]
self.assertTrue(deleted_host.get_deleted())
self.assert_(
test_utils.is_timestamp_between(epoch_before, epoch_after,
deleted_host.get_deletedEpoch()))
deleted_switches = healthnmon_db_api.virtual_switch_get_all_by_filters(
self.admin_context, {"id": (virSw1.get_id(), virSw2.get_id())},
None, None)
for deleted_switch in deleted_switches:
self.assertTrue(deleted_switch.get_deleted())
self.assert_(
test_utils.is_timestamp_between(
epoch_before, epoch_after,
deleted_switch.get_deletedEpoch()))
for deleted_portgrp in deleted_switch.get_portGroups():
self.assertTrue(deleted_portgrp.get_deleted())
self.assert_(
test_utils.is_timestamp_between(
epoch_before, epoch_after,
deleted_portgrp.get_deletedEpoch()))
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')
