def test_refresh_ip_choose_ipv4(self):
nic1 = Mock()
nic1.network = 'A Network'
nic1.ipAddress = ['192.168.1.1']
nic2 = Mock()
nic2.network = 'A Network'
nic2.ipAddress = ['2001:0db8:0a0b:12f0:0000:0000:0000:0001']
guest = Mock()
guest.toolsStatus = 'toolsOk'
guest.net = [nic1, nic2]
vm = Mock()
vm.guest = guest
pyvmomi_service = Mock()
pyvmomi_service.find_by_uuid = Mock(return_value=vm)
ip_regex = self._create_custom_param('ip_regex', '')
refresh_ip_timeout = self._create_custom_param('refresh_ip_timeout',
'10')
resource_instance = create_autospec(ResourceInfo)
resource_instance.ResourceModelName = 'Generic Deployed App'
resource_instance.ResourceAttributes = {
'vm_uuis': '123',
'cloud_provider': 'vCenter'
}
resource_instance.VmDetails = create_autospec(ResourceInfoVmDetails)
resource_instance.VmDetails.VmCustomParams = [
ip_regex, refresh_ip_timeout
]
refresh_ip_command = RefreshIpCommand(pyvmomi_service,
ResourceModelParser(), Mock())
session = Mock()
session.UpdateResourceAddress = Mock(return_value=True)
session.GetResourceDetails = Mock(return_value=resource_instance)
si = Mock()
center_resource_model = VMwarevCenterResourceModel()
center_resource_model.default_datacenter = 'QualiSB'
center_resource_model.holding_network = 'anetwork'
cancellation_context = Mock()
# Act
refresh_ip_command.refresh_ip(
si=si,
session=session,
vcenter_data_model=center_resource_model,
vm_uuid='machine1',
resource_name='default_network',
cancellation_context=cancellation_context,
logger=Mock())
# Assert
self.assertTrue(
session.UpdateResourceAddress.called_with('machine1',
'192.168.1.1'))
def integrationtest_refresh_ip(self):
resource_context = Mock()
resource_context.attributes = {"vCenter Template": "vCenter/Alex/test"}
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('vCenter-Server-Appliance-yaniv')
refresh_ip_command.refresh_ip(uuid, '')
pass
def test_refresh_ip_choose_ip_by_regex(self):
nic1 = Mock()
nic1.network = 'A Network'
nic1.ipAddress = ['192.168.1.1']
nic2 = Mock()
nic2.network = 'A Network'
nic2.ipAddress = ['111.111.111.111']
guest = Mock()
guest.toolsStatus = 'toolsOk'
guest.net = [nic1, nic2]
vm = Mock()
vm.guest = guest
pyvmomi_service = Mock()
pyvmomi_service.find_by_uuid = Mock(return_value=vm)
ip_regex = self._create_custom_param('ip_regex', '192\.168\..*')
refresh_ip_timeout = self._create_custom_param('refresh_ip_timeout',
'10')
resource_model = create_autospec(GenericDeployedAppResourceModel)
resource_model.fullname = 'Generic Deployed App'
resource_model.vm_uuid = '123',
resource_model.cloud_provider = 'vCenter'
resource_model.vm_custom_params = [ip_regex, refresh_ip_timeout]
refresh_ip_command = RefreshIpCommand(pyvmomi_service,
ResourceModelParser(), Mock())
session = Mock()
session.UpdateResourceAddress = Mock(return_value=True)
session.GetResourceDetails = Mock(return_value=resource_model)
si = Mock()
center_resource_model = VMwarevCenterResourceModel()
center_resource_model.default_datacenter = 'QualiSB'
center_resource_model.holding_network = 'anetwork'
cancellation_context = Mock()
# Act
refresh_ip_command.refresh_ip(
si=si,
session=session,
vcenter_data_model=center_resource_model,
resource_model=resource_model,
cancellation_context=cancellation_context,
logger=Mock(),
app_request_json=Mock())
# Assert
self.assertTrue(
session.UpdateResourceAddress.called_with('machine1',
'192.168.1.1'))
def test_refresh_ip_choose_ipv4(self):
nic1 = Mock()
nic1.network = 'A Network'
nic1.ipAddress = ['192.168.1.1']
nic2 = Mock()
nic2.network = 'A Network'
nic2.ipAddress = ['2001:0db8:0a0b:12f0:0000:0000:0000:0001']
guest = Mock()
guest.toolsStatus = 'toolsOk'
guest.net = [nic1, nic2]
vm = Mock()
vm.guest = guest
pyvmomi_service = Mock()
pyvmomi_service.find_by_uuid = Mock(return_value=vm)
ip_regex = self._create_custom_param('ip_regex', '')
refresh_ip_timeout = self._create_custom_param('refresh_ip_timeout', '10')
resource_instance = create_autospec(ResourceInfo)
resource_instance.ResourceModelName = 'Generic Deployed App'
resource_instance.ResourceAttributes = {'vm_uuis': '123',
'cloud_provider': 'vCenter'
}
resource_instance.VmDetails = create_autospec(ResourceInfoVmDetails)
resource_instance.VmDetails.VmCustomParams = [ip_regex, refresh_ip_timeout]
refresh_ip_command = RefreshIpCommand(pyvmomi_service, ResourceModelParser(), Mock())
session = Mock()
session.UpdateResourceAddress = Mock(return_value=True)
session.GetResourceDetails = Mock(return_value=resource_instance)
si = Mock()
center_resource_model = VMwarevCenterResourceModel()
center_resource_model.default_datacenter = 'QualiSB'
center_resource_model.holding_network = 'anetwork'
cancellation_context = Mock()
# Act
refresh_ip_command.refresh_ip(
si=si,
session=session,
vcenter_data_model=center_resource_model,
vm_uuid='machine1',
resource_name='default_network',
cancellation_context=cancellation_context,
logger=Mock(),
app_request_json=Mock())
# Assert
self.assertTrue(session.UpdateResourceAddress.called_with('machine1', '192.168.1.1'))
def test_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)
si = py_vmomi_service.connect(credentials.host, credentials.username,
credentials.password)
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('VM Deployment_afee8038')
ip = refresh_ip_command.refresh_ip(
si, '4222f613-e714-6204-8095-e282796985fb',
'VM Deployment_afee8038')
print ip
pass
