| # -*- coding: utf-8 -*- |
| |
| ## |
| # Copyright 2016-2017 VMware Inc. |
| # This file is part of ETSI OSM |
| # All Rights Reserved. |
| # |
| # Licensed under the Apache License, Version 2.0 (the "License"); you may |
| # not use this file except in compliance with the License. You may obtain |
| # a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT |
| # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the |
| # License for the specific language governing permissions and limitations |
| # under the License. |
| # |
| # For those usages not covered by the Apache License, Version 2.0 please |
| # contact: osslegalrouting@vmware.com |
| ## |
| |
| |
| from osm_ro.vimconn_vmware import vimconnector |
| from osm_ro.vimconn import vimconnUnexpectedResponse,vimconnNotFoundException |
| from pyvcloud.vcloudair import VCA,VCS |
| from pyvcloud.vapp import VAPP |
| from pyvcloud import Http |
| from pyvcloud.schema.vcd.v1_5.schemas.vcloud import vdcType,networkType,catalogType, \ |
| vAppType,taskType |
| import unittest |
| import mock |
| import test_vimconn_vmware_xml_response as xml_resp |
| |
| class TestVimconn_VMware(unittest.TestCase): |
| def setUp(self): |
| config = { "admin_password": "admin", |
| "admin_username":"user", |
| "nsx_user": "nsx", |
| "nsx_password": "nsx", |
| "nsx_manager":"https://test-nsx" } |
| |
| self.vca = VCA(host='test', |
| username='user', |
| service_type='standalone', |
| version='5.9', |
| verify=False, |
| log=False) |
| |
| self.session = VCS('https://test/api/session', |
| 'test', |
| 'test', |
| None, |
| 'https://test/api/org/a93c', |
| 'https://test/api/org/a93c', |
| version='5.9') |
| |
| self.vim = vimconnector(uuid='12354', |
| name='test', |
| tenant_id='abc1234', |
| tenant_name='test', |
| url='https://test', |
| config=config) |
| |
| |
| @mock.patch.object(vimconnector,'get_vdc_details') |
| @mock.patch.object(vimconnector,'connect') |
| @mock.patch.object(VCA,'get_networks') |
| def test_get_network_not_found(self,get_networks, connect, get_vdc_details): |
| vdc_xml_resp = xml_resp.vdc_xml_response |
| # created vdc object |
| vdc = vdcType.parseString(vdc_xml_resp,True) |
| # assumed return value from VIM connector |
| get_vdc_details.return_value = vdc |
| self.vim.vca = self.vim.connect() |
| network_xml_resp = xml_resp.network_xml_response |
| networks = networkType.parseString(network_xml_resp, True) |
| (self.vim.vca).get_networks.return_value = [networks] |
| # call to VIM connector method with invalid id |
| self.assertRaises(vimconnNotFoundException,self.vim.get_network,'mgmt-net') |
| |
| @mock.patch.object(vimconnector,'get_vdc_details') |
| @mock.patch.object(vimconnector,'connect') |
| @mock.patch.object(VCA,'get_networks') |
| def test_get_network(self,get_networks, connect, get_vdc_details): |
| net_id = '5c04dc6d-6096-47c6-b72b-68f19013d491' |
| vdc_xml_resp = xml_resp.vdc_xml_response |
| # created vdc object |
| vdc = vdcType.parseString(vdc_xml_resp,True) |
| # created network object |
| network_xml_resp = xml_resp.network_xml_response |
| networks = networkType.parseString(network_xml_resp, True) |
| # assumed return value from VIM connector |
| get_vdc_details.return_value = vdc |
| self.vim.vca = self.vim.connect() |
| # assumed return value from VIM connector |
| (self.vim.vca).get_networks.return_value = [networks] |
| # call to VIM connector method with network_id |
| result = self.vim.get_network(net_id) |
| # assert verified expected and return result from VIM connector |
| self.assertEqual(net_id, result['id']) |
| |
| @mock.patch.object(vimconnector,'get_vdc_details') |
| @mock.patch.object(vimconnector,'connect') |
| @mock.patch.object(VCA,'get_networks') |
| def test_get_network_list_not_found(self,get_networks, connect, get_vdc_details): |
| vdc_xml_resp = xml_resp.vdc_xml_response |
| # created vdc object |
| vdc = vdcType.parseString(vdc_xml_resp,True) |
| # assumed return value from VIM connector |
| get_vdc_details.return_value = vdc |
| self.vim.vca = self.vim.connect() |
| |
| network_xml_resp = xml_resp.network_xml_response |
| networks = networkType.parseString(network_xml_resp, True) |
| (self.vim.vca).get_networks.return_value = [networks] |
| # call to VIM connector method with network_id |
| result = self.vim.get_network_list({'id':'45hdfg-345nb-345'}) |
| |
| # assert verified expected and return result from VIM connector |
| self.assertEqual(list(), result) |
| |
| @mock.patch.object(vimconnector,'get_vdc_details') |
| @mock.patch.object(vimconnector,'connect') |
| @mock.patch.object(VCA,'get_networks') |
| def test_get_network_list(self,get_networks, connect, get_vdc_details): |
| vdc_xml_resp = xml_resp.vdc_xml_response |
| net_id = '5c04dc6d-6096-47c6-b72b-68f19013d491' |
| vdc = vdcType.parseString(vdc_xml_resp,True) |
| # created network object |
| network_xml_resp = xml_resp.network_xml_response |
| networks = networkType.parseString(network_xml_resp, True) |
| # assumed return value from VIM connector |
| get_vdc_details.return_value = vdc |
| self.vim.vca = self.vim.connect() |
| # assumed return value from VIM connector |
| (self.vim.vca).get_networks.return_value = [networks] |
| |
| # call to VIM connector method with network_id |
| result = self.vim.get_network_list({'id': net_id}) |
| # assert verified expected and return result from VIM connector |
| for item in result: |
| self.assertEqual(item.get('id'), net_id) |
| self.assertEqual(item.get('status'), 'ACTIVE') |
| self.assertEqual(item.get('shared'), False) |
| |
| @mock.patch.object(vimconnector,'create_network_rest') |
| def test_new_network(self, create_network_rest): |
| create_net_xml_resp = xml_resp.create_network_xml_response |
| net_name = 'Test_network' |
| net_type = 'bridge' |
| # assumed return value from VIM connector |
| create_network_rest.return_value = create_net_xml_resp |
| # call to VIM connector method with network_id |
| result = self.vim.new_network(net_name, net_type) |
| # assert verified expected and return result from VIM connector |
| self.assertEqual(result, 'df1956fa-da04-419e-a6a2-427b6f83788f') |
| |
| @mock.patch.object(vimconnector, 'create_network_rest') |
| def test_new_network_not_created(self, create_network_rest): |
| # assumed return value from VIM connector |
| create_network_rest.return_value = """<?xml version="1.0" encoding="UTF-8"?> |
| <OrgVdcNetwork></OrgVdcNetwork>""" |
| |
| # assert verified expected and return result from VIM connector |
| self.assertRaises(vimconnUnexpectedResponse,self.vim.new_network, |
| 'test_net', |
| 'bridge') |
| |
| @mock.patch.object(vimconnector, 'connect') |
| @mock.patch.object(vimconnector, 'get_network_action') |
| @mock.patch.object(vimconnector, 'connect_as_admin') |
| @mock.patch.object(vimconnector, 'delete_network_action') |
| def test_delete_network(self, delete_network_action, connect_as_admin, get_network_action, connect): |
| delete_net_xml_resp = xml_resp.delete_network_xml_response |
| # assumed return value from VIM connector |
| connect.return_value = self.vca |
| self.vim.vca = self.vim.connect() |
| get_network_action.return_value = delete_net_xml_resp |
| connect_as_admin.return_value = self.vca |
| delete_network_action.return_value = True |
| # call to VIM connector method with network_id |
| result = self.vim.delete_network('0a55e5d1-43a2-4688-bc92-cb304046bf87') |
| # assert verified expected and return result from VIM connector |
| self.assertEqual(result, '0a55e5d1-43a2-4688-bc92-cb304046bf87') |
| |
| @mock.patch.object(vimconnector, 'get_vcd_network') |
| def test_delete_network_not_found(self, get_vcd_network): |
| # assumed return value from VIM connector |
| get_vcd_network.return_value = False |
| # assert verified expected and return result from VIM connector |
| self.assertRaises(vimconnNotFoundException,self.vim.delete_network, |
| '2a23e5d1-42a2-0648-bc92-cb508046bf87') |
| |
| def test_get_flavor(self): |
| flavor_data = {'a646eb8a-95bd-4e81-8321-5413ee72b62e': {'disk': 10, |
| 'vcpus': 1, |
| 'ram': 1024}} |
| vimconnector.flavorlist = flavor_data |
| result = self.vim.get_flavor('a646eb8a-95bd-4e81-8321-5413ee72b62e') |
| # assert verified expected and return result from VIM connector |
| self.assertEqual(result, flavor_data['a646eb8a-95bd-4e81-8321-5413ee72b62e']) |
| |
| def test_get_flavor_not_found(self): |
| vimconnector.flavorlist = {} |
| # assert verified expected and return result from VIM connector |
| self.assertRaises(vimconnNotFoundException,self.vim.get_flavor, |
| 'a646eb8a-95bd-4e81-8321-5413ee72b62e') |
| |
| def test_new_flavor(self): |
| flavor_data = {'disk': 10, 'vcpus': 1, 'ram': 1024} |
| result = self.vim.new_flavor(flavor_data) |
| # assert verified expected and return result from VIM connector |
| self.assertIsNotNone(result) |
| |
| def test_delete_flavor(self): |
| flavor_data = {'2cb3dffb-5c51-4355-8406-28553ead28ac': {'disk': 10, |
| 'vcpus': 1, |
| 'ram': 1024}} |
| vimconnector.flavorlist = flavor_data |
| # return value from VIM connector |
| result = self.vim.delete_flavor('2cb3dffb-5c51-4355-8406-28553ead28ac') |
| # assert verified expected and return result from VIM connector |
| self.assertEqual(result, '2cb3dffb-5c51-4355-8406-28553ead28ac') |
| |
| @mock.patch.object(vimconnector,'connect_as_admin') |
| @mock.patch.object(vimconnector,'connect') |
| @mock.patch.object(Http,'get') |
| @mock.patch.object(VCS,'get_vcloud_headers') |
| def test_delete_image_not_found(self, get_vcloud_headers, get, connect, connect_as_admin): |
| # assumed return value from VIM connector |
| connect.return_value = self.vca |
| self.vim.vca = self.vim.connect() |
| # assumed return value from VIM connector |
| connect_as_admin.return_value = self.vca |
| self.vca.host = connect_as_admin.return_value.host |
| self.vca.vcloud_session = self.session |
| get_vcloud_headers.return_value = {'Accept':'application/*+xml;version=5.9', |
| 'x-vcloud-authorization': '638bfee6cb5f435abc3480f480817254'} |
| get_vcloud_headers.return_value = self.vca.vcloud_session.get_vcloud_headers() |
| # assert verified expected and return result from VIM connector |
| self.assertRaises(vimconnNotFoundException, self.vim.delete_image, 'invali3453') |
| |
| @mock.patch.object(vimconnector,'connect') |
| @mock.patch.object(VCA,'get_catalogs') |
| def test_get_image_list(self, get_catalogs, connect): |
| catalog1 = xml_resp.catalog1_xml_response |
| catalog2 = xml_resp.catalog2_xml_response |
| |
| catalogs = [catalogType.parseString(cat, True) for cat in catalog1, catalog2] |
| connect.return_value = self.vca |
| self.vim.vca = self.vim.connect() |
| # assumed return value from VIM connector |
| self.vim.vca.get_catalogs.return_value = catalogs |
| result = self.vim.get_image_list({'id': '32ccb082-4a65-41f6-bcd6-38942e8a3829'}) |
| # assert verified expected and return result from VIM connector |
| for item in result: |
| self.assertEqual(item['id'], '32ccb082-4a65-41f6-bcd6-38942e8a3829') |
| |
| @mock.patch.object(vimconnector,'get_vapp_details_rest') |
| @mock.patch.object(vimconnector,'get_vdc_details') |
| def test_get_vminstance(self, get_vdc_details, get_vapp_details_rest): |
| vapp_info = {'status': '4', |
| 'acquireMksTicket': {'href': 'https://localhost/api/vApp/vm-47d12505-5968-4e16-95a7-18743edb0c8b/screen/action/acquireMksTicket', |
| 'type': 'application/vnd.vmware.vcloud.mksTicket+xml', 'rel': 'screen:acquireMksTicket'}, |
| 'vm_virtual_hardware': {'disk_edit_href': 'https://localhost/api/vApp/vm-47d12505-5968-4e16-95a7-18743edb0c8b/virtualHardwareSection/disks', 'disk_size': '40960'}, |
| 'name': 'Test1_vm-69a18104-8413-4cb8-bad7-b5afaec6f9fa', |
| 'created': '2017-09-21T01:15:31.627-07:00', |
| 'IsEnabled': 'true', |
| 'EndAddress': '12.16.24.199', |
| 'interfaces': [{'MACAddress': '00:50:56:01:12:a2', |
| 'NetworkConnectionIndex': '0', |
| 'network': 'testing_T6nODiW4-68f68d93-0350-4d86-b40b-6e74dedf994d', |
| 'IpAddressAllocationMode': 'DHCP', |
| 'IsConnected': 'true', |
| 'IpAddress': '12.16.24.200'}], |
| 'ovfDescriptorUploaded': 'true', |
| 'nestedHypervisorEnabled': 'false', |
| 'Gateway': '12.16.24.1', |
| 'acquireTicket': {'href': 'https://localhost/api/vApp/vm-47d12505-5968-4e16-95a7-18743edb0c8b/screen/action/acquireTicket', |
| 'rel': 'screen:acquireTicket'}, |
| 'vmuuid': '47d12505-5968-4e16-95a7-18743edb0c8b', |
| 'Netmask': '255.255.255.0', |
| 'StartAddress': '12.16.24.100', |
| 'primarynetwork': '0', |
| 'networkname': 'External-Network-1074', |
| 'IsInherited': 'false', |
| 'deployed': 'true'} |
| # created vdc object |
| vdc_xml_resp = xml_resp.vdc_xml_response |
| vdc = vdcType.parseString(vdc_xml_resp,True) |
| # assumed return value from VIM connector |
| get_vdc_details.return_value = vdc |
| get_vapp_details_rest.return_value = vapp_info |
| |
| result = self.vim.get_vminstance('47d12505-5968-4e16-95a7-18743edb0c8b') |
| # assert verified expected and return result from VIM connector |
| self.assertEqual(result['status'], 'ACTIVE') |
| self.assertEqual(result['hostId'], '47d12505-5968-4e16-95a7-18743edb0c8b') |
| |
| @mock.patch.object(VCA,'get_vapp') |
| @mock.patch.object(vimconnector,'connect') |
| @mock.patch.object(vimconnector,'get_namebyvappid') |
| @mock.patch.object(vimconnector,'get_vdc_details') |
| def test_delete_vminstance(self, get_vdc_details, get_namebyvappid, connect, vapp): |
| vm_id = '4f6a9b49-e92d-4935-87a1-0e4dc9c3a069' |
| vm_name = 'Test1_vm-69a18104-8413-4cb8-bad7-b5afaec6f9fa' |
| # created vdc object |
| vdc_xml_resp = xml_resp.vdc_xml_response |
| vdc = vdcType.parseString(vdc_xml_resp,True) |
| # assumed return value from VIM connector |
| connect.return_value = self.vca |
| self.vim.vca = self.vim.connect() |
| get_namebyvappid.return_name = vm_name |
| vapp.return_value = None |
| # call to VIM connector method |
| result = self.vim.delete_vminstance(vm_id) |
| # assert verified expected and return result from VIM connector |
| self.assertEqual(result, vm_id) |
| |
| @mock.patch.object(vimconnector,'get_network_id_by_name') |
| @mock.patch.object(vimconnector,'get_vm_pci_details') |
| @mock.patch.object(VCA,'get_vapp') |
| @mock.patch.object(vimconnector,'connect') |
| @mock.patch.object(vimconnector,'get_namebyvappid') |
| @mock.patch.object(vimconnector,'get_vdc_details') |
| def test_refresh_vms_status(self, get_vdc_details, get_namebyvappid, connect, |
| get_vapp, get_vm_pci_details, |
| get_network_id_by_name): |
| headers = {'Accept':'application/*+xml;version=5.9', |
| 'x-vcloud-authorization': '638bfee6cb5f435abc3480f480817254'} |
| vm_id = '05e6047b-6938-4275-8940-22d1ea7245b8' |
| |
| vapp_resp = xml_resp.vapp_xml_response |
| # created vdc object |
| vdc_xml_resp = xml_resp.vdc_xml_response |
| vdc = vdcType.parseString(vdc_xml_resp,True) |
| # assumed return value from VIM connector |
| get_vdc_details.return_value = vdc |
| connect.return_value = self.vca |
| |
| self.vim.vca = self.vim.connect() |
| get_namebyvappid.return_value = 'Test1_vm-69a18104-8413-4cb8-bad7-b5afaec6f9fa' |
| get_vm_pci_details.return_value = {'host_name': 'test-esx-1.corp.local', 'host_ip': '12.19.24.31'} |
| get_vapp.return_value = VAPP(vAppType.parseString(vapp_resp, True), headers, False) |
| get_network_id_by_name.return_value = '47d12505-5968-4e16-95a7-18743edb0c8b' |
| # call to VIM connector method |
| result = self.vim.refresh_vms_status([vm_id]) |
| for attr in result[vm_id]: |
| if attr == 'status': |
| # assert verified expected and return result from VIM connector |
| self.assertEqual(result[vm_id][attr], 'ACTIVE') |
| |
| @mock.patch.object(vimconnector,'get_vcd_network') |
| def test_refresh_nets_status(self, get_vcd_network): |
| net_id = 'c2d0f28f-d38b-4588-aecc-88af3d4af58b' |
| network_dict = {'status': '1','isShared': 'false','IpScope': '', |
| 'EndAddress':'12.19.21.15', |
| 'name': 'testing_gwyRXlvWYL1-9ebb6d7b-5c74-472f-be77-963ed050d44d', |
| 'Dns1': '12.19.21.10', 'IpRanges': '', |
| 'Gateway': '12.19.21.23', 'Netmask': '255.255.255.0', |
| 'RetainNetInfoAcrossDeployments': 'false', |
| 'IpScopes': '', 'IsEnabled': 'true', 'DnsSuffix': 'corp.local', |
| 'StartAddress': '12.19.21.11', 'IpRange': '', |
| 'Configuration': '', 'FenceMode': 'bridged', |
| 'IsInherited': 'true', 'uuid': 'c2d0f28f-d38b-4588-aecc-88af3d4af58b'} |
| # assumed return value from VIM connector |
| get_vcd_network.return_value = network_dict |
| result = self.vim.refresh_nets_status([net_id]) |
| # assert verified expected and return result from VIM connector |
| for attr in result[net_id]: |
| if attr == 'status': |
| self.assertEqual(result[net_id][attr], 'ACTIVE') |
| |
| @mock.patch.object(VCA,'block_until_completed') |
| @mock.patch.object(VCA,'get_vapp') |
| @mock.patch.object(vimconnector,'connect') |
| @mock.patch.object(vimconnector,'get_namebyvappid') |
| @mock.patch.object(vimconnector,'get_vdc_details') |
| def test_action_vminstance(self, get_vdc_details, get_namebyvappid, connect, get_vapp, block): |
| task_resp = xml_resp.task_xml |
| vm_id = '05e6047b-6938-4275-8940-22d1ea7245b8' |
| # created vdc object |
| vdc_xml_resp = xml_resp.vdc_xml_response |
| vdc = vdcType.parseString(vdc_xml_resp,True) |
| # assumed return value from VIM connector |
| get_vdc_details.return_value = vdc |
| get_namebyvappid.return_value = 'Test1_vm-69a18104-8413-4cb8-bad7-b5afaec6f9fa' |
| connect.return_value = self.vca |
| self.vim.vca = self.vim.connect() |
| get_vapp.return_value.undeploy.return_value = taskType.parseString(task_resp, True) |
| block.return_value = True |
| # call to VIM connector method |
| result = self.vim.action_vminstance(vm_id,{'shutdown': None}) |
| # assert verified expected and return result from VIM connector |
| self.assertEqual(result, None) |