def __init__(self):
update_environment()
self.py_vmomi_service = pyVmomiService(SmartConnect, Disconnect)
self.data_retriever_service = CloudshellDataRetrieverService()
self.connection_details_retriever = ResourceConnectionDetailsRetriever(self.data_retriever_service)
self.si = None
def integrationtest(self):
resource_connection_details_retriever = Mock()
credentials = TestCredentials()
resource_connection_details_retriever.connection_details = Mock(
return_value=VCenterConnectionDetails(credentials.host, credentials.username, credentials.password))
py_vmomi_service = pyVmomiService(SmartConnect, Disconnect)
synchronous_task_waiter = SynchronousTaskWaiter()
dv_port_group_creator = DvPortGroupCreator(py_vmomi_service, synchronous_task_waiter)
pg_name_generator = DvPortGroupNameGenerator
vnic_to_network_mapper = VnicToNetworkMapper(pg_name_generator)
virtual_machine_port_group_configurer = VirtualMachinePortGroupConfigurer(py_vmomi_service,
synchronous_task_waiter,
vnic_to_network_mapper,
VNicService())
virtual_switch_to_machine_connector = VirtualSwitchToMachineConnector(dv_port_group_creator,
virtual_machine_port_group_configurer)
si = py_vmomi_service.connect(credentials.host, credentials.username,
credentials.password,
credentials.port)
virtual_machine_path = 'Boris'
virtual_machine_name = 'boris1'
vm = py_vmomi_service.get_obj(si.content, [vim.VirtualMachine], virtual_machine_name)
vm_uuid = vm.config.uuid
#vm_uuid = self.get_vm_uuid(py_vmomi_service, si, virtual_machine_name)
port_group_path = 'QualiSB'
dv_switch_path = 'QualiSB'
dv_switch_name = 'dvSwitch'
dv_port_name = 'boris_group59'
def test_get_vm_by_uuid_vm_with_path(self):
"""
Checks whether the function can grab object by uuid
"""
'#arrange'
pv_service = pyVmomiService(None, None)
dc = Mock(spec=vim.Datacenter)
pv_service.find_item_in_path_by_type = Mock(return_value=dc)
si = create_autospec(spec=vim.ServiceInstance)
si.RetrieveContent = Mock()
si.content = create_autospec(spec=vim.ServiceInstanceContent())
si.content.searchIndex = Mock()
si.content.searchIndex.FindByUuid = Mock(
return_value='b8e4da4e-a2ff-11e5-bf7f-feff819cdc9f')
'#act'
result = pv_service.find_by_uuid(
si, 'b8e4da4e-a2ff-11e5-bf7f-feff819cdc9f', True, 'path/path/path')
'#assert'
self.assertTrue(result)
self.assertTrue(
pv_service.find_item_in_path_by_type.called_with(
si, 'path/path/path', vim.Datacenter))
def test_deploy_template(self):
# arrange
cred = TestCredentials()
pv_service = pyVmomiService(SmartConnect, Disconnect)
si = pv_service.connect(cred.host, cred.username, cred.password)
deployer = VirtualMachineDeployer(pv_service, generate_unique_name)
# vm = pv_service.find_vm_by_name(si, 'QualiSB/Raz', '2')
deploy_params = DeployDataHolder({
"template_model": {
"vCenter_resource_name": "QualiSB",
"vm_folder": "QualiSB/Raz",
"template_name": "2"
},
"vm_cluster_model": {
"cluster_name": "QualiSB Cluster",
"resource_pool": "IT"
},
"datastore_name": "eric ds cluster",
"power_on": False
})
deploy_command = DeployFromTemplateCommand(deployer)
# act
result = deploy_command.execute_deploy_from_template(si, deploy_params)
# assert
self.assertIsNotNone(result)
def integrationtest_refresh_ip(self):
resource_context = Mock()
resource_context.attributes = {"vCenter Template": "vCenter/Boris/Boris2-win7"}
qualipy_helpers = Mock()
qualipy_helpers.get_resource_context_details = Mock(return_value=resource_context)
credentials = TestCredentials()
py_vmomi_service = pyVmomiService(SmartConnect, Disconnect)
cloudshell_data_retriever_service = Mock()
cloudshell_data_retriever_service.getVCenterConnectionDetails = Mock(
return_value=VCenterConnectionDetails(credentials.host, credentials.username, credentials.password))
vm_resource = Mock()
vm_resource.default_network = 'default'
resource_model_parser = Mock()
resource_model_parser.convert_to_resource_model = Mock(return_value=vm_resource)
refresh_ip_command = RefreshIpCommand(py_vmomi_service, cloudshell_data_retriever_service, qualipy_helpers,
resource_model_parser)
uuid = helpers.get_uuid('Boris2-win7')
refresh_ip_command.refresh_ip(uuid, '')
pass
def get_uuid(virtual_machine_name):
credentials = TestCredentials()
py_vmomi_service = pyVmomiService(SmartConnect, Disconnect)
si = py_vmomi_service.connect(credentials.host, credentials.username,
credentials.password, credentials.port)
vm_uuid = get_vm_uuid(py_vmomi_service, si, virtual_machine_name)
return vm_uuid
def integration_clone_vm_destory(self):
"""
Checks whether clone_vm and destroy methods works
"""
'#arrange'
cred = TestCredentials()
pv_service = pyVmomiService(SmartConnect, Disconnect)
si = pv_service.connect(cred.host, cred.username, cred.password)
params = pv_service.CloneVmParameters(si=si,
template_name='DC0_C0_RP0_VM20',
vm_name='my_clone',
vm_folder='DC0')
'#act'
now = datetime.now()
res = pv_service.clone_vm(params)
logger.debug('clone took: %s' % (str(datetime.now() - now)))
'#assert'
self.assertTrue(type(res.vm), vim.VirtualMachine)
'#tear down'
now = datetime.now()
if res.error is None and res.vm is not None:
destroyed = pv_service.destroy_vm(res.vm)
logger.debug('destroy took: %s' % (str(datetime.now() - now)))
self.assertIsNone(destroyed)
def integrationtest_refresh_ip(self):
resource_context = Mock()
resource_context.attributes = {
"vCenter Template": "vCenter/Boris/Boris2-win7"
}
qualipy_helpers = Mock()
qualipy_helpers.get_resource_context_details = Mock(
return_value=resource_context)
credentials = TestCredentials()
py_vmomi_service = pyVmomiService(SmartConnect, Disconnect)
cloudshell_data_retriever_service = Mock()
cloudshell_data_retriever_service.getVCenterConnectionDetails = Mock(
return_value=VCenterConnectionDetails(
credentials.host, credentials.username, credentials.password))
vm_resource = Mock()
vm_resource.default_network = 'default'
resource_model_parser = Mock()
resource_model_parser.convert_to_resource_model = Mock(
return_value=vm_resource)
refresh_ip_command = RefreshIpCommand(
py_vmomi_service, cloudshell_data_retriever_service,
qualipy_helpers, resource_model_parser)
uuid = helpers.get_uuid('Boris2-win7')
refresh_ip_command.refresh_ip(uuid, '')
pass
def test_get_object_by_path_checks_childEntity(self):
"""
Checks whether the function can grab from child entities
"""
'#arrange'
pv_service = pyVmomiService(None, None)
def search_child(*args, **keys):
if args[0].name == pv_service.ChildEntity:
return True
return False
class FolderMock:
childEntity = None
folder = Mock(spec=FolderMock())
folder.name = pv_service.ChildEntity
get_folder = MagicMock(return_value=folder)
pv_service.get_folder = get_folder
si = create_autospec(spec=vim.ServiceInstance)
si.RetrieveContent = Mock()
si.content = create_autospec(spec=vim.ServiceInstanceContent())
si.content.searchIndex = Mock()
si.content.searchIndex.FindChild = MagicMock(side_effect=search_child)
'#act'
result = pv_service.find_obj_by_path(si, '', '', '')
'#assert'
self.assertTrue(result)
def integration_test_connect_B(self):
py_vmomi_service = pyVmomiService(SmartConnect, Disconnect)
cred = TestCredentials()
si = py_vmomi_service.connect(cred.host, cred.username, cred.password, cred.port)
vm_uuid = py_vmomi_service.find_vm_by_name(si, 'QualiSB/Boris', 'Boris2-win7').config.uuid
mapping = VmNetworkMapping()
mapping.vlan_id = '114'
# mapping.dv_port_name = 65
mapping.dv_switch_name = 'dvSwitch'
mapping.dv_switch_path = 'QualiSB'
mapping.port_group_path = 'QualiSB'
mapping.vlan_spec = 'Trunk'
vlan_spec = VlanSpecFactory()
range_fac = VLanIdRangeParser()
synchronous_task_waiter = SynchronousTaskWaiter()
name_gen = DvPortGroupNameGenerator()
mapper = VnicToNetworkMapper(name_gen)
dv_port_group_creator = DvPortGroupCreator(py_vmomi_service, synchronous_task_waiter)
virtual_machine_port_group_configurer = VirtualMachinePortGroupConfigurer(py_vmomi_service,
synchronous_task_waiter,
mapper,
VNicService())
connector = VirtualSwitchToMachineConnector(dv_port_group_creator, virtual_machine_port_group_configurer)
vnic_updater = VnicUpdater(helpers)
command = VirtualSwitchConnectCommand(py_vmomi_service, connector, name_gen, vlan_spec, range_fac, vnic_updater)
command.connect_to_networks(si, vm_uuid, [mapping], 'QualiSB/anetwork')
def integration_test_connect_A(self):
py_vmomi_service = pyVmomiService(SmartConnect, Disconnect)
cred = TestCredentials()
si = py_vmomi_service.connect(cred.host, cred.username, cred.password, cred.port)
synchronous_task_waiter = SynchronousTaskWaiter()
mapper = VnicToNetworkMapper(DvPortGroupNameGenerator())
dv_port_group_creator = DvPortGroupCreator(py_vmomi_service, synchronous_task_waiter)
virtual_machine_port_group_configurer = VirtualMachinePortGroupConfigurer(py_vmomi_service,
synchronous_task_waiter,
mapper,
VNicService())
mapping = VmNetworkMapping()
mapping.vlan_id = [vim.NumericRange(start=65, end=65)]
mapping.dv_port_name = DvPortGroupNameGenerator().generate_port_group_name(65)
mapping.dv_switch_name = 'dvSwitch'
mapping.dv_switch_path = 'QualiSB'
mapping.port_group_path = 'QualiSB'
mapping.vlan_spec = vim.dvs.VmwareDistributedVirtualSwitch.TrunkVlanSpec()
connector = VirtualSwitchToMachineConnector(
dv_port_group_creator,
virtual_machine_port_group_configurer)
vm = py_vmomi_service.find_vm_by_name(si, 'QualiSB/Raz', '2')
# Act
connector.connect_by_mapping(si, vm, [mapping], None)
pass
def integration_clone_vm_destory(self):
"""
Checks whether clone_vm and destroy methods works
"""
'#arrange'
cred = TestCredentials()
pv_service = pyVmomiService(SmartConnect, Disconnect)
si = pv_service.connect(cred.host, cred.username, cred.password)
params = pv_service.CloneVmParameters(si=si,
template_name='DC0_C0_RP0_VM20',
vm_name='my_clone',
vm_folder='DC0')
'#act'
now = datetime.now()
res = pv_service.clone_vm(params)
logger.debug('clone took: %s' % (str(datetime.now() - now)))
'#assert'
self.assertTrue(type(res.vm), vim.VirtualMachine)
'#tear down'
now = datetime.now()
if res.error is None and res.vm is not None:
destroyed = pv_service.destroy_vm(res.vm)
logger.debug('destroy took: %s' % (str(datetime.now() - now)))
self.assertIsNone(destroyed)
def test_get_object_by_path_checks_childEntity(self):
"""
Checks whether the function can grab from child entities
"""
'#arrange'
pv_service = pyVmomiService(None, None)
def search_child(*args, **keys):
if args[0].name == pv_service.ChildEntity:
return True
return False
class FolderMock:
childEntity = None
folder = Mock(spec=FolderMock())
folder.name = pv_service.ChildEntity
get_folder = MagicMock(return_value=folder)
pv_service.get_folder = get_folder
si = create_autospec(spec=vim.ServiceInstance)
si.RetrieveContent = Mock()
si.content = create_autospec(spec=vim.ServiceInstanceContent())
si.content.searchIndex = Mock()
si.content.searchIndex.FindChild = MagicMock(side_effect=search_child)
'#act'
result = pv_service.find_obj_by_path(si, '', '', '')
'#assert'
self.assertTrue(result)
def test_deploy_template(self):
# arrange
cred = TestCredentials()
pv_service = pyVmomiService(SmartConnect, Disconnect)
si = pv_service.connect(cred.host, cred.username, cred.password)
deployer = VirtualMachineDeployer(pv_service, generate_unique_name)
# vm = pv_service.find_vm_by_name(si, 'QualiSB/Raz', '2')
deploy_params = DeployDataHolder(
{
"template_model": {
"vCenter_resource_name": "QualiSB",
"vm_folder": "QualiSB/Raz",
"template_name": "2",
},
"vm_cluster_model": {"cluster_name": "QualiSB Cluster", "resource_pool": "IT"},
"datastore_name": "eric ds cluster",
"power_on": False,
}
)
deploy_command = DeployFromTemplateCommand(deployer)
# act
result = deploy_command.execute_deploy_from_template(si, deploy_params)
# assert
self.assertIsNotNone(result)
def test_find_item_in_path_by_type_complex_path(self):
"""
Checks whether the function can grab object by uuid
"""
'#arrange'
class counter:
i = 0
def side_eff(*args, **kwargs):
if counter.i != 1:
counter.i = counter.i + 1;
return 'not yet'
else:
counter.i = counter.i + 1;
return Mock(spec=vim.Datacenter)
pv_service = pyVmomiService(None, None)
si = create_autospec(spec=vim.ServiceInstance)
si.RetrieveContent = Mock()
si.content = create_autospec(spec=vim.ServiceInstanceContent())
si.content.searchIndex = Mock()
si.content.rootFolder = Mock()
si.content.searchIndex.FindChild = Mock(side_effect=side_eff)
'#act'
result = pv_service.find_item_in_path_by_type(si, 'test//dc/asd', vim.Datacenter)
'#assert'
self.assertTrue(isinstance(result, vim.Datacenter))
self.assertTrue(counter.i, 1)
def test_find_item_in_path_by_type_complex_path(self):
"""
Checks whether the function can grab object by uuid
"""
'#arrange'
class counter:
i = 0
def side_eff(*args, **kwargs):
if counter.i != 1:
counter.i = counter.i + 1
return 'not yet'
else:
counter.i = counter.i + 1
return Mock(spec=vim.Datacenter)
pv_service = pyVmomiService(None, None)
si = create_autospec(spec=vim.ServiceInstance)
si.RetrieveContent = Mock()
si.content = create_autospec(spec=vim.ServiceInstanceContent())
si.content.searchIndex = Mock()
si.content.rootFolder = Mock()
si.content.searchIndex.FindChild = Mock(side_effect=side_eff)
'#act'
result = pv_service.find_item_in_path_by_type(si, 'test//dc/asd',
vim.Datacenter)
'#assert'
self.assertTrue(isinstance(result, vim.Datacenter))
self.assertTrue(counter.i, 1)
def __init__(self):
py_vmomi_service = pyVmomiService(SmartConnect, Disconnect)
cloudshell_data_retriever_service = CloudshellDataRetrieverService()
resource_connection_details_retriever = ResourceConnectionDetailsRetriever(
helpers, cloudshell_data_retriever_service)
resource_remover = CloudshellResourceRemover(helpers)
command_wrapper = CommandWrapper(getLogger, py_vmomi_service)
template_deployer = VirtualMachineDeployer(py_vmomi_service,
generate_unique_name)
deploy_from_template_command = DeployFromTemplateCommand(
template_deployer)
resource_model_parser = ResourceModelParser()
vc_model_retriever = VCenterDataModelRetriever(helpers,
resource_model_parser)
vc_data_model = vc_model_retriever.get_vcenter_data_model()
#vnic_to_network_mapper = VnicToNetworkMapper(name_generator, vc_data_model.default_network)
port_group_name_generator = DvPortGroupNameGenerator()
vnic_to_network_mapper = VnicToNetworkMapper(port_group_name_generator)
# Virtual Switch Connect
synchronous_task_waiter = SynchronousTaskWaiter()
dv_port_group_creator = DvPortGroupCreator(py_vmomi_service,
synchronous_task_waiter)
virtual_machine_port_group_configurer = VirtualMachinePortGroupConfigurer(
py_vmomi_service, synchronous_task_waiter, vnic_to_network_mapper,
VNicService())
virtual_switch_to_machine_connector = VirtualSwitchToMachineConnector(
dv_port_group_creator, virtual_machine_port_group_configurer)
virtual_switch_connect_command = VirtualSwitchConnectCommand(
py_vmomi_service, virtual_switch_to_machine_connector,
DvPortGroupNameGenerator(), VlanSpecFactory(), VLanIdRangeParser(),
getLogger('VirtualSwitchConnectCommand'))
# Virtual Switch Revoke
virtual_switch_disconnect_command = \
VirtualSwitchToMachineDisconnectCommand(py_vmomi_service,
virtual_machine_port_group_configurer,
vc_data_model.default_network)
destroy_virtual_machine_command = DestroyVirtualMachineCommand(
py_vmomi_service, resource_remover,
virtual_switch_disconnect_command)
# Power Command
vm_power_management_command = VirtualMachinePowerManagementCommand(
py_vmomi_service, synchronous_task_waiter)
# Refresh IP command
refresh_ip_command = RefreshIpCommand(
py_vmomi_service, cloudshell_data_retriever_service, helpers,
resource_model_parser)
self.commandExecuterService = CommandExecuterService(
jsonpickle, helpers, command_wrapper,
resource_connection_details_retriever, vc_data_model,
destroy_virtual_machine_command, deploy_from_template_command,
virtual_switch_connect_command, virtual_switch_disconnect_command,
vm_power_management_command, refresh_ip_command)
def test_get_folder_deep_and_complex_path(self):
"""
Checks when path is deep and complex, goes through all the folder types
"""
'#arrange'
pv_service = pyVmomiService(None, None)
def find_child_mock(*args):
root = args[0]
if hasattr(root, pv_service.ChildEntity):
for folder in root.childEntity:
if folder.name == args[1]:
return folder
else:
for folder in root:
if folder.name == args[1]:
return folder
return None
si = create_autospec(spec=vim.ServiceInstance)
si.RetrieveContent = Mock()
si.content = create_autospec(spec=vim.ServiceInstanceContent())
si.content.searchIndex = Mock()
si.content.searchIndex.FindChild = MagicMock(
side_effect=find_child_mock)
first_folder = Mock(spec=[], name='first')
first_folder.name = 'first'
second_folder = Mock(spec=[], name='second')
second_folder.name = 'second'
third_folder = Mock(spec=[], name='third')
third_folder.name = 'third'
fourth_folder = Mock(spec=[], name='fourth')
fourth_folder.name = 'fourth'
fifth_folder = Mock(spec=[], name='fifth')
fifth_folder.name = 'fifth'
sixth_folder = Mock(spec=[], name='sixth')
sixth_folder.name = 'sixth'
si.content.rootFolder = Mock()
si.content.rootFolder.name = 'rootFolder'
si.content.rootFolder.childEntity = [first_folder, second_folder]
first_folder.vmFolder = [second_folder, sixth_folder]
second_folder.networkFolder = [fourth_folder, third_folder]
third_folder.hostFolder = [third_folder, fourth_folder]
fourth_folder.datacenterFolder = [fifth_folder]
fifth_folder.datastoreFolder = [sixth_folder]
'#act'
result = pv_service.get_folder(
si, 'first/second/third/fourth/fifth/sixth')
'#assert'
self.assertEqual(result, sixth_folder)
def get_uuid(virtual_machine_name):
credentials = TestCredentials()
py_vmomi_service = pyVmomiService(SmartConnect, Disconnect)
si = py_vmomi_service.connect(credentials.host, credentials.username,
credentials.password,
credentials.port)
vm_uuid = get_vm_uuid(py_vmomi_service, si, virtual_machine_name)
return vm_uuid
def __init__(self):
update_environment()
self.py_vmomi_service = pyVmomiService(SmartConnect, Disconnect)
self.data_retriever_service = CloudshellDataRetrieverService()
self.connection_details_retriever = ResourceConnectionDetailsRetriever(
self.data_retriever_service)
self.si = None
def test_get_folder_deep_and_complex_path(self):
"""
Checks when path is deep and complex, goes through all the folder types
"""
'#arrange'
pv_service = pyVmomiService(None, None)
def find_child_mock(*args):
root = args[0]
if hasattr(root, pv_service.ChildEntity):
for folder in root.childEntity:
if folder.name == args[1]:
return folder
else:
for folder in root:
if folder.name == args[1]:
return folder
return None
si = create_autospec(spec=vim.ServiceInstance)
si.RetrieveContent = Mock()
si.content = create_autospec(spec=vim.ServiceInstanceContent())
si.content.searchIndex = Mock()
si.content.searchIndex.FindChild = MagicMock(side_effect=find_child_mock)
first_folder = Mock(spec=[], name='first')
first_folder.name = 'first'
second_folder = Mock(spec=[], name='second')
second_folder.name = 'second'
third_folder = Mock(spec=[], name='third')
third_folder.name = 'third'
fourth_folder = Mock(spec=[], name='fourth')
fourth_folder.name = 'fourth'
fifth_folder = Mock(spec=[], name='fifth')
fifth_folder.name = 'fifth'
sixth_folder = Mock(spec=[], name='sixth')
sixth_folder.name = 'sixth'
si.content.rootFolder = Mock()
si.content.rootFolder.name = 'rootFolder'
si.content.rootFolder.childEntity = [first_folder, second_folder]
first_folder.vmFolder = [second_folder, sixth_folder]
second_folder.networkFolder = [fourth_folder, third_folder]
third_folder.hostFolder = [third_folder, fourth_folder]
fourth_folder.datacenterFolder = [fifth_folder]
fifth_folder.datastoreFolder = [sixth_folder]
'#act'
result = pv_service.get_folder(si, 'first/second/third/fourth/fifth/sixth')
'#assert'
self.assertEqual(result, sixth_folder)
def test_power_management(self):
# arrange
cred = TestCredentials()
pv_service = pyVmomiService(SmartConnect, Disconnect)
si = pv_service.connect(cred.host, cred.username, cred.password)
uuid = pv_service.find_vm_by_name(si, 'QualiSB/Raz', '2').config.uuid
power_manager = VirtualMachinePowerManagementCommand(
pv_service, task_waiter.SynchronousTaskWaiter())
power_manager.power_on(si, uuid)
power_manager.power_off(si, uuid)
def test_power_management(self):
# arrange
cred = TestCredentials()
pv_service = pyVmomiService(SmartConnect, Disconnect)
si = pv_service.connect(cred.host, cred.username, cred.password)
uuid = pv_service.find_vm_by_name(si, 'QualiSB/Raz', '2').config.uuid
power_manager = VirtualMachinePowerManagementCommand(pv_service,
task_waiter.SynchronousTaskWaiter())
power_manager.power_on(si, uuid)
power_manager.power_off(si, uuid)
def integration_test(self):
py_vmomi_service = pyVmomiService(SmartConnect, Disconnect)
cred = TestCredentials()
si = py_vmomi_service.connect(cred.host, cred.username, cred.password, cred.port)
vm = py_vmomi_service.find_vm_by_name(si, 'QualiSB/Alex', 'Ubuntu_a287f573')
os.environ['VM_UUID'] = vm.config.uuid
os.environ['VLAN_ID'] = '25'
os.environ['VLAN_SPEC_TYPE'] = 'Access'
bootstrapper = Bootstrapper()
executer_service = bootstrapper.get_command_executer_service()
executer_service.connect()
def test_get_vm_by_uuid_vm_without_uuid(self):
"""
Checks whether the function can grab object by uuid
"""
'#arrange'
pv_service = pyVmomiService(None, None)
si = create_autospec(spec=vim.ServiceInstance)
si.RetrieveContent = Mock()
si.content = create_autospec(spec=vim.ServiceInstanceContent())
'#act'
result = pv_service.find_by_uuid(si, None)
'#assert'
self.assertIsNone(result)
def test_get_object_by_path_no_nested_objs(self):
"""
Checks whether the function returns 'None' if it doesn't find an object
"""
'#arrange'
pv_service = pyVmomiService(None, None)
get_folder = MagicMock(return_value=Mock(spec=[]))
pv_service.get_folder = get_folder
si = create_autospec(spec=vim.ServiceInstance)
si.RetrieveContent = Mock()
si.content = create_autospec(spec=vim.ServiceInstanceContent())
'#act'
self.assertRaises(KeyError, pv_service.find_obj_by_path,si, '', '', '')
def test_clone_vm_si_is_none(self):
"""
Checks clone_vm
"""
'#arrange'
pv_service = pyVmomiService(None, None)
params = pv_service.CloneVmParameters(si=None,
template_name=None,
vm_name=None,
vm_folder=None)
'#act'
res = pv_service.clone_vm(params)
'#assert'
self.assertTrue(res.error is not None)
def test_get_vm_by_uuid_vm_without_uuid(self):
"""
Checks whether the function can grab object by uuid
"""
'#arrange'
pv_service = pyVmomiService(None, None)
si = create_autospec(spec=vim.ServiceInstance)
si.RetrieveContent = Mock()
si.content = create_autospec(spec=vim.ServiceInstanceContent())
'#act'
result = pv_service.find_by_uuid(si, None)
'#assert'
self.assertIsNone(result)
def integration_test(self):
py_vmomi_service = pyVmomiService(SmartConnect, Disconnect)
cred = TestCredentials()
si = py_vmomi_service.connect(cred.host, cred.username, cred.password,
cred.port)
vm = py_vmomi_service.find_vm_by_name(si, 'QualiSB/Alex',
'Ubuntu_a287f573')
os.environ['VM_UUID'] = vm.config.uuid
os.environ['VLAN_ID'] = '25'
os.environ['VLAN_SPEC_TYPE'] = 'Access'
bootstrapper = Bootstrapper()
executer_service = bootstrapper.get_command_executer_service()
executer_service.connect()
def test_clone_vm_si_is_none(self):
"""
Checks clone_vm
"""
'#arrange'
pv_service = pyVmomiService(None, None)
params = pv_service.CloneVmParameters(si=None,
template_name=None,
vm_name=None,
vm_folder=None)
'#act'
res = pv_service.clone_vm(params)
'#assert'
self.assertTrue(res.error is not None)
def test_get_object_by_path_no_nested_objs(self):
"""
Checks whether the function returns 'None' if it doesn't find an object
"""
'#arrange'
pv_service = pyVmomiService(None, None)
get_folder = MagicMock(return_value=Mock(spec=[]))
pv_service.get_folder = get_folder
si = create_autospec(spec=vim.ServiceInstance)
si.RetrieveContent = Mock()
si.content = create_autospec(spec=vim.ServiceInstanceContent())
'#act'
self.assertRaises(KeyError, pv_service.find_obj_by_path, si, '', '',
'')
def test_clone_vm_template_name_is_none(self):
"""
Checks clone_vm
"""
'#arrange'
si = create_autospec(spec=vim.ServiceInstance)
pv_service = pyVmomiService(None, None)
params = pv_service.CloneVmParameters(si=si,
template_name=None,
vm_name=None,
vm_folder=None)
'#act'
res = pv_service.clone_vm(params)
'#assert'
self.assertTrue(res.error is not None)
def test_clone_vm_template_name_is_none(self):
"""
Checks clone_vm
"""
'#arrange'
si = create_autospec(spec=vim.ServiceInstance)
pv_service = pyVmomiService(None, None)
params = pv_service.CloneVmParameters(si=si,
template_name=None,
vm_name=None,
vm_folder=None)
'#act'
res = pv_service.clone_vm(params)
'#assert'
self.assertTrue(res.error is not None)
def integration_test_update_vnics(self):
dev_helpers.attach_to_cloudshell_as("admin", "admin", "Global", "1205e711-edf7-4b12-8a5e-e0ff53768ce7")
details = helpers.get_reservation_context_details()
py_vmomi_service = pyVmomiService(SmartConnect, Disconnect)
cred = TestCredentials()
si = py_vmomi_service.connect(cred.host, cred.username, cred.password, cred.port)
vm = py_vmomi_service.find_vm_by_name(si, 'QualiSB/Alex', 'Ubuntu_a287f573')
nics = [x for x in vm.config.hardware.device
if isinstance(x, vim.vm.device.VirtualEthernetCard)]
for nic in nics:
network_name = nic.backing.network.name
mac_address = nic.macAddress
print network_name + mac_address
def test_find_item_in_path_by_type_path_None(self):
"""
Checks whether the function can grab object by uuid
"""
'#arrange'
pv_service = pyVmomiService(None, None)
si = create_autospec(spec=vim.ServiceInstance)
si.RetrieveContent = Mock()
si.content = create_autospec(spec=vim.ServiceInstanceContent())
si.content.searchIndex = Mock()
si.content.rootFolder = Mock
'#act'
result = pv_service.find_item_in_path_by_type(si, None, 'not none')
'#assert'
self.assertEqual(result, si.content.rootFolder)
def test_find_item_in_path_by_type_path_None(self):
"""
Checks whether the function can grab object by uuid
"""
'#arrange'
pv_service = pyVmomiService(None, None)
si = create_autospec(spec=vim.ServiceInstance)
si.RetrieveContent = Mock()
si.content = create_autospec(spec=vim.ServiceInstanceContent())
si.content.searchIndex = Mock()
si.content.rootFolder = Mock
'#act'
result = pv_service.find_item_in_path_by_type(si, None, 'not none')
'#assert'
self.assertEqual(result, si.content.rootFolder)
def test_get_folder_deep_path(self):
"""
Checks when path is deep, more then two
"""
'#arrange'
pv_service = pyVmomiService(None, None)
def find_child_mock(*args):
root = args[0]
for folder in root.childEntity:
if folder.name == args[1]:
return folder
return None
si = create_autospec(spec=vim.ServiceInstance)
si.RetrieveContent = Mock()
si.content = create_autospec(spec=vim.ServiceInstanceContent())
si.content.searchIndex = Mock()
si.content.searchIndex.FindChild = MagicMock(
side_effect=find_child_mock)
inner_folder = Mock()
inner_folder.name = 'inner'
inner_decoy_folder = Mock()
inner_decoy_folder.name = 'decoy'
inner_deep_folder = Mock()
inner_deep_folder.name = 'inner_deep_folder'
inner_folder.childEntity = [inner_deep_folder, inner_decoy_folder]
inner_decoy_folder.childEntity = [inner_folder]
si.content.rootFolder = Mock()
si.content.rootFolder.childEntity = [inner_decoy_folder, inner_folder]
si.content.rootFolder.name = 'rootFolder'
'#act'
result = pv_service.get_folder(si, 'decoy/inner/inner_deep_folder/')
'#assert'
self.assertEqual(result, inner_deep_folder)
def test_get_network_by_name_is_network_type(self):
"""
Checks whether the function can passes network type
"""
'#arrange'
pv_service = pyVmomiService(None, None)
def find_obj_by_path_mock(*args, **kwargs):
return args[3] == pv_service.Network
si = create_autospec(spec=vim.ServiceInstance)
si.RetrieveContent = Mock()
si.content = create_autospec(spec=vim.ServiceInstanceContent())
pv_service.find_obj_by_path = Mock(side_effect=find_obj_by_path_mock)
'#act'
result = pv_service.find_network_by_name(si, '', '')
'#assert'
self.assertTrue(result)
def test_get_vm_by_uuid_vm_without_path(self):
"""
Checks whether the function can grab object by uuid
"""
'#arrange'
pv_service = pyVmomiService(None, None)
si = create_autospec(spec=vim.ServiceInstance)
si.RetrieveContent = Mock()
si.content = create_autospec(spec=vim.ServiceInstanceContent())
si.content.searchIndex = Mock()
si.content.searchIndex.FindByUuid = Mock(return_value='b8e4da4e-a2ff-11e5-bf7f-feff819cdc9f')
'#act'
result = pv_service.find_by_uuid(si, 'b8e4da4e-a2ff-11e5-bf7f-feff819cdc9f')
'#assert'
self.assertTrue(result)
self.assertTrue(si.content.searchIndex.FindByUuid.called)
def test_clone_vm_resource_pool_is_not_empty(self):
"""
Checks clone_vm
"""
'#arrange'
si = Mock(spec=vim.ServiceInstance)
vim_mock = Mock()
vim_mock.vm = Mock()
vim_mock.vm.RelocateSpec = Mock()
vim_mock.vm.CloneSpec = Mock()
vim_mock.Datacenter = vim.Datacenter
vim_mock.Datastore = vim.Datastore
vim_mock.ServiceInstance = vim.ServiceInstance
datacenter = Mock(spec=vim.Datacenter)
template = Mock(spec=vim.VirtualMachine)
template.datastore = [Mock()]
pv_service = pyVmomiService(None, None, vim_mock)
pv_service.find_vm_by_name = Mock(return_value=template)
pv_service.get_obj = Mock()
pv_service.get_folder = Mock(return_value=datacenter)
pv_service.wait_for_task = Mock()
params = pv_service.CloneVmParameters(si=si,
template_name='my_temp',
vm_name='my_name',
vm_folder='my_folder',
resource_pool='my_resource_pool')
'#act'
res = pv_service.clone_vm(params)
'#assert'
self.assertIsNone(res.error)
self.assertTrue(vim_mock.vm.RelocateSpec.called)
self.assertTrue(vim_mock.vm.CloneSpec.called)
self.assertTrue(pv_service.get_folder.called)
self.assertTrue(pv_service.find_vm_by_name.called)
self.assertTrue(pv_service.get_obj.called)
self.assertTrue(pv_service.wait_for_task.called)
def test_get_folder_deep_path(self):
"""
Checks when path is deep, more then two
"""
'#arrange'
pv_service = pyVmomiService(None, None)
def find_child_mock(*args):
root = args[0]
for folder in root.childEntity:
if folder.name == args[1]:
return folder
return None
si = create_autospec(spec=vim.ServiceInstance)
si.RetrieveContent = Mock()
si.content = create_autospec(spec=vim.ServiceInstanceContent())
si.content.searchIndex = Mock()
si.content.searchIndex.FindChild = MagicMock(side_effect=find_child_mock)
inner_folder = Mock()
inner_folder.name = 'inner'
inner_decoy_folder = Mock()
inner_decoy_folder.name = 'decoy'
inner_deep_folder = Mock()
inner_deep_folder.name = 'inner_deep_folder'
inner_folder.childEntity = [inner_deep_folder, inner_decoy_folder]
inner_decoy_folder.childEntity = [inner_folder]
si.content.rootFolder = Mock()
si.content.rootFolder.childEntity = [inner_decoy_folder, inner_folder]
si.content.rootFolder.name = 'rootFolder'
'#act'
result = pv_service.get_folder(si, 'decoy/inner/inner_deep_folder/')
'#assert'
self.assertEqual(result, inner_deep_folder)
def test_clone_vm_resource_pool_is_not_empty(self):
"""
Checks clone_vm
"""
'#arrange'
si = Mock(spec=vim.ServiceInstance)
vim_mock = Mock()
vim_mock.vm = Mock()
vim_mock.vm.RelocateSpec = Mock()
vim_mock.vm.CloneSpec = Mock()
vim_mock.Datacenter = vim.Datacenter
vim_mock.Datastore = vim.Datastore
vim_mock.ServiceInstance = vim.ServiceInstance
datacenter = Mock(spec=vim.Datacenter)
template = Mock(spec=vim.VirtualMachine)
template.datastore = [Mock()]
pv_service = pyVmomiService(None, None, vim_mock)
pv_service.find_vm_by_name = Mock(return_value=template)
pv_service.get_obj = Mock()
pv_service.get_folder = Mock(return_value=datacenter)
pv_service.wait_for_task = Mock()
params = pv_service.CloneVmParameters(si=si,
template_name='my_temp',
vm_name='my_name',
vm_folder='my_folder',
resource_pool='my_resource_pool')
'#act'
res = pv_service.clone_vm(params)
'#assert'
self.assertIsNone(res.error)
self.assertTrue(vim_mock.vm.RelocateSpec.called)
self.assertTrue(vim_mock.vm.CloneSpec.called)
self.assertTrue(pv_service.get_folder.called)
self.assertTrue(pv_service.find_vm_by_name.called)
self.assertTrue(pv_service.get_obj.called)
self.assertTrue(pv_service.wait_for_task.called)
def test_get_folder_path_empty(self):
"""
Checks if the receiving path is empty, the function returns root folder
"""
'#arrange'
folder_name = 'rootFolder'
pv_service = pyVmomiService(None, None)
si = create_autospec(spec=vim.ServiceInstance)
si.RetrieveContent = Mock()
si.content = create_autospec(spec=vim.ServiceInstanceContent())
si.content.rootFolder = Mock()
si.content.rootFolder.name = folder_name
'#act'
result = pv_service.get_folder(si, '')
'#assert'
self.assertEqual(result.name, folder_name)
def test_get_folder_path_empty(self):
"""
Checks if the receiving path is empty, the function returns root folder
"""
'#arrange'
folder_name = 'rootFolder'
pv_service = pyVmomiService(None, None)
si = create_autospec(spec=vim.ServiceInstance)
si.RetrieveContent = Mock()
si.content = create_autospec(spec=vim.ServiceInstanceContent())
si.content.rootFolder = Mock()
si.content.rootFolder.name = folder_name
'#act'
result = pv_service.get_folder(si, '')
'#assert'
self.assertEqual(result.name, folder_name)
def test_get_network_by_name_is_network_type(self):
"""
Checks whether the function can passes network type
"""
'#arrange'
pv_service = pyVmomiService(None, None)
def find_obj_by_path_mock(*args, **kwargs):
return args[3] == pv_service.Network
si = create_autospec(spec=vim.ServiceInstance)
si.RetrieveContent = Mock()
si.content = create_autospec(spec=vim.ServiceInstanceContent())
pv_service.find_obj_by_path = Mock(side_effect=find_obj_by_path_mock)
'#act'
result = pv_service.find_network_by_name(si, '', '')
'#assert'
self.assertTrue(result)
def integration_test_update_vnics(self):
dev_helpers.attach_to_cloudshell_as(
"admin", "admin", "Global", "1205e711-edf7-4b12-8a5e-e0ff53768ce7")
details = helpers.get_reservation_context_details()
py_vmomi_service = pyVmomiService(SmartConnect, Disconnect)
cred = TestCredentials()
si = py_vmomi_service.connect(cred.host, cred.username, cred.password,
cred.port)
vm = py_vmomi_service.find_vm_by_name(si, 'QualiSB/Alex',
'Ubuntu_a287f573')
nics = [
x for x in vm.config.hardware.device
if isinstance(x, vim.vm.device.VirtualEthernetCard)
]
for nic in nics:
network_name = nic.backing.network.name
mac_address = nic.macAddress
print network_name + mac_address
def test_destroy_vm_by_name(self):
"""
Checks whether the vm found and call to be destroy
"""
'#arrange'
pv_service = pyVmomiService(None, None)
si = create_autospec(spec=vim.ServiceInstance)
si.RetrieveContent = Mock()
si.content = create_autospec(spec=vim.ServiceInstanceContent())
si.content.searchIndex = Mock()
pv_service.find_vm_by_name = Mock(return_value=Mock(name='vm'))
pv_service.destroy_vm = Mock(return_value=True)
'#act'
result = pv_service.destroy_vm_by_name(si, 'vm_name:name', 'fake/path')
'#assert'
self.assertTrue(result)
self.assertTrue(pv_service.find_vm_by_name.called)
self.assertTrue(pv_service.destroy_vm.called)
def test_destroy_vm_by_uuid(self):
"""
Checks whether the vm found and call to be destroy
"""
'#arrange'
pv_service = pyVmomiService(None, None)
si = create_autospec(spec=vim.ServiceInstance)
si.RetrieveContent = Mock()
si.content = create_autospec(spec=vim.ServiceInstanceContent())
si.content.searchIndex = Mock()
pv_service.find_by_uuid = Mock(return_value=Mock(name='vm'))
pv_service.destroy_vm = Mock(return_value=True)
'#act'
result = pv_service.destroy_vm_by_uuid(si, 'thisuni-vers-ally-uniq-ueidentifier', 'fake/path')
'#assert'
self.assertTrue(result)
self.assertTrue(pv_service.find_by_uuid.called)
self.assertTrue(pv_service.destroy_vm.called)
def test_get_object_by_path_no_folder_found(self):
"""
Checks if the receiving path that does not exist
"""
'#arrange'
folder_name = 'rootFolder'
pv_service = pyVmomiService(None, None)
si = create_autospec(spec=vim.ServiceInstance)
si.RetrieveContent = Mock()
si.content = create_autospec(spec=vim.ServiceInstanceContent())
si.content.rootFolder = Mock()
si.content.rootFolder.name = folder_name
pv_service.get_folder = Mock(return_value=None)
'#act'
self.assertRaises(KeyError, pv_service.find_obj_by_path, si, 'nothing/to/be/found', 'fake_vm', pv_service.VM)
'#assert'
self.assertTrue(pv_service.get_folder.called)
def test_get_object_by_path_no_folder_found(self):
"""
Checks if the receiving path that does not exist
"""
'#arrange'
folder_name = 'rootFolder'
pv_service = pyVmomiService(None, None)
si = create_autospec(spec=vim.ServiceInstance)
si.RetrieveContent = Mock()
si.content = create_autospec(spec=vim.ServiceInstanceContent())
si.content.rootFolder = Mock()
si.content.rootFolder.name = folder_name
pv_service.get_folder = Mock(return_value=None)
'#act'
self.assertRaises(KeyError, pv_service.find_obj_by_path, si,
'nothing/to/be/found', 'fake_vm', pv_service.VM)
'#assert'
self.assertTrue(pv_service.get_folder.called)
def __init__(self):
py_vmomi_service = pyVmomiService(SmartConnect, Disconnect)
cloudshell_data_retriever_service = CloudshellDataRetrieverService()
resource_connection_details_retriever = ResourceConnectionDetailsRetriever(helpers,
cloudshell_data_retriever_service)
resource_remover = CloudshellResourceRemover(helpers)
command_wrapper = CommandWrapper(getLogger, py_vmomi_service)
template_deployer = VirtualMachineDeployer(py_vmomi_service, generate_unique_name)
deploy_from_template_command = DeployFromTemplateCommand(template_deployer)
resource_model_parser = ResourceModelParser()
vc_model_retriever = VCenterDataModelRetriever(helpers, resource_model_parser)
vc_data_model = vc_model_retriever.get_vcenter_data_model()
#vnic_to_network_mapper = VnicToNetworkMapper(name_generator, vc_data_model.default_network)
port_group_name_generator = DvPortGroupNameGenerator()
vnic_to_network_mapper = VnicToNetworkMapper(port_group_name_generator)
# Virtual Switch Connect
synchronous_task_waiter = SynchronousTaskWaiter()
dv_port_group_creator = DvPortGroupCreator(py_vmomi_service, synchronous_task_waiter)
virtual_machine_port_group_configurer = VirtualMachinePortGroupConfigurer(py_vmomi_service,
synchronous_task_waiter,
vnic_to_network_mapper,
VNicService())
virtual_switch_to_machine_connector = VirtualSwitchToMachineConnector(dv_port_group_creator,
virtual_machine_port_group_configurer)
virtual_switch_connect_command = VirtualSwitchConnectCommand(py_vmomi_service,
virtual_switch_to_machine_connector,
DvPortGroupNameGenerator(),
VlanSpecFactory(),
VLanIdRangeParser(),
getLogger('VirtualSwitchConnectCommand'))
# Virtual Switch Revoke
virtual_switch_disconnect_command = \
VirtualSwitchToMachineDisconnectCommand(py_vmomi_service,
virtual_machine_port_group_configurer,
vc_data_model.default_network)
destroy_virtual_machine_command = DestroyVirtualMachineCommand(py_vmomi_service,
resource_remover,
virtual_switch_disconnect_command)
# Power Command
vm_power_management_command = VirtualMachinePowerManagementCommand(py_vmomi_service,
synchronous_task_waiter)
# Refresh IP command
refresh_ip_command = RefreshIpCommand(py_vmomi_service, cloudshell_data_retriever_service, helpers,
resource_model_parser)
self.commandExecuterService = CommandExecuterService(jsonpickle,
helpers,
command_wrapper,
resource_connection_details_retriever,
vc_data_model,
destroy_virtual_machine_command,
deploy_from_template_command,
virtual_switch_connect_command,
virtual_switch_disconnect_command,
vm_power_management_command,
refresh_ip_command)
def __init__(self, method_ame="runTest"):
super(SearchObjectsPerfTest, self).__init__(method_ame)
cred = TestCredentials()
self.pv_service = pyVmomiService(SmartConnect, Disconnect)
self.si = self.pv_service.connect(cred.host, cred.username, cred.password)
self.logger = getLogger("performance")
