def setUp(self):
self.portgroup_name = DvPortGroupNameGenerator()
self.connector = Mock()
self.disconnector = Mock()
self.model_parser = Mock()
self.vc_data_model = Mock()
self.vc_data_model.reserved_networks = 'restricted_network1,restricted_network2'
self.vc_data_model.default_dvswitch = 'dvSwitch'
self.vc_data_model.default_network = 'Anetwork'
self.vc_data_model.default_datacenter = 'datacenter'
self.vc_data_model.holding_network = 'Holding Network'
self.si = Mock()
self.ConnectionCommandOrchestrator = ConnectionCommandOrchestrator(
self.connector, self.disconnector, self.model_parser)
def setUp(self):
self.portgroup_name = DvPortGroupNameGenerator()
self.connector = Mock()
self.disconnector = Mock()
self.model_parser = Mock()
self.vc_data_model = Mock()
self.vc_data_model.reserved_networks = 'restricted_network1,restricted_network2'
self.vc_data_model.default_dvswitch = 'dvSwitch'
self.vc_data_model.default_network = 'Anetwork'
self.vc_data_model.default_datacenter = 'datacenter'
self.vc_data_model.holding_network = 'Holding Network'
self.si = Mock()
self.ConnectionCommandOrchestrator = ConnectionCommandOrchestrator(self.connector,
self.disconnector,
self.model_parser)
class TestCommandOrchestrator(TestCase):
def setUp(self):
self.portgroup_name = DvPortGroupNameGenerator()
self.connector = Mock()
self.disconnector = Mock()
self.model_parser = Mock()
self.vc_data_model = Mock()
self.vc_data_model.reserved_networks = 'restricted_network1,restricted_network2'
self.vc_data_model.default_dvswitch = 'dvSwitch'
self.vc_data_model.default_network = 'Anetwork'
self.vc_data_model.default_datacenter = 'datacenter'
self.vc_data_model.holding_network = 'Holding Network'
self.vc_data_model.default_port_group_location = '/'
self.si = Mock()
self.ConnectionCommandOrchestrator = ConnectionCommandOrchestrator(self.connector,
self.disconnector,
self.model_parser)
def test_connect_bulk1(self):
"""
Simple connect vm to vlan 'Access' mode
"""
request, expected = self._get_test1_params()
results = self.ConnectionCommandOrchestrator.connect_bulk(si=self.si,
logger=Mock(),
vcenter_data_model=self.vc_data_model,
request=request)
self._assert_as_expected(results, expected)
def test_connect_bulk2(self):
"""
Simple disconnect
"""
request, expected = self._get_test2_params()
results = self.ConnectionCommandOrchestrator.connect_bulk(si=self.si,
logger=Mock(),
vcenter_data_model=self.vc_data_model,
request=request)
self._assert_as_expected(results, expected)
def test_connect_bulk3(self):
"""
Complex connect multiple vms and multiple vlan types
"""
request, expected = self._get_test3_params()
results = self.ConnectionCommandOrchestrator.connect_bulk(si=self.si,
logger=Mock(),
vcenter_data_model=self.vc_data_model,
request=request)
self._assert_as_expected(results, expected)
def test_connect_bulk4(self):
"""
Simple disconnect
"""
request, expected = self._get_test4_params()
self.assertRaises(ValueError,
self.ConnectionCommandOrchestrator.connect_bulk, self.si, Mock(), self.vc_data_model, request)
def test_connect_bulk5(self):
"""
Disconnect returns error
"""
request, expected = self._get_test5_params()
res = self.ConnectionCommandOrchestrator.connect_bulk(si=self.si,
logger=Mock(),
vcenter_data_model=self.vc_data_model,
request=request)
self._assert_as_expected(res, expected)
def test_connect_bulk6(self):
"""
Connect returns error
"""
request, expected = self._get_test6_params()
res = self.ConnectionCommandOrchestrator.connect_bulk(si=self.si,
logger=Mock(),
vcenter_data_model=self.vc_data_model,
request=request)
self._assert_as_expected(res, expected)
def _assert_as_expected(self, res, exp):
for r in res:
for e in exp:
found = e.actionId == r.actionId
if found:
for prop in [prop for prop in dir(r) if not prop.startswith('__')]:
self.assertEqual(getattr(r, prop), getattr(e, prop))
break
def _set_connect_to_networks_by_request(self, request):
a = DeployDataHolder(request['driverRequest'])
res = []
for action in a.actions:
vm_uuid = self.ConnectionCommandOrchestrator._get_vm_uuid(action)
vnic_name = self.ConnectionCommandOrchestrator._validate_vnic_name(
self.ConnectionCommandOrchestrator._get_vnic_name(action))
vlan_id = action.connectionParams.vlanId
mode = action.connectionParams.mode
interface = self._get_interface_name(action)
r = ConnectionResult(mac_address=interface,
vnic_name=vnic_name,
requested_vnic=vnic_name,
vm_uuid=vm_uuid,
network_name=self.portgroup_name.generate_port_group_name(vlan_id, mode),
network_key='aa')
res.append(r)
self.connector.connect_to_networks = Mock(return_value=res)
def _set_disconnect_from_networks(self, request):
a = DeployDataHolder(request['driverRequest'])
res = []
for action in a.actions:
vnic_name = self.ConnectionCommandOrchestrator._validate_vnic_name(
self.ConnectionCommandOrchestrator._get_vnic_name(action))
interface = self._get_interface_name(action)
r = VNicDeviceMapper(vnic=Mock(), requested_vnic=vnic_name, network=Mock(), connect=False, mac=interface)
res.append(r)
self.disconnector.disconnect_from_networks = Mock(return_value=res)
def _get_interface_name(self, action):
interface_attributes = [attr.attributeValue
for attr in action.connectorAttributes
if attr.attributeName == 'Interface']
interface = 'aa' if not interface_attributes else interface_attributes[0]
return interface
def _get_connect_excepted_results(self, request, error_msg=None):
a = DeployDataHolder(request['driverRequest'])
res = []
for action in a.actions:
r = ActionResult()
r.actionId = action.actionId
r.type = action.type
r.errorMessage = error_msg
r.infoMessage = 'VLAN successfully set' if not error_msg else None
r.success = True if not error_msg else False
r.updatedInterface = self._get_interface_name(action)
res.append(r)
return res
def _get_disconnect_excepted_results(self, request, error_msg=None):
a = DeployDataHolder(request['driverRequest'])
res = []
for action in a.actions:
r = ActionResult()
r.actionId = action.actionId
r.type = action.type
r.errorMessage = error_msg
r.infoMessage = 'VLAN Successfully removed' if not error_msg else None
r.success = True if not error_msg else False
r.updatedInterface = self._get_interface_name(action)
res.append(r)
return res
def _get_test1_params(self):
request = {
'driverRequest': {
'actions': [
{
"connectorAttributes": [],
"connectionParams": {
"vlanId": "2",
"mode": "Access",
"vlanServiceAttributes": [
{
"attributeName": "Access Mode",
"attributeValue": "Access",
"type": "vlanServiceAttribute"
}
],
"type": "setVlanParameter"
},
"actionId": "ee8a3dc8-eb4b-4141-92ad-58bbd8430cad_376046bb-1ef0-4ecc-bb52-c0c7a9c75b1c",
"actionTarget": {
"fullName": "VM Deployment1_9602ad34",
"fullAddress": "N/A/NA",
"type": "actionTarget"
},
"customActionAttributes": [
{
"attributeName": "VM_UUID",
"attributeValue": "42220ae6-2fa8-b4cd-14e4-16fbad2798f6",
"type": "customAttribute"
}
],
"type": "setVlan"
}
]
}
}
self._set_connect_to_networks_by_request(request)
expected = self._get_connect_excepted_results(request)
return jsonpickle.encode(request), expected
def _get_test2_params(self):
request = {
"driverRequest": {
"actions": [
{
"connectionId": "d5369772-66de-4003-85cc-e94a57c20f1e",
"connectionParams": {
"vlanId": "654",
"mode": "Access",
"vlanServiceAttributes": [
{
"attributeName": "Access Mode",
"attributeValue": "Access",
"type": "vlanServiceAttribute"
},
{
"attributeName": "VLAN ID",
"attributeValue": "654",
"type": "vlanServiceAttribute"
},
{
"attributeName": "Virtual Network",
"attributeValue": "654",
"type": "vlanServiceAttribute"
}
],
"type": "setVlanParameter"
},
"connectorAttributes": [
{
"attributeName": "Interface",
"attributeValue": "00:50:56:a2:5f:07",
"type": "connectorAttribute"
}
],
"actionId": "d5369772-66de-4003-85cc-e94a57c20f1e_30fb5d10-1ce9-4e9f-b41e-ce0d271fd7ab",
"actionTarget": {
"fullName": "Temp4_a443da02",
"fullAddress": "192.168.65.33",
"type": "actionTarget"
},
"customActionAttributes": [
{
"attributeName": "VM_UUID",
"attributeValue": "422279bf-5899-789f-ca44-2c62075b3d2d",
"type": "customAttribute"
}
],
"type": "removeVlan"
}
]
}
}
self._set_disconnect_from_networks(request)
expected = self._get_disconnect_excepted_results(request)
return jsonpickle.encode(request), expected
def _get_test3_params(self):
request = {
'driverRequest': {
'actions': [
{
"connectorAttributes": [],
"connectionParams": {
"vlanId": "2",
"mode": "Access",
"vlanServiceAttributes": [
{
"attributeName": "Access Mode",
"attributeValue": "Access",
"type": "vlanServiceAttribute"
}
],
"type": "setVlanParameter"
},
"actionId": "1",
"actionTarget": {
"fullName": "VM Deployment1_9602ad34",
"fullAddress": "N/A/NA",
"type": "actionTarget"
},
"customActionAttributes": [
{
"attributeName": "VM_UUID",
"attributeValue": "42220ae6-2fa8-b4cd-14e4-16fbad2798f6",
"type": "customAttribute"
}
],
"type": "setVlan"
},
{
"connectorAttributes": [],
"connectionParams": {
"vlanId": "2-10",
"mode": "Access",
"vlanServiceAttributes": [
{
"attributeName": "Access Mode",
"attributeValue": "Trunk",
"type": "vlanServiceAttribute"
}
],
"type": "setVlanParameter"
},
"actionId": "2",
"actionTarget": {
"fullName": "VM Deployment1_9602ad34",
"fullAddress": "N/A/NA",
"type": "actionTarget"
},
"customActionAttributes": [
{
"attributeName": "VM_UUID",
"attributeValue": "42220ae6-2fa8-b4cd-14e4-16fbad2798f6",
"type": "customAttribute"
}
],
"type": "setVlan"
},
{
"connectorAttributes": [],
"connectionParams": {
"vlanId": "2",
"mode": "Access",
"vlanServiceAttributes": [
{
"attributeName": "Access Mode",
"attributeValue": "Access",
"type": "vlanServiceAttribute"
}
],
"type": "setVlanParameter"
},
"actionId": "3",
"actionTarget": {
"fullName": "VM Deployment1_9602ad34",
"fullAddress": "N/A/NA",
"type": "actionTarget"
},
"customActionAttributes": [
{
"attributeName": "VM_UUID",
"attributeValue": "42220ae6-2fa8-b4cd-14e4-16fbad2798f6",
"type": "customAttribute"
},
{
"attributeName": "Vnic Name",
"attributeValue": "1",
"type": "customAttribute"
}
],
"type": "setVlan"
},
{
"connectorAttributes": [],
"connectionParams": {
"vlanId": "2",
"mode": "Access",
"vlanServiceAttributes": [
{
"attributeName": "Access Mode",
"attributeValue": "Access",
"type": "vlanServiceAttribute"
}
],
"type": "setVlanParameter"
},
"actionId": "4",
"actionTarget": {
"fullName": "VM Deployment1_9602ad34",
"fullAddress": "N/A/NA",
"type": "actionTarget"
},
"customActionAttributes": [
{
"attributeName": "VM_UUID",
"attributeValue": "1234",
"type": "customAttribute"
},
{
"attributeName": "Vnic Name",
"attributeValue": "1",
"type": "customAttribute"
}
],
"type": "setVlan"
}
]
}
}
self._set_connect_to_networks_by_request(request)
expected = self._get_connect_excepted_results(request)
return jsonpickle.encode(request), expected
def _get_test4_params(self):
request = {
"driverRequest": {
"actions": [
{
"connectionId": "d5369772-66de-4003-85cc-e94a57c20f1e",
"connectionParams": {
"vlanId": "654",
"mode": "Access",
"vlanServiceAttributes": [
{
"attributeName": "Access Mode",
"attributeValue": "Access",
"type": "vlanServiceAttribute"
},
{
"attributeName": "VLAN ID",
"attributeValue": "654",
"type": "vlanServiceAttribute"
},
{
"attributeName": "Virtual Network",
"attributeValue": "654",
"type": "vlanServiceAttribute"
}
],
"type": "setVlanParameter"
},
"connectorAttributes": [
{
"attributeName": "Interface",
"attributeValue": "00:50:56:a2:5f:07",
"type": "connectorAttribute"
}
],
"actionId": "d5369772-66de-4003-85cc-e94a57c20f1e_30fb5d10-1ce9-4e9f-b41e-ce0d271fd7ab",
"actionTarget": {
"fullName": "Temp4_a443da02",
"fullAddress": "192.168.65.33",
"type": "actionTarget"
},
"customActionAttributes": [
],
"type": "removeVlan"
}
]
}
}
self._set_disconnect_from_networks(request)
expected = self._get_disconnect_excepted_results(request, 'VM_UUID is missing on action attributes')
return jsonpickle.encode(request), expected
def _get_test5_params(self):
request = {
"driverRequest": {
"actions": [
{
"connectionId": "d5369772-66de-4003-85cc-e94a57c20f1e",
"connectionParams": {
"vlanId": "654",
"mode": "Access",
"vlanServiceAttributes": [
{
"attributeName": "Access Mode",
"attributeValue": "Access",
"type": "vlanServiceAttribute"
},
{
"attributeName": "VLAN ID",
"attributeValue": "654",
"type": "vlanServiceAttribute"
},
{
"attributeName": "Virtual Network",
"attributeValue": "654",
"type": "vlanServiceAttribute"
}
],
"type": "setVlanParameter"
},
"connectorAttributes": [
{
"attributeName": "Interface",
"attributeValue": "00:50:56:a2:5f:07",
"type": "connectorAttribute"
}
],
"actionId": "d5369772-66de-4003-85cc-e94a57c20f1e_30fb5d10-1ce9-4e9f-b41e-ce0d271fd7ab",
"actionTarget": {
"fullName": "Temp4_a443da02",
"fullAddress": "192.168.65.33",
"type": "actionTarget"
},
"customActionAttributes": [
{
"attributeName": "VM_UUID",
"attributeValue": "42220ae6-2fa8-b4cd-14e4-16fbad2798f6",
"type": "customAttribute"
}
],
"type": "removeVlan"
}
]
}
}
self.disconnector.disconnect_from_networks = Mock()
self.disconnector.disconnect_from_networks.side_effect = ValueError('vnic not found')
expected = self._get_disconnect_excepted_results(request, 'vnic not found')
return jsonpickle.encode(request), expected
def _get_test6_params(self):
request = {
'driverRequest': {
'actions': [
{
"connectorAttributes": [],
"connectionParams": {
"vlanId": "2",
"mode": "Access",
"vlanServiceAttributes": [
{
"attributeName": "Access Mode",
"attributeValue": "Access",
"type": "vlanServiceAttribute"
}
],
"type": "setVlanParameter"
},
"actionId": "ee8a3dc8-eb4b-4141-92ad-58bbd8430cad_376046bb-1ef0-4ecc-bb52-c0c7a9c75b1c",
"actionTarget": {
"fullName": "VM Deployment1_9602ad34",
"fullAddress": "N/A/NA",
"type": "actionTarget"
},
"customActionAttributes": [
{
"attributeName": "VM_UUID",
"attributeValue": "42220ae6-2fa8-b4cd-14e4-16fbad2798f6",
"type": "customAttribute"
}
],
"type": "setVlan"
}
]
}
}
self.connector.connect_to_networks = Mock()
self.connector.connect_to_networks.side_effect = ValueError('vnic not found')
expected = self._get_connect_excepted_results(request, 'vnic not found')
# here we don't don't get vnic input input
for e in expected:
e.updatedInterface = None
return jsonpickle.encode(request), expected
def __init__(self):
"""
Initialize the driver session, this function is called everytime a new instance of the driver is created
in here the driver is going to be bootstrapped
:param context: models.QualiDriverModels.InitCommandContext
"""
self.cs_helper = CloudshellDriverHelper()
synchronous_task_waiter = SynchronousTaskWaiter()
pv_service = pyVmomiService(connect=SmartConnect,
disconnect=Disconnect,
task_waiter=synchronous_task_waiter)
self.resource_model_parser = ResourceModelParser()
port_group_name_generator = DvPortGroupNameGenerator()
vnic_to_network_mapper = VnicToNetworkMapper(
quali_name_generator=port_group_name_generator)
resource_remover = CloudshellResourceRemover()
ovf_service = OvfImageDeployerService(self.resource_model_parser)
self.vm_loader = VMLoader(pv_service)
vm_deployer = VirtualMachineDeployer(
pv_service=pv_service,
name_generator=generate_unique_name,
ovf_service=ovf_service,
cs_helper=self.cs_helper,
resource_model_parser=ResourceModelParser())
dv_port_group_creator = DvPortGroupCreator(
pyvmomi_service=pv_service,
synchronous_task_waiter=synchronous_task_waiter)
virtual_machine_port_group_configurer = \
VirtualMachinePortGroupConfigurer(pyvmomi_service=pv_service,
synchronous_task_waiter=synchronous_task_waiter,
vnic_to_network_mapper=vnic_to_network_mapper,
vnic_service=VNicService(),
name_gen=port_group_name_generator)
virtual_switch_to_machine_connector = VirtualSwitchToMachineConnector(
dv_port_group_creator, virtual_machine_port_group_configurer)
# Command Wrapper
self.command_wrapper = CommandWrapper(
pv_service=pv_service,
cloud_shell_helper=self.cs_helper,
resource_model_parser=self.resource_model_parser,
context_based_logger_factory=ContextBasedLoggerFactory())
# Deploy Command
self.deploy_command = DeployCommand(deployer=vm_deployer)
# Virtual Switch Revoke
self.virtual_switch_disconnect_command = \
VirtualSwitchToMachineDisconnectCommand(
pyvmomi_service=pv_service,
port_group_configurer=virtual_machine_port_group_configurer,
resource_model_parser=self.resource_model_parser)
# Virtual Switch Connect
virtual_switch_connect_command = \
VirtualSwitchConnectCommand(
pv_service=pv_service,
virtual_switch_to_machine_connector=virtual_switch_to_machine_connector,
dv_port_group_name_generator=DvPortGroupNameGenerator(),
vlan_spec_factory=VlanSpecFactory(),
vlan_id_range_parser=VLanIdRangeParser())
self.connection_orchestrator = ConnectionCommandOrchestrator(
connector=virtual_switch_connect_command,
disconnector=self.virtual_switch_disconnect_command,
resource_model_parser=self.resource_model_parser)
# Destroy VM Command
self.destroy_virtual_machine_command = \
DestroyVirtualMachineCommand(pv_service=pv_service,
resource_remover=resource_remover,
disconnector=self.virtual_switch_disconnect_command)
# Power Command
self.vm_power_management_command = \
VirtualMachinePowerManagementCommand(pyvmomi_service=pv_service,
synchronous_task_waiter=synchronous_task_waiter)
ip_manager = VMIPManager()
# Refresh IP command
self.refresh_ip_command = RefreshIpCommand(
pyvmomi_service=pv_service,
resource_model_parser=ResourceModelParser(),
ip_manager=ip_manager)
# Save Snapshot
self.snapshot_saver = SaveSnapshotCommand(
pyvmomi_service=pv_service, task_waiter=synchronous_task_waiter)
# Snapshot Restorer
self.snapshot_restorer = SnapshotRestoreCommand(
pyvmomi_service=pv_service, task_waiter=synchronous_task_waiter)
self.snapshots_retriever = RetrieveSnapshotsCommand(
pyvmomi_service=pv_service)
class TestCommandOrchestrator(TestCase):
def setUp(self):
self.portgroup_name = DvPortGroupNameGenerator()
self.connector = Mock()
self.disconnector = Mock()
self.model_parser = Mock()
self.vc_data_model = Mock()
self.vc_data_model.reserved_networks = 'restricted_network1,restricted_network2'
self.vc_data_model.default_dvswitch = 'dvSwitch'
self.vc_data_model.default_network = 'Anetwork'
self.vc_data_model.default_datacenter = 'datacenter'
self.vc_data_model.holding_network = 'Holding Network'
self.si = Mock()
self.ConnectionCommandOrchestrator = ConnectionCommandOrchestrator(
self.connector, self.disconnector, self.model_parser)
def test_connect_bulk_dvswitch_is_None(self):
"""
tests the error by missing dvswitch
"""
vc_data_model = Mock()
vc_data_model.reserved_networks = 'restricted_network1,restricted_network2'
vc_data_model.default_dvswitch = None
vc_data_model.default_network = 'Anetwork'
vc_data_model.default_datacenter = 'datacenter'
vc_data_model.holding_network = 'Holding Network'
request, expected = self._get_missing_dv_switch_params()
results = self.ConnectionCommandOrchestrator.connect_bulk(
si=self.si,
logger=Mock(),
vcenter_data_model=vc_data_model,
request=request)
self._assert_as_expected(results, expected)
def test_connect_bulk1(self):
"""
Simple connect vm to vlan 'Access' mode
"""
request, expected = self._get_test1_params()
results = self.ConnectionCommandOrchestrator.connect_bulk(
si=self.si,
logger=Mock(),
vcenter_data_model=self.vc_data_model,
request=request)
self._assert_as_expected(results, expected)
def test_connect_bulk2(self):
"""
Simple disconnect
"""
request, expected = self._get_test2_params()
results = self.ConnectionCommandOrchestrator.connect_bulk(
si=self.si,
logger=Mock(),
vcenter_data_model=self.vc_data_model,
request=request)
self._assert_as_expected(results, expected)
def test_connect_bulk3(self):
"""
Complex connect multiple vms and multiple vlan types
"""
request, expected = self._get_test3_params()
results = self.ConnectionCommandOrchestrator.connect_bulk(
si=self.si,
logger=Mock(),
vcenter_data_model=self.vc_data_model,
request=request)
self._assert_as_expected(results, expected)
def test_connect_bulk4(self):
"""
Simple disconnect
"""
request, expected = self._get_test4_params()
self.assertRaises(ValueError,
self.ConnectionCommandOrchestrator.connect_bulk,
self.si, Mock(), self.vc_data_model, request)
def test_connect_bulk5(self):
"""
Disconnect returns error
"""
request, expected = self._get_test5_params()
res = self.ConnectionCommandOrchestrator.connect_bulk(
si=self.si,
logger=Mock(),
vcenter_data_model=self.vc_data_model,
request=request)
self._assert_as_expected(res, expected)
def test_connect_bulk6(self):
"""
Connect returns error
"""
request, expected = self._get_test6_params()
res = self.ConnectionCommandOrchestrator.connect_bulk(
si=self.si,
logger=Mock(),
vcenter_data_model=self.vc_data_model,
request=request)
self._assert_as_expected(res, expected)
def test_connect_bulk7(self):
"""
Simple connect vm to vlan 'Access' mode
"""
request, expected = self._get_test7_params()
results = self.ConnectionCommandOrchestrator.connect_bulk(
si=self.si,
logger=Mock(),
vcenter_data_model=self.vc_data_model,
request=request)
self._assert_as_expected(results, expected)
def test_connect_bulk8(self):
"""
Simple connect vm to vlan 'Access' mode
"""
request, expected = self._get_test8_params()
results = self.ConnectionCommandOrchestrator.connect_bulk(
si=self.si,
logger=Mock(),
vcenter_data_model=self.vc_data_model,
request=request)
self._assert_as_expected(results, expected)
def test_connect_bulk9(self):
"""
Simple disconnect
"""
request, expected = self._get_test9_params()
results = self.ConnectionCommandOrchestrator.connect_bulk(
si=self.si,
logger=Mock(),
vcenter_data_model=self.vc_data_model,
request=request)
self._assert_as_expected(results, expected)
def _assert_as_expected(self, res, exp):
for r in res:
for e in exp:
found = e.actionId == r.actionId
if found:
for prop in [
prop for prop in dir(r)
if not prop.startswith('__')
]:
self.assertEqual(getattr(r, prop), getattr(e, prop))
break
def _set_connect_to_networks_by_request(self, request):
a = DeployDataHolder(request['driverRequest'])
res = []
for action in a.actions:
vm_uuid = self.ConnectionCommandOrchestrator._get_vm_uuid(action)
vnic_name = self.ConnectionCommandOrchestrator._validate_vnic_name(
self.ConnectionCommandOrchestrator._get_vnic_name(action))
if vnic_name:
vnic_name = vnic_name.strip().split(',')
else:
vnic_name = [None]
for vnic in vnic_name:
vlan_id = action.connectionParams.vlanId
mode = action.connectionParams.mode
interface = self._get_interface_name(action)
name = 'Network adapter {0}'.format(
vnic) if vnic and vnic.isdigit() else vnic
r = ConnectionResult(
mac_address=interface,
vnic_name=name,
requested_vnic=name,
vm_uuid=vm_uuid,
network_name=self.portgroup_name.generate_port_group_name(
'dvSwitch', vlan_id, mode),
network_key='aa')
res.append(r)
self.connector.connect_to_networks = Mock(return_value=res)
def _set_disconnect_from_networks(self, request):
a = DeployDataHolder(request['driverRequest'])
res = []
for action in a.actions:
vnic_name = self.ConnectionCommandOrchestrator._validate_vnic_name(
self.ConnectionCommandOrchestrator._get_vnic_name(action))
interface = self._get_interface_name(action)
r = VNicDeviceMapper(vnic=Mock(),
requested_vnic=vnic_name,
network=Mock(),
connect=False,
mac=interface)
res.append(r)
self.disconnector.disconnect_from_networks = Mock(return_value=res)
def _get_interface_name_for_result(self, action):
interface_attributes = [
attr.attributeValue for attr in action.connectorAttributes
if attr.attributeName == 'Interface'
]
vnics = [
attr.attributeValue for attr in action.connectorAttributes
if attr.attributeName == 'Requested Source vNIC Name'
]
if vnics:
vnics = vnics[0].strip().split(',')
interface = 'aa'
for vnic in vnics[1:]:
interface = '{0},{1}'.format(
interface,
'aa' if not interface_attributes else interface_attributes[0])
return interface
def _get_interface_name(self, action):
interface_attributes = [
attr.attributeValue for attr in action.connectorAttributes
if attr.attributeName == 'Interface'
]
interface = 'aa' if not interface_attributes else interface_attributes[
0]
return interface
def _get_connect_excepted_results(self,
request,
error_msg=None,
ignore_interface=False):
a = DeployDataHolder(request['driverRequest'])
res = []
for action in a.actions:
r = ActionResult()
r.actionId = action.actionId
r.type = action.type
r.errorMessage = error_msg
r.infoMessage = 'VLAN successfully set' if not error_msg else None
r.success = True if not error_msg else False
r.updatedInterface = None if ignore_interface else self._get_interface_name_for_result(
action)
res.append(r)
return res
def _get_disconnect_excepted_results(self, request, error_msg=None):
a = DeployDataHolder(request['driverRequest'])
res = []
for action in a.actions:
r = ActionResult()
r.actionId = action.actionId
r.type = action.type
r.errorMessage = error_msg
r.infoMessage = 'VLAN Successfully removed' if not error_msg else None
r.success = True if not error_msg else False
r.updatedInterface = self._get_interface_name(action)
res.append(r)
return res
def _get_missing_dv_switch_params(self):
request = {
'driverRequest': {
'actions': [{
"connectorAttributes": [],
"connectionParams": {
"vlanId":
"2",
"mode":
"Access",
"vlanServiceAttributes": [{
"attributeName":
"Access Mode",
"attributeValue":
"Access",
"type":
"vlanServiceAttribute"
}],
"type":
"setVlanParameter"
},
"actionId":
"ee8a3dc8-eb4b-4141-92ad-58bbd8430cad_376046bb-1ef0-4ecc-bb52-c0c7a9c75b1c",
"actionTarget": {
"fullName": "VM Deployment1_9602ad34",
"fullAddress": "N/A/NA",
"type": "actionTarget"
},
"customActionAttributes": [{
"attributeName": "VM_UUID",
"attributeValue":
"42220ae6-2fa8-b4cd-14e4-16fbad2798f6",
"type": "customAttribute"
}],
"type":
"setVlan"
}]
}
}
self._set_connect_to_networks_by_request(request)
expected = self._get_connect_excepted_results(
request,
error_msg=
'Please set the attribute "Default DvSwitch" in order to execute any connectivity changes',
ignore_interface=True)
return jsonpickle.encode(request), expected
def _get_test1_params(self):
request = {
'driverRequest': {
'actions': [{
"connectorAttributes": [],
"connectionParams": {
"vlanId":
"2",
"mode":
"Access",
"vlanServiceAttributes": [{
"attributeName":
"Access Mode",
"attributeValue":
"Access",
"type":
"vlanServiceAttribute"
}],
"type":
"setVlanParameter"
},
"actionId":
"ee8a3dc8-eb4b-4141-92ad-58bbd8430cad_376046bb-1ef0-4ecc-bb52-c0c7a9c75b1c",
"actionTarget": {
"fullName": "VM Deployment1_9602ad34",
"fullAddress": "N/A/NA",
"type": "actionTarget"
},
"customActionAttributes": [{
"attributeName": "VM_UUID",
"attributeValue":
"42220ae6-2fa8-b4cd-14e4-16fbad2798f6",
"type": "customAttribute"
}],
"type":
"setVlan"
}]
}
}
self._set_connect_to_networks_by_request(request)
expected = self._get_connect_excepted_results(request)
return jsonpickle.encode(request), expected
def _get_test2_params(self):
request = {
"driverRequest": {
"actions": [{
"connectionId":
"d5369772-66de-4003-85cc-e94a57c20f1e",
"connectionParams": {
"vlanId":
"654",
"mode":
"Access",
"vlanServiceAttributes": [{
"attributeName":
"Access Mode",
"attributeValue":
"Access",
"type":
"vlanServiceAttribute"
}, {
"attributeName":
"VLAN ID",
"attributeValue":
"654",
"type":
"vlanServiceAttribute"
}, {
"attributeName":
"Virtual Network",
"attributeValue":
"654",
"type":
"vlanServiceAttribute"
}],
"type":
"setVlanParameter"
},
"connectorAttributes": [{
"attributeName": "Interface",
"attributeValue": "00:50:56:a2:5f:07",
"type": "connectorAttribute"
}],
"actionId":
"d5369772-66de-4003-85cc-e94a57c20f1e_30fb5d10-1ce9-4e9f-b41e-ce0d271fd7ab",
"actionTarget": {
"fullName": "Temp4_a443da02",
"fullAddress": "192.168.65.33",
"type": "actionTarget"
},
"customActionAttributes": [{
"attributeName": "VM_UUID",
"attributeValue":
"422279bf-5899-789f-ca44-2c62075b3d2d",
"type": "customAttribute"
}],
"type":
"removeVlan"
}]
}
}
self._set_disconnect_from_networks(request)
expected = self._get_disconnect_excepted_results(request)
return jsonpickle.encode(request), expected
def _get_test3_params(self):
request = {
'driverRequest': {
'actions': [{
"connectorAttributes": [],
"connectionParams": {
"vlanId":
"2",
"mode":
"Access",
"vlanServiceAttributes": [{
"attributeName":
"Access Mode",
"attributeValue":
"Access",
"type":
"vlanServiceAttribute"
}],
"type":
"setVlanParameter"
},
"actionId":
"1",
"actionTarget": {
"fullName": "VM Deployment1_9602ad34",
"fullAddress": "N/A/NA",
"type": "actionTarget"
},
"customActionAttributes": [{
"attributeName": "VM_UUID",
"attributeValue":
"42220ae6-2fa8-b4cd-14e4-16fbad2798f6",
"type": "customAttribute"
}],
"type":
"setVlan"
},
{
"connectorAttributes": [],
"connectionParams": {
"vlanId":
"2-10",
"mode":
"Access",
"vlanServiceAttributes": [{
"attributeName":
"Access Mode",
"attributeValue":
"Trunk",
"type":
"vlanServiceAttribute"
}],
"type":
"setVlanParameter"
},
"actionId":
"2",
"actionTarget": {
"fullName": "VM Deployment1_9602ad34",
"fullAddress": "N/A/NA",
"type": "actionTarget"
},
"customActionAttributes": [{
"attributeName":
"VM_UUID",
"attributeValue":
"42220ae6-2fa8-b4cd-14e4-16fbad2798f6",
"type":
"customAttribute"
}],
"type":
"setVlan"
}, {
"connectorAttributes": [],
"connectionParams": {
"vlanId":
"2",
"mode":
"Access",
"vlanServiceAttributes": [{
"attributeName":
"Access Mode",
"attributeValue":
"Access",
"type":
"vlanServiceAttribute"
}],
"type":
"setVlanParameter"
},
"actionId":
"3",
"actionTarget": {
"fullName": "VM Deployment1_9602ad34",
"fullAddress": "N/A/NA",
"type": "actionTarget"
},
"customActionAttributes": [{
"attributeName":
"VM_UUID",
"attributeValue":
"42220ae6-2fa8-b4cd-14e4-16fbad2798f6",
"type":
"customAttribute"
}, {
"attributeName":
"Vnic Name",
"attributeValue":
"1",
"type":
"customAttribute"
}],
"type":
"setVlan"
}, {
"connectorAttributes": [],
"connectionParams": {
"vlanId":
"2",
"mode":
"Access",
"vlanServiceAttributes": [{
"attributeName":
"Access Mode",
"attributeValue":
"Access",
"type":
"vlanServiceAttribute"
}],
"type":
"setVlanParameter"
},
"actionId":
"4",
"actionTarget": {
"fullName": "VM Deployment1_9602ad34",
"fullAddress": "N/A/NA",
"type": "actionTarget"
},
"customActionAttributes": [{
"attributeName":
"VM_UUID",
"attributeValue":
"1234",
"type":
"customAttribute"
}, {
"attributeName":
"Vnic Name",
"attributeValue":
"1",
"type":
"customAttribute"
}],
"type":
"setVlan"
}]
}
}
self._set_connect_to_networks_by_request(request)
expected = self._get_connect_excepted_results(request)
return jsonpickle.encode(request), expected
def _get_test4_params(self):
request = {
"driverRequest": {
"actions": [{
"connectionId":
"d5369772-66de-4003-85cc-e94a57c20f1e",
"connectionParams": {
"vlanId":
"654",
"mode":
"Access",
"vlanServiceAttributes": [{
"attributeName":
"Access Mode",
"attributeValue":
"Access",
"type":
"vlanServiceAttribute"
}, {
"attributeName":
"VLAN ID",
"attributeValue":
"654",
"type":
"vlanServiceAttribute"
}, {
"attributeName":
"Virtual Network",
"attributeValue":
"654",
"type":
"vlanServiceAttribute"
}],
"type":
"setVlanParameter"
},
"connectorAttributes": [{
"attributeName": "Interface",
"attributeValue": "00:50:56:a2:5f:07",
"type": "connectorAttribute"
}],
"actionId":
"d5369772-66de-4003-85cc-e94a57c20f1e_30fb5d10-1ce9-4e9f-b41e-ce0d271fd7ab",
"actionTarget": {
"fullName": "Temp4_a443da02",
"fullAddress": "192.168.65.33",
"type": "actionTarget"
},
"customActionAttributes": [],
"type":
"removeVlan"
}]
}
}
self._set_disconnect_from_networks(request)
expected = self._get_disconnect_excepted_results(
request, 'VM_UUID is missing on action attributes')
return jsonpickle.encode(request), expected
def _get_test5_params(self):
request = {
"driverRequest": {
"actions": [{
"connectionId":
"d5369772-66de-4003-85cc-e94a57c20f1e",
"connectionParams": {
"vlanId":
"654",
"mode":
"Access",
"vlanServiceAttributes": [{
"attributeName":
"Access Mode",
"attributeValue":
"Access",
"type":
"vlanServiceAttribute"
}, {
"attributeName":
"VLAN ID",
"attributeValue":
"654",
"type":
"vlanServiceAttribute"
}, {
"attributeName":
"Virtual Network",
"attributeValue":
"654",
"type":
"vlanServiceAttribute"
}],
"type":
"setVlanParameter"
},
"connectorAttributes": [{
"attributeName": "Interface",
"attributeValue": "00:50:56:a2:5f:07",
"type": "connectorAttribute"
}],
"actionId":
"d5369772-66de-4003-85cc-e94a57c20f1e_30fb5d10-1ce9-4e9f-b41e-ce0d271fd7ab",
"actionTarget": {
"fullName": "Temp4_a443da02",
"fullAddress": "192.168.65.33",
"type": "actionTarget"
},
"customActionAttributes": [{
"attributeName": "VM_UUID",
"attributeValue":
"42220ae6-2fa8-b4cd-14e4-16fbad2798f6",
"type": "customAttribute"
}],
"type":
"removeVlan"
}]
}
}
self.disconnector.disconnect_from_networks = Mock()
self.disconnector.disconnect_from_networks.side_effect = ValueError(
'vnic not found')
expected = self._get_disconnect_excepted_results(
request, 'vnic not found')
return jsonpickle.encode(request), expected
def _get_test6_params(self):
request = {
'driverRequest': {
'actions': [{
"connectorAttributes": [],
"connectionParams": {
"vlanId":
"2",
"mode":
"Access",
"vlanServiceAttributes": [{
"attributeName":
"Access Mode",
"attributeValue":
"Access",
"type":
"vlanServiceAttribute"
}],
"type":
"setVlanParameter"
},
"actionId":
"ee8a3dc8-eb4b-4141-92ad-58bbd8430cad_376046bb-1ef0-4ecc-bb52-c0c7a9c75b1c",
"actionTarget": {
"fullName": "VM Deployment1_9602ad34",
"fullAddress": "N/A/NA",
"type": "actionTarget"
},
"customActionAttributes": [{
"attributeName": "VM_UUID",
"attributeValue":
"42220ae6-2fa8-b4cd-14e4-16fbad2798f6",
"type": "customAttribute"
}],
"type":
"setVlan"
}]
}
}
self.connector.connect_to_networks = Mock()
self.connector.connect_to_networks.side_effect = ValueError(
'vnic not found')
expected = self._get_connect_excepted_results(request,
'vnic not found')
# here we don't don't get vnic input input
for e in expected:
e.updatedInterface = None
return jsonpickle.encode(request), expected
def _get_test7_params(self):
request = {
"driverRequest": {
"actions": [{
"connectionId":
"6bd00213-6cfa-435a-abf5-6c839dae0f44",
"connectionParams": {
"vlanId":
"2",
"mode":
"Access",
"vlanServiceAttributes": [{
"attributeName":
"QnQ",
"attributeValue":
"False",
"type":
"vlanServiceAttribute"
}, {
"attributeName":
"CTag",
"attributeValue":
"",
"type":
"vlanServiceAttribute"
}, {
"attributeName":
"Allocation Ranges",
"attributeValue":
"2-4094",
"type":
"vlanServiceAttribute"
}, {
"attributeName":
"Isolation Level",
"attributeValue":
"Exclusive",
"type":
"vlanServiceAttribute"
}, {
"attributeName":
"Access Mode",
"attributeValue":
"Access",
"type":
"vlanServiceAttribute"
}, {
"attributeName":
"VLAN ID",
"attributeValue":
"",
"type":
"vlanServiceAttribute"
}, {
"attributeName":
"Pool Name",
"attributeValue":
"",
"type":
"vlanServiceAttribute"
}, {
"attributeName":
"Virtual Network",
"attributeValue":
"2",
"type":
"vlanServiceAttribute"
}],
"type":
"setVlanParameter"
},
"connectorAttributes": [{
"attributeName": "Requested Source vNIC Name",
"attributeValue": "1,2",
"type": "connectorAttribute"
}],
"actionId":
"6bd00213-6cfa-435a-abf5-6c839dae0f44_c6b69177-d251-4dba-ba88-d7c5bb259505",
"actionTarget": {
"fullName": "temp_b648f80f",
"fullAddress": "NA",
"type": "actionTarget"
},
"customActionAttributes": [{
"attributeName": "auto_power_off",
"attributeValue": "auto_power_off",
"type": "customAttribute"
}, {
"attributeName": "ip_regex",
"attributeValue": "ip_regex",
"type": "customAttribute"
}, {
"attributeName": "wait_for_ip",
"attributeValue": "wait_for_ip",
"type": "customAttribute"
}, {
"attributeName": "autoload",
"attributeValue": "autoload",
"type": "customAttribute"
}, {
"attributeName": "auto_delete",
"attributeValue": "auto_delete",
"type": "customAttribute"
}, {
"attributeName": "auto_power_on",
"attributeValue": "auto_power_on",
"type": "customAttribute"
}, {
"attributeName": "refresh_ip_timeout",
"attributeValue": "refresh_ip_timeout",
"type": "customAttribute"
}, {
"attributeName": "VM_UUID",
"attributeValue":
"422203f6-eadd-9f88-5dc8-00c17f49fa21",
"type": "customAttribute"
}, {
"attributeName": "Vnic Name",
"attributeValue": "1,2",
"type": "customAttribute"
}],
"type":
"setVlan"
}]
}
}
self._set_connect_to_networks_by_request(request)
expected = self._get_connect_excepted_results(request)
return jsonpickle.encode(request), expected
def _get_test8_params(self):
request = {
"driverRequest": {
"actions": [{
"connectionId":
"6bd00213-6cfa-435a-abf5-6c839dae0f44",
"connectionParams": {
"vlanId":
"2",
"mode":
"Access",
"vlanServiceAttributes": [{
"attributeName":
"QnQ",
"attributeValue":
"False",
"type":
"vlanServiceAttribute"
}, {
"attributeName":
"CTag",
"attributeValue":
"",
"type":
"vlanServiceAttribute"
}, {
"attributeName":
"Allocation Ranges",
"attributeValue":
"2-4094",
"type":
"vlanServiceAttribute"
}, {
"attributeName":
"Isolation Level",
"attributeValue":
"Exclusive",
"type":
"vlanServiceAttribute"
}, {
"attributeName":
"Access Mode",
"attributeValue":
"Access",
"type":
"vlanServiceAttribute"
}, {
"attributeName":
"VLAN ID",
"attributeValue":
"",
"type":
"vlanServiceAttribute"
}, {
"attributeName":
"Pool Name",
"attributeValue":
"",
"type":
"vlanServiceAttribute"
}, {
"attributeName":
"Virtual Network",
"attributeValue":
"2",
"type":
"vlanServiceAttribute"
}],
"type":
"setVlanParameter"
},
"connectorAttributes": [{
"attributeName": "Requested Source vNIC Name",
"attributeValue": "1,2",
"type": "connectorAttribute"
}],
"actionId":
"6bd00213-6cfa-435a-abf5-6c839dae0f44_c6b69177-d251-4dba-ba88-d7c5bb259505",
"actionTarget": {
"fullName": "temp_b648f80f",
"fullAddress": "NA",
"type": "actionTarget"
},
"customActionAttributes": [{
"attributeName": "auto_power_off",
"attributeValue": "auto_power_off",
"type": "customAttribute"
}, {
"attributeName": "ip_regex",
"attributeValue": "ip_regex",
"type": "customAttribute"
}, {
"attributeName": "wait_for_ip",
"attributeValue": "wait_for_ip",
"type": "customAttribute"
}, {
"attributeName": "autoload",
"attributeValue": "autoload",
"type": "customAttribute"
}, {
"attributeName": "auto_delete",
"attributeValue": "auto_delete",
"type": "customAttribute"
}, {
"attributeName": "auto_power_on",
"attributeValue": "auto_power_on",
"type": "customAttribute"
}, {
"attributeName": "refresh_ip_timeout",
"attributeValue": "refresh_ip_timeout",
"type": "customAttribute"
}, {
"attributeName": "VM_UUID",
"attributeValue":
"422203f6-eadd-9f88-5dc8-00c17f49fa21",
"type": "customAttribute"
}, {
"attributeName": "Vnic Name",
"attributeValue":
"Network Adapter 1,Network Adapter 2",
"type": "customAttribute"
}],
"type":
"setVlan"
}]
}
}
self._set_connect_to_networks_by_request(request)
expected = self._get_connect_excepted_results(request)
return jsonpickle.encode(request), expected
def _get_test9_params(self):
request = {
"driverRequest": {
"actions": [{
"connectionId":
"4dd285fa-d271-4023-b7d1-881553a8f59f",
"connectionParams": {
"vlanId":
"2",
"mode":
"Access",
"vlanServiceAttributes": [{
"attributeName":
"QnQ",
"attributeValue":
"False",
"type":
"vlanServiceAttribute"
}, {
"attributeName":
"CTag",
"attributeValue":
"",
"type":
"vlanServiceAttribute"
}, {
"attributeName":
"Allocation Ranges",
"attributeValue":
"2-4094",
"type":
"vlanServiceAttribute"
}, {
"attributeName":
"Isolation Level",
"attributeValue":
"Exclusive",
"type":
"vlanServiceAttribute"
}, {
"attributeName":
"Access Mode",
"attributeValue":
"Access",
"type":
"vlanServiceAttribute"
}, {
"attributeName":
"VLAN ID",
"attributeValue":
"",
"type":
"vlanServiceAttribute"
}, {
"attributeName":
"Pool Name",
"attributeValue":
"",
"type":
"vlanServiceAttribute"
}, {
"attributeName":
"Virtual Network",
"attributeValue":
"2",
"type":
"vlanServiceAttribute"
}],
"type":
"setVlanParameter"
},
"connectorAttributes": [{
"attributeName": "Interface",
"attributeValue": "00:50:56:a2:37:3b",
"type": "connectorAttribute"
}],
"actionId":
"4dd285fa-d271-4023-b7d1-881553a8f59f_c6b69177-d251-4dba-ba88-d7c5bb259505",
"actionTarget": {
"fullName": "temp_b648f80f",
"fullAddress": "NA",
"type": "actionTarget"
},
"customActionAttributes": [{
"attributeName": "auto_power_off",
"attributeValue": "auto_power_off",
"type": "customAttribute"
}, {
"attributeName": "ip_regex",
"attributeValue": "ip_regex",
"type": "customAttribute"
}, {
"attributeName": "wait_for_ip",
"attributeValue": "wait_for_ip",
"type": "customAttribute"
}, {
"attributeName": "autoload",
"attributeValue": "autoload",
"type": "customAttribute"
}, {
"attributeName": "auto_delete",
"attributeValue": "auto_delete",
"type": "customAttribute"
}, {
"attributeName": "auto_power_on",
"attributeValue": "auto_power_on",
"type": "customAttribute"
}, {
"attributeName": "refresh_ip_timeout",
"attributeValue": "refresh_ip_timeout",
"type": "customAttribute"
}, {
"attributeName": "VM_UUID",
"attributeValue":
"422203f6-eadd-9f88-5dc8-00c17f49fa21",
"type": "customAttribute"
}],
"type":
"removeVlan"
}]
}
}
self._set_disconnect_from_networks(request)
expected = self._get_disconnect_excepted_results(request)
return jsonpickle.encode(request), expected
class TestCommandOrchestrator(TestCase):
def setUp(self):
self.portgroup_name = DvPortGroupNameGenerator()
self.connector = Mock()
self.disconnector = Mock()
self.model_parser = Mock()
self.vc_data_model = Mock()
self.vc_data_model.reserved_networks = 'restricted_network1,restricted_network2'
self.vc_data_model.default_dvswitch = 'dvSwitch'
self.vc_data_model.default_network = 'Anetwork'
self.vc_data_model.default_datacenter = 'datacenter'
self.vc_data_model.holding_network = 'Holding Network'
self.si = Mock()
self.ConnectionCommandOrchestrator = ConnectionCommandOrchestrator(self.connector,
self.disconnector,
self.model_parser)
def test_connect_bulk_dvswitch_is_None(self):
"""
tests the error by missing dvswitch
"""
vc_data_model = Mock()
vc_data_model.reserved_networks = 'restricted_network1,restricted_network2'
vc_data_model.default_dvswitch = None
vc_data_model.default_network = 'Anetwork'
vc_data_model.default_datacenter = 'datacenter'
vc_data_model.holding_network = 'Holding Network'
request, expected = self._get_missing_dv_switch_params()
results = self.ConnectionCommandOrchestrator.connect_bulk(si=self.si,
logger=Mock(),
vcenter_data_model=vc_data_model,
request=request)
self._assert_as_expected(results, expected)
def test_connect_bulk1(self):
"""
Simple connect vm to vlan 'Access' mode
"""
request, expected = self._get_test1_params()
results = self.ConnectionCommandOrchestrator.connect_bulk(si=self.si,
logger=Mock(),
vcenter_data_model=self.vc_data_model,
request=request)
self._assert_as_expected(results, expected)
def test_connect_bulk2(self):
"""
Simple disconnect
"""
request, expected = self._get_test2_params()
results = self.ConnectionCommandOrchestrator.connect_bulk(si=self.si,
logger=Mock(),
vcenter_data_model=self.vc_data_model,
request=request)
self._assert_as_expected(results, expected)
def test_connect_bulk3(self):
"""
Complex connect multiple vms and multiple vlan types
"""
request, expected = self._get_test3_params()
results = self.ConnectionCommandOrchestrator.connect_bulk(si=self.si,
logger=Mock(),
vcenter_data_model=self.vc_data_model,
request=request)
self._assert_as_expected(results, expected)
def test_connect_bulk4(self):
"""
Simple disconnect
"""
request, expected = self._get_test4_params()
self.assertRaises(ValueError,
self.ConnectionCommandOrchestrator.connect_bulk, self.si, Mock(), self.vc_data_model, request)
def test_connect_bulk5(self):
"""
Disconnect returns error
"""
request, expected = self._get_test5_params()
res = self.ConnectionCommandOrchestrator.connect_bulk(si=self.si,
logger=Mock(),
vcenter_data_model=self.vc_data_model,
request=request)
self._assert_as_expected(res, expected)
def test_connect_bulk6(self):
"""
Connect returns error
"""
request, expected = self._get_test6_params()
res = self.ConnectionCommandOrchestrator.connect_bulk(si=self.si,
logger=Mock(),
vcenter_data_model=self.vc_data_model,
request=request)
self._assert_as_expected(res, expected)
def test_connect_bulk7(self):
"""
Simple connect vm to vlan 'Access' mode
"""
request, expected = self._get_test7_params()
results = self.ConnectionCommandOrchestrator.connect_bulk(si=self.si,
logger=Mock(),
vcenter_data_model=self.vc_data_model,
request=request)
self._assert_as_expected(results, expected)
def test_connect_bulk8(self):
"""
Simple connect vm to vlan 'Access' mode
"""
request, expected = self._get_test8_params()
results = self.ConnectionCommandOrchestrator.connect_bulk(si=self.si,
logger=Mock(),
vcenter_data_model=self.vc_data_model,
request=request)
self._assert_as_expected(results, expected)
def test_connect_bulk9(self):
"""
Simple disconnect
"""
request, expected = self._get_test9_params()
results = self.ConnectionCommandOrchestrator.connect_bulk(si=self.si,
logger=Mock(),
vcenter_data_model=self.vc_data_model,
request=request)
self._assert_as_expected(results, expected)
def _assert_as_expected(self, res, exp):
for r in res:
for e in exp:
found = e.actionId == r.actionId
if found:
for prop in [prop for prop in dir(r) if not prop.startswith('__')]:
self.assertEqual(getattr(r, prop), getattr(e, prop))
break
def _set_connect_to_networks_by_request(self, request):
a = DeployDataHolder(request['driverRequest'])
res = []
for action in a.actions:
vm_uuid = self.ConnectionCommandOrchestrator._get_vm_uuid(action)
vnic_name = self.ConnectionCommandOrchestrator._validate_vnic_name(
self.ConnectionCommandOrchestrator._get_vnic_name(action))
if vnic_name:
vnic_name = vnic_name.strip().split(',')
else:
vnic_name = [None]
for vnic in vnic_name:
vlan_id = action.connectionParams.vlanId
mode = action.connectionParams.mode
interface = self._get_interface_name(action)
name = 'Network adapter {0}'.format(vnic) if vnic and vnic.isdigit() else vnic
r = ConnectionResult(mac_address=interface,
vnic_name=name,
requested_vnic=name,
vm_uuid=vm_uuid,
network_name=self.portgroup_name.generate_port_group_name('dvSwitch', vlan_id,
mode),
network_key='aa')
res.append(r)
self.connector.connect_to_networks = Mock(return_value=res)
def _set_disconnect_from_networks(self, request):
a = DeployDataHolder(request['driverRequest'])
res = []
for action in a.actions:
vnic_name = self.ConnectionCommandOrchestrator._validate_vnic_name(
self.ConnectionCommandOrchestrator._get_vnic_name(action))
interface = self._get_interface_name(action)
r = VNicDeviceMapper(vnic=Mock(), requested_vnic=vnic_name, network=Mock(), connect=False, mac=interface)
res.append(r)
self.disconnector.disconnect_from_networks = Mock(return_value=res)
def _get_interface_name_for_result(self, action):
interface_attributes = [attr.attributeValue
for attr in action.connectorAttributes
if attr.attributeName == 'Interface']
vnics = [attr.attributeValue
for attr in action.connectorAttributes
if attr.attributeName == 'Requested Source vNIC Name']
if vnics:
vnics = vnics[0].strip().split(',')
interface = 'aa'
for vnic in vnics[1:]:
interface = '{0},{1}'.format(interface, 'aa' if not interface_attributes else interface_attributes[0])
return interface
def _get_interface_name(self, action):
interface_attributes = [attr.attributeValue
for attr in action.connectorAttributes
if attr.attributeName == 'Interface']
interface = 'aa' if not interface_attributes else interface_attributes[0]
return interface
def _get_connect_excepted_results(self, request, error_msg=None, ignore_interface=False):
a = DeployDataHolder(request['driverRequest'])
res = []
for action in a.actions:
r = ActionResult()
r.actionId = action.actionId
r.type = action.type
r.errorMessage = error_msg
r.infoMessage = 'VLAN successfully set' if not error_msg else None
r.success = True if not error_msg else False
r.updatedInterface = None if ignore_interface else self._get_interface_name_for_result(action)
res.append(r)
return res
def _get_disconnect_excepted_results(self, request, error_msg=None):
a = DeployDataHolder(request['driverRequest'])
res = []
for action in a.actions:
r = ActionResult()
r.actionId = action.actionId
r.type = action.type
r.errorMessage = error_msg
r.infoMessage = 'VLAN Successfully removed' if not error_msg else None
r.success = True if not error_msg else False
r.updatedInterface = self._get_interface_name(action)
res.append(r)
return res
def _get_missing_dv_switch_params(self):
request = {
'driverRequest': {
'actions': [
{
"connectorAttributes": [],
"connectionParams": {
"vlanId": "2",
"mode": "Access",
"vlanServiceAttributes": [
{
"attributeName": "Access Mode",
"attributeValue": "Access",
"type": "vlanServiceAttribute"
}
],
"type": "setVlanParameter"
},
"actionId": "ee8a3dc8-eb4b-4141-92ad-58bbd8430cad_376046bb-1ef0-4ecc-bb52-c0c7a9c75b1c",
"actionTarget": {
"fullName": "VM Deployment1_9602ad34",
"fullAddress": "N/A/NA",
"type": "actionTarget"
},
"customActionAttributes": [
{
"attributeName": "VM_UUID",
"attributeValue": "42220ae6-2fa8-b4cd-14e4-16fbad2798f6",
"type": "customAttribute"
}
],
"type": "setVlan"
}
]
}
}
self._set_connect_to_networks_by_request(request)
expected = self._get_connect_excepted_results(request,
error_msg='Please set the attribute "Default DvSwitch" in order to execute any connectivity changes',
ignore_interface=True)
return jsonpickle.encode(request), expected
def _get_test1_params(self):
request = {
'driverRequest': {
'actions': [
{
"connectorAttributes": [],
"connectionParams": {
"vlanId": "2",
"mode": "Access",
"vlanServiceAttributes": [
{
"attributeName": "Access Mode",
"attributeValue": "Access",
"type": "vlanServiceAttribute"
}
],
"type": "setVlanParameter"
},
"actionId": "ee8a3dc8-eb4b-4141-92ad-58bbd8430cad_376046bb-1ef0-4ecc-bb52-c0c7a9c75b1c",
"actionTarget": {
"fullName": "VM Deployment1_9602ad34",
"fullAddress": "N/A/NA",
"type": "actionTarget"
},
"customActionAttributes": [
{
"attributeName": "VM_UUID",
"attributeValue": "42220ae6-2fa8-b4cd-14e4-16fbad2798f6",
"type": "customAttribute"
}
],
"type": "setVlan"
}
]
}
}
self._set_connect_to_networks_by_request(request)
expected = self._get_connect_excepted_results(request)
return jsonpickle.encode(request), expected
def _get_test2_params(self):
request = {
"driverRequest": {
"actions": [
{
"connectionId": "d5369772-66de-4003-85cc-e94a57c20f1e",
"connectionParams": {
"vlanId": "654",
"mode": "Access",
"vlanServiceAttributes": [
{
"attributeName": "Access Mode",
"attributeValue": "Access",
"type": "vlanServiceAttribute"
},
{
"attributeName": "VLAN ID",
"attributeValue": "654",
"type": "vlanServiceAttribute"
},
{
"attributeName": "Virtual Network",
"attributeValue": "654",
"type": "vlanServiceAttribute"
}
],
"type": "setVlanParameter"
},
"connectorAttributes": [
{
"attributeName": "Interface",
"attributeValue": "00:50:56:a2:5f:07",
"type": "connectorAttribute"
}
],
"actionId": "d5369772-66de-4003-85cc-e94a57c20f1e_30fb5d10-1ce9-4e9f-b41e-ce0d271fd7ab",
"actionTarget": {
"fullName": "Temp4_a443da02",
"fullAddress": "192.168.65.33",
"type": "actionTarget"
},
"customActionAttributes": [
{
"attributeName": "VM_UUID",
"attributeValue": "422279bf-5899-789f-ca44-2c62075b3d2d",
"type": "customAttribute"
}
],
"type": "removeVlan"
}
]
}
}
self._set_disconnect_from_networks(request)
expected = self._get_disconnect_excepted_results(request)
return jsonpickle.encode(request), expected
def _get_test3_params(self):
request = {
'driverRequest': {
'actions': [
{
"connectorAttributes": [],
"connectionParams": {
"vlanId": "2",
"mode": "Access",
"vlanServiceAttributes": [
{
"attributeName": "Access Mode",
"attributeValue": "Access",
"type": "vlanServiceAttribute"
}
],
"type": "setVlanParameter"
},
"actionId": "1",
"actionTarget": {
"fullName": "VM Deployment1_9602ad34",
"fullAddress": "N/A/NA",
"type": "actionTarget"
},
"customActionAttributes": [
{
"attributeName": "VM_UUID",
"attributeValue": "42220ae6-2fa8-b4cd-14e4-16fbad2798f6",
"type": "customAttribute"
}
],
"type": "setVlan"
},
{
"connectorAttributes": [],
"connectionParams": {
"vlanId": "2-10",
"mode": "Access",
"vlanServiceAttributes": [
{
"attributeName": "Access Mode",
"attributeValue": "Trunk",
"type": "vlanServiceAttribute"
}
],
"type": "setVlanParameter"
},
"actionId": "2",
"actionTarget": {
"fullName": "VM Deployment1_9602ad34",
"fullAddress": "N/A/NA",
"type": "actionTarget"
},
"customActionAttributes": [
{
"attributeName": "VM_UUID",
"attributeValue": "42220ae6-2fa8-b4cd-14e4-16fbad2798f6",
"type": "customAttribute"
}
],
"type": "setVlan"
},
{
"connectorAttributes": [],
"connectionParams": {
"vlanId": "2",
"mode": "Access",
"vlanServiceAttributes": [
{
"attributeName": "Access Mode",
"attributeValue": "Access",
"type": "vlanServiceAttribute"
}
],
"type": "setVlanParameter"
},
"actionId": "3",
"actionTarget": {
"fullName": "VM Deployment1_9602ad34",
"fullAddress": "N/A/NA",
"type": "actionTarget"
},
"customActionAttributes": [
{
"attributeName": "VM_UUID",
"attributeValue": "42220ae6-2fa8-b4cd-14e4-16fbad2798f6",
"type": "customAttribute"
},
{
"attributeName": "Vnic Name",
"attributeValue": "1",
"type": "customAttribute"
}
],
"type": "setVlan"
},
{
"connectorAttributes": [],
"connectionParams": {
"vlanId": "2",
"mode": "Access",
"vlanServiceAttributes": [
{
"attributeName": "Access Mode",
"attributeValue": "Access",
"type": "vlanServiceAttribute"
}
],
"type": "setVlanParameter"
},
"actionId": "4",
"actionTarget": {
"fullName": "VM Deployment1_9602ad34",
"fullAddress": "N/A/NA",
"type": "actionTarget"
},
"customActionAttributes": [
{
"attributeName": "VM_UUID",
"attributeValue": "1234",
"type": "customAttribute"
},
{
"attributeName": "Vnic Name",
"attributeValue": "1",
"type": "customAttribute"
}
],
"type": "setVlan"
}
]
}
}
self._set_connect_to_networks_by_request(request)
expected = self._get_connect_excepted_results(request)
return jsonpickle.encode(request), expected
def _get_test4_params(self):
request = {
"driverRequest": {
"actions": [
{
"connectionId": "d5369772-66de-4003-85cc-e94a57c20f1e",
"connectionParams": {
"vlanId": "654",
"mode": "Access",
"vlanServiceAttributes": [
{
"attributeName": "Access Mode",
"attributeValue": "Access",
"type": "vlanServiceAttribute"
},
{
"attributeName": "VLAN ID",
"attributeValue": "654",
"type": "vlanServiceAttribute"
},
{
"attributeName": "Virtual Network",
"attributeValue": "654",
"type": "vlanServiceAttribute"
}
],
"type": "setVlanParameter"
},
"connectorAttributes": [
{
"attributeName": "Interface",
"attributeValue": "00:50:56:a2:5f:07",
"type": "connectorAttribute"
}
],
"actionId": "d5369772-66de-4003-85cc-e94a57c20f1e_30fb5d10-1ce9-4e9f-b41e-ce0d271fd7ab",
"actionTarget": {
"fullName": "Temp4_a443da02",
"fullAddress": "192.168.65.33",
"type": "actionTarget"
},
"customActionAttributes": [
],
"type": "removeVlan"
}
]
}
}
self._set_disconnect_from_networks(request)
expected = self._get_disconnect_excepted_results(request, 'VM_UUID is missing on action attributes')
return jsonpickle.encode(request), expected
def _get_test5_params(self):
request = {
"driverRequest": {
"actions": [
{
"connectionId": "d5369772-66de-4003-85cc-e94a57c20f1e",
"connectionParams": {
"vlanId": "654",
"mode": "Access",
"vlanServiceAttributes": [
{
"attributeName": "Access Mode",
"attributeValue": "Access",
"type": "vlanServiceAttribute"
},
{
"attributeName": "VLAN ID",
"attributeValue": "654",
"type": "vlanServiceAttribute"
},
{
"attributeName": "Virtual Network",
"attributeValue": "654",
"type": "vlanServiceAttribute"
}
],
"type": "setVlanParameter"
},
"connectorAttributes": [
{
"attributeName": "Interface",
"attributeValue": "00:50:56:a2:5f:07",
"type": "connectorAttribute"
}
],
"actionId": "d5369772-66de-4003-85cc-e94a57c20f1e_30fb5d10-1ce9-4e9f-b41e-ce0d271fd7ab",
"actionTarget": {
"fullName": "Temp4_a443da02",
"fullAddress": "192.168.65.33",
"type": "actionTarget"
},
"customActionAttributes": [
{
"attributeName": "VM_UUID",
"attributeValue": "42220ae6-2fa8-b4cd-14e4-16fbad2798f6",
"type": "customAttribute"
}
],
"type": "removeVlan"
}
]
}
}
self.disconnector.disconnect_from_networks = Mock()
self.disconnector.disconnect_from_networks.side_effect = ValueError('vnic not found')
expected = self._get_disconnect_excepted_results(request, 'vnic not found')
return jsonpickle.encode(request), expected
def _get_test6_params(self):
request = {
'driverRequest': {
'actions': [
{
"connectorAttributes": [],
"connectionParams": {
"vlanId": "2",
"mode": "Access",
"vlanServiceAttributes": [
{
"attributeName": "Access Mode",
"attributeValue": "Access",
"type": "vlanServiceAttribute"
}
],
"type": "setVlanParameter"
},
"actionId": "ee8a3dc8-eb4b-4141-92ad-58bbd8430cad_376046bb-1ef0-4ecc-bb52-c0c7a9c75b1c",
"actionTarget": {
"fullName": "VM Deployment1_9602ad34",
"fullAddress": "N/A/NA",
"type": "actionTarget"
},
"customActionAttributes": [
{
"attributeName": "VM_UUID",
"attributeValue": "42220ae6-2fa8-b4cd-14e4-16fbad2798f6",
"type": "customAttribute"
}
],
"type": "setVlan"
}
]
}
}
self.connector.connect_to_networks = Mock()
self.connector.connect_to_networks.side_effect = ValueError('vnic not found')
expected = self._get_connect_excepted_results(request, 'vnic not found')
# here we don't don't get vnic input input
for e in expected:
e.updatedInterface = None
return jsonpickle.encode(request), expected
def _get_test7_params(self):
request = {"driverRequest": {"actions": [{"connectionId": "6bd00213-6cfa-435a-abf5-6c839dae0f44",
"connectionParams": {"vlanId": "2", "mode": "Access",
"vlanServiceAttributes": [
{"attributeName": "QnQ",
"attributeValue": "False",
"type": "vlanServiceAttribute"},
{"attributeName": "CTag",
"attributeValue": "",
"type": "vlanServiceAttribute"},
{"attributeName": "Allocation Ranges",
"attributeValue": "2-4094",
"type": "vlanServiceAttribute"},
{"attributeName": "Isolation Level",
"attributeValue": "Exclusive",
"type": "vlanServiceAttribute"},
{"attributeName": "Access Mode",
"attributeValue": "Access",
"type": "vlanServiceAttribute"},
{"attributeName": "VLAN ID",
"attributeValue": "",
"type": "vlanServiceAttribute"},
{"attributeName": "Pool Name",
"attributeValue": "",
"type": "vlanServiceAttribute"},
{"attributeName": "Virtual Network",
"attributeValue": "2",
"type": "vlanServiceAttribute"}],
"type": "setVlanParameter"},
"connectorAttributes": [
{"attributeName": "Requested Source vNIC Name",
"attributeValue": "1,2", "type": "connectorAttribute"}],
"actionId": "6bd00213-6cfa-435a-abf5-6c839dae0f44_c6b69177-d251-4dba-ba88-d7c5bb259505",
"actionTarget": {"fullName": "temp_b648f80f", "fullAddress": "NA",
"type": "actionTarget"}, "customActionAttributes": [
{"attributeName": "auto_power_off", "attributeValue": "auto_power_off", "type": "customAttribute"},
{"attributeName": "ip_regex", "attributeValue": "ip_regex", "type": "customAttribute"},
{"attributeName": "wait_for_ip", "attributeValue": "wait_for_ip", "type": "customAttribute"},
{"attributeName": "autoload", "attributeValue": "autoload", "type": "customAttribute"},
{"attributeName": "auto_delete", "attributeValue": "auto_delete", "type": "customAttribute"},
{"attributeName": "auto_power_on", "attributeValue": "auto_power_on", "type": "customAttribute"},
{"attributeName": "refresh_ip_timeout", "attributeValue": "refresh_ip_timeout",
"type": "customAttribute"},
{"attributeName": "VM_UUID", "attributeValue": "422203f6-eadd-9f88-5dc8-00c17f49fa21",
"type": "customAttribute"},
{"attributeName": "Vnic Name", "attributeValue": "1,2", "type": "customAttribute"}],
"type": "setVlan"}]}}
self._set_connect_to_networks_by_request(request)
expected = self._get_connect_excepted_results(request)
return jsonpickle.encode(request), expected
def _get_test8_params(self):
request = {"driverRequest": {"actions": [{"connectionId": "6bd00213-6cfa-435a-abf5-6c839dae0f44",
"connectionParams": {"vlanId": "2", "mode": "Access",
"vlanServiceAttributes": [
{"attributeName": "QnQ",
"attributeValue": "False",
"type": "vlanServiceAttribute"},
{"attributeName": "CTag",
"attributeValue": "",
"type": "vlanServiceAttribute"},
{
"attributeName": "Allocation Ranges",
"attributeValue": "2-4094",
"type": "vlanServiceAttribute"},
{
"attributeName": "Isolation Level",
"attributeValue": "Exclusive",
"type": "vlanServiceAttribute"},
{
"attributeName": "Access Mode",
"attributeValue": "Access",
"type": "vlanServiceAttribute"},
{
"attributeName": "VLAN ID",
"attributeValue": "",
"type": "vlanServiceAttribute"},
{
"attributeName": "Pool Name",
"attributeValue": "",
"type": "vlanServiceAttribute"},
{
"attributeName": "Virtual Network",
"attributeValue": "2",
"type": "vlanServiceAttribute"}],
"type": "setVlanParameter"},
"connectorAttributes": [
{"attributeName": "Requested Source vNIC Name",
"attributeValue": "1,2", "type": "connectorAttribute"}],
"actionId": "6bd00213-6cfa-435a-abf5-6c839dae0f44_c6b69177-d251-4dba-ba88-d7c5bb259505",
"actionTarget": {"fullName": "temp_b648f80f", "fullAddress": "NA",
"type": "actionTarget"}, "customActionAttributes": [
{"attributeName": "auto_power_off", "attributeValue": "auto_power_off", "type": "customAttribute"},
{"attributeName": "ip_regex", "attributeValue": "ip_regex", "type": "customAttribute"},
{"attributeName": "wait_for_ip", "attributeValue": "wait_for_ip", "type": "customAttribute"},
{"attributeName": "autoload", "attributeValue": "autoload", "type": "customAttribute"},
{"attributeName": "auto_delete", "attributeValue": "auto_delete", "type": "customAttribute"},
{"attributeName": "auto_power_on", "attributeValue": "auto_power_on", "type": "customAttribute"},
{"attributeName": "refresh_ip_timeout", "attributeValue": "refresh_ip_timeout",
"type": "customAttribute"},
{"attributeName": "VM_UUID", "attributeValue": "422203f6-eadd-9f88-5dc8-00c17f49fa21",
"type": "customAttribute"},
{"attributeName": "Vnic Name", "attributeValue": "Network Adapter 1,Network Adapter 2",
"type": "customAttribute"}], "type": "setVlan"}]}}
self._set_connect_to_networks_by_request(request)
expected = self._get_connect_excepted_results(request)
return jsonpickle.encode(request), expected
def _get_test9_params(self):
request = {"driverRequest":{"actions":[{"connectionId":"4dd285fa-d271-4023-b7d1-881553a8f59f","connectionParams":{"vlanId":"2","mode":"Access","vlanServiceAttributes":[{"attributeName":"QnQ","attributeValue":"False","type":"vlanServiceAttribute"},{"attributeName":"CTag","attributeValue":"","type":"vlanServiceAttribute"},{"attributeName":"Allocation Ranges","attributeValue":"2-4094","type":"vlanServiceAttribute"},{"attributeName":"Isolation Level","attributeValue":"Exclusive","type":"vlanServiceAttribute"},{"attributeName":"Access Mode","attributeValue":"Access","type":"vlanServiceAttribute"},{"attributeName":"VLAN ID","attributeValue":"","type":"vlanServiceAttribute"},{"attributeName":"Pool Name","attributeValue":"","type":"vlanServiceAttribute"},{"attributeName":"Virtual Network","attributeValue":"2","type":"vlanServiceAttribute"}],"type":"setVlanParameter"},"connectorAttributes":[{"attributeName":"Interface","attributeValue":"00:50:56:a2:37:3b","type":"connectorAttribute"}],"actionId":"4dd285fa-d271-4023-b7d1-881553a8f59f_c6b69177-d251-4dba-ba88-d7c5bb259505","actionTarget":{"fullName":"temp_b648f80f","fullAddress":"NA","type":"actionTarget"},"customActionAttributes":[{"attributeName":"auto_power_off","attributeValue":"auto_power_off","type":"customAttribute"},{"attributeName":"ip_regex","attributeValue":"ip_regex","type":"customAttribute"},{"attributeName":"wait_for_ip","attributeValue":"wait_for_ip","type":"customAttribute"},{"attributeName":"autoload","attributeValue":"autoload","type":"customAttribute"},{"attributeName":"auto_delete","attributeValue":"auto_delete","type":"customAttribute"},{"attributeName":"auto_power_on","attributeValue":"auto_power_on","type":"customAttribute"},{"attributeName":"refresh_ip_timeout","attributeValue":"refresh_ip_timeout","type":"customAttribute"},{"attributeName":"VM_UUID","attributeValue":"422203f6-eadd-9f88-5dc8-00c17f49fa21","type":"customAttribute"}],"type":"removeVlan"}]}}
self._set_disconnect_from_networks(request)
expected = self._get_disconnect_excepted_results(request)
return jsonpickle.encode(request), expected
