Refactor the PBM support
Having a Pbm class that inherits from Vim is a design mistake that we need to fix before start using Pbm features in other projects like Nova. This patch introduces a new base class 'Service' which provides common functionality for invoking vSphere APIs and both Vim and Pbm inherit from it. That will allow to further evolve our APIs and add features which are specific to only Vim or Pbm. Existing clients which use the Vim object through VMwareAPISession are not impacted by this change. The interface of VMwareAPISession is unchanged. Change-Id: Icf54e3d0305b30c73d0ff7d9c85da1893392c3aa
This commit is contained in:
Radoslav Gerganov
@@ -124,54 +124,34 @@ class VMwareAPISessionTest(base.TestCase):
|
||||
self.VimMock.assert_called_with(protocol=api_session._scheme,
|
||||
host=VMwareAPISessionTest.SERVER_IP,
|
||||
port=VMwareAPISessionTest.PORT,
|
||||
wsdl_loc=api_session._vim_wsdl_loc)
|
||||
wsdl_url=api_session._vim_wsdl_loc)
|
||||
|
||||
@mock.patch.object(pbm, 'PBMClient')
|
||||
def test_pbm(self, pbm_client_mock):
|
||||
@mock.patch.object(pbm, 'Pbm')
|
||||
def test_pbm(self, pbm_mock):
|
||||
api_session = self._create_api_session(True)
|
||||
api_session._pbm_wsdl_loc = mock.Mock()
|
||||
pbm = mock.Mock()
|
||||
pbm_client_mock.return_value = pbm
|
||||
vim_obj = api_session.vim
|
||||
cookie = mock.Mock()
|
||||
vim_obj.get_http_cookie.return_value = cookie
|
||||
api_session._pbm_wsdl_loc = mock.Mock()
|
||||
|
||||
pbm = mock.Mock()
|
||||
pbm_mock.return_value = pbm
|
||||
api_session._get_session_cookie = mock.Mock(return_value=cookie)
|
||||
|
||||
self.assertEqual(pbm, api_session.pbm)
|
||||
pbm.set_cookie.assert_called_once_with(cookie)
|
||||
|
||||
def test_get_session_cookie(self):
|
||||
api_session = self._create_api_session(False)
|
||||
vim_obj = api_session.vim
|
||||
|
||||
cookie_value = 'xyz'
|
||||
cookie = mock.Mock()
|
||||
cookie.name = 'vmware_soap_session'
|
||||
cookie.value = cookie_value
|
||||
vim_obj.client.options.transport.cookiejar = [cookie]
|
||||
|
||||
self.assertEqual(cookie_value, api_session._get_session_cookie())
|
||||
|
||||
def test_get_session_cookie_with_no_cookie(self):
|
||||
api_session = self._create_api_session(False)
|
||||
vim_obj = api_session.vim
|
||||
|
||||
cookie = mock.Mock()
|
||||
cookie.name = 'cookie'
|
||||
cookie.value = 'xyz'
|
||||
vim_obj.client.options.transport.cookiejar = [cookie]
|
||||
|
||||
self.assertIsNone(api_session._get_session_cookie())
|
||||
pbm.set_soap_cookie.assert_called_once_with(cookie)
|
||||
|
||||
def test_create_session(self):
|
||||
session = mock.Mock()
|
||||
session.key = "12345"
|
||||
api_session = self._create_api_session(False)
|
||||
cookie = mock.Mock()
|
||||
vim_obj = api_session.vim
|
||||
vim_obj.Login.return_value = session
|
||||
vim_obj.get_http_cookie.return_value = cookie
|
||||
|
||||
pbm = mock.Mock()
|
||||
api_session._pbm = pbm
|
||||
cookie = mock.Mock()
|
||||
api_session._get_session_cookie = mock.Mock(return_value=cookie)
|
||||
|
||||
api_session._create_session()
|
||||
session_manager = vim_obj.service_content.sessionManager
|
||||
@@ -180,7 +160,7 @@ class VMwareAPISessionTest(base.TestCase):
|
||||
password=VMwareAPISessionTest.PASSWORD)
|
||||
self.assertFalse(vim_obj.TerminateSession.called)
|
||||
self.assertEqual(session.key, api_session._session_id)
|
||||
pbm.set_cookie.assert_called_once_with(cookie)
|
||||
pbm.set_soap_cookie.assert_called_once_with(cookie)
|
||||
|
||||
def test_create_session_with_existing_session(self):
|
||||
old_session_key = '12345'
|
||||
|
||||
290
tests/test_service.py
Normal file
290
tests/test_service.py
Normal file
@@ -0,0 +1,290 @@
|
||||
# Copyright (c) 2014 VMware, Inc.
|
||||
# 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.
|
||||
|
||||
import httplib
|
||||
import urllib2
|
||||
|
||||
import mock
|
||||
import suds
|
||||
|
||||
from oslo.vmware import exceptions
|
||||
from oslo.vmware import service
|
||||
from oslo.vmware import vim_util
|
||||
from tests import base
|
||||
|
||||
|
||||
class ServiceMessagePluginTest(base.TestCase):
|
||||
"""Test class for ServiceMessagePlugin."""
|
||||
|
||||
def test_add_attribute_for_value(self):
|
||||
node = mock.Mock()
|
||||
node.name = 'value'
|
||||
plugin = service.ServiceMessagePlugin()
|
||||
plugin.add_attribute_for_value(node)
|
||||
node.set.assert_called_once_with('xsi:type', 'xsd:string')
|
||||
|
||||
def test_marshalled(self):
|
||||
plugin = service.ServiceMessagePlugin()
|
||||
context = mock.Mock()
|
||||
plugin.marshalled(context)
|
||||
context.envelope.prune.assert_called_once_with()
|
||||
context.envelope.walk.assert_called_once_with(
|
||||
plugin.add_attribute_for_value)
|
||||
|
||||
|
||||
class ServiceTest(base.TestCase):
|
||||
|
||||
def setUp(self):
|
||||
super(ServiceTest, self).setUp()
|
||||
patcher = mock.patch('suds.client.Client')
|
||||
self.addCleanup(patcher.stop)
|
||||
self.SudsClientMock = patcher.start()
|
||||
|
||||
def test_retrieve_properties_ex_fault_checker_with_empty_response(self):
|
||||
try:
|
||||
service.Service._retrieve_properties_ex_fault_checker(None)
|
||||
assert False
|
||||
except exceptions.VimFaultException as ex:
|
||||
self.assertEqual([exceptions.NOT_AUTHENTICATED],
|
||||
ex.fault_list)
|
||||
|
||||
def test_retrieve_properties_ex_fault_checker(self):
|
||||
fault_list = ['FileFault', 'VimFault']
|
||||
missing_set = []
|
||||
for fault in fault_list:
|
||||
missing_elem = mock.Mock()
|
||||
missing_elem.fault.fault.__class__.__name__ = fault
|
||||
missing_set.append(missing_elem)
|
||||
obj_cont = mock.Mock()
|
||||
obj_cont.missingSet = missing_set
|
||||
response = mock.Mock()
|
||||
response.objects = [obj_cont]
|
||||
|
||||
try:
|
||||
service.Service._retrieve_properties_ex_fault_checker(response)
|
||||
assert False
|
||||
except exceptions.VimFaultException as ex:
|
||||
self.assertEqual(fault_list, ex.fault_list)
|
||||
|
||||
def test_request_handler(self):
|
||||
managed_object = 'VirtualMachine'
|
||||
resp = mock.Mock()
|
||||
|
||||
def side_effect(mo, **kwargs):
|
||||
self.assertEqual(managed_object, mo._type)
|
||||
self.assertEqual(managed_object, mo.value)
|
||||
return resp
|
||||
|
||||
svc_obj = service.Service()
|
||||
attr_name = 'powerOn'
|
||||
service_mock = svc_obj.client.service
|
||||
setattr(service_mock, attr_name, side_effect)
|
||||
ret = svc_obj.powerOn(managed_object)
|
||||
self.assertEqual(resp, ret)
|
||||
|
||||
def test_request_handler_with_retrieve_properties_ex_fault(self):
|
||||
managed_object = 'Datacenter'
|
||||
|
||||
def side_effect(mo, **kwargs):
|
||||
self.assertEqual(managed_object, mo._type)
|
||||
self.assertEqual(managed_object, mo.value)
|
||||
return None
|
||||
|
||||
svc_obj = service.Service()
|
||||
attr_name = 'retrievePropertiesEx'
|
||||
service_mock = svc_obj.client.service
|
||||
setattr(service_mock, attr_name, side_effect)
|
||||
self.assertRaises(exceptions.VimFaultException,
|
||||
svc_obj.retrievePropertiesEx,
|
||||
managed_object)
|
||||
|
||||
def test_request_handler_with_web_fault(self):
|
||||
managed_object = 'VirtualMachine'
|
||||
fault_list = ['Fault']
|
||||
|
||||
def side_effect(mo, **kwargs):
|
||||
self.assertEqual(managed_object, mo._type)
|
||||
self.assertEqual(managed_object, mo.value)
|
||||
fault_string = mock.Mock()
|
||||
fault_string.getText.return_value = "MyFault"
|
||||
|
||||
fault_children = mock.Mock()
|
||||
fault_children.name = "name"
|
||||
fault_children.getText.return_value = "value"
|
||||
child = mock.Mock()
|
||||
child.get.return_value = fault_list[0]
|
||||
child.getChildren.return_value = [fault_children]
|
||||
detail = mock.Mock()
|
||||
detail.getChildren.return_value = [child]
|
||||
doc = mock.Mock()
|
||||
doc.childAtPath = mock.Mock(side_effect=[fault_string, detail])
|
||||
raise suds.WebFault(None, doc)
|
||||
|
||||
svc_obj = service.Service()
|
||||
attr_name = 'powerOn'
|
||||
service_mock = svc_obj.client.service
|
||||
setattr(service_mock, attr_name, side_effect)
|
||||
|
||||
try:
|
||||
svc_obj.powerOn(managed_object)
|
||||
except exceptions.VimFaultException as ex:
|
||||
self.assertEqual(fault_list, ex.fault_list)
|
||||
self.assertEqual({'name': 'value'}, ex.details)
|
||||
self.assertEqual("MyFault", ex.msg)
|
||||
|
||||
def test_request_handler_with_attribute_error(self):
|
||||
managed_object = 'VirtualMachine'
|
||||
svc_obj = service.Service()
|
||||
# no powerOn method in Service
|
||||
service_mock = mock.Mock(spec=service.Service)
|
||||
svc_obj.client.service = service_mock
|
||||
self.assertRaises(exceptions.VimAttributeException,
|
||||
svc_obj.powerOn,
|
||||
managed_object)
|
||||
|
||||
def test_request_handler_with_http_cannot_send_error(self):
|
||||
managed_object = 'VirtualMachine'
|
||||
|
||||
def side_effect(mo, **kwargs):
|
||||
self.assertEqual(managed_object, mo._type)
|
||||
self.assertEqual(managed_object, mo.value)
|
||||
raise httplib.CannotSendRequest()
|
||||
|
||||
svc_obj = service.Service()
|
||||
attr_name = 'powerOn'
|
||||
service_mock = svc_obj.client.service
|
||||
setattr(service_mock, attr_name, side_effect)
|
||||
self.assertRaises(exceptions.VimSessionOverLoadException,
|
||||
svc_obj.powerOn,
|
||||
managed_object)
|
||||
|
||||
def test_request_handler_with_http_response_not_ready_error(self):
|
||||
managed_object = 'VirtualMachine'
|
||||
|
||||
def side_effect(mo, **kwargs):
|
||||
self.assertEqual(managed_object, mo._type)
|
||||
self.assertEqual(managed_object, mo.value)
|
||||
raise httplib.ResponseNotReady()
|
||||
|
||||
svc_obj = service.Service()
|
||||
attr_name = 'powerOn'
|
||||
service_mock = svc_obj.client.service
|
||||
setattr(service_mock, attr_name, side_effect)
|
||||
self.assertRaises(exceptions.VimSessionOverLoadException,
|
||||
svc_obj.powerOn,
|
||||
managed_object)
|
||||
|
||||
def test_request_handler_with_http_cannot_send_header_error(self):
|
||||
managed_object = 'VirtualMachine'
|
||||
|
||||
def side_effect(mo, **kwargs):
|
||||
self.assertEqual(managed_object, mo._type)
|
||||
self.assertEqual(managed_object, mo.value)
|
||||
raise httplib.CannotSendHeader()
|
||||
|
||||
svc_obj = service.Service()
|
||||
attr_name = 'powerOn'
|
||||
service_mock = svc_obj.client.service
|
||||
setattr(service_mock, attr_name, side_effect)
|
||||
self.assertRaises(exceptions.VimSessionOverLoadException,
|
||||
svc_obj.powerOn,
|
||||
managed_object)
|
||||
|
||||
def test_request_handler_with_url_error(self):
|
||||
managed_object = 'VirtualMachine'
|
||||
|
||||
def side_effect(mo, **kwargs):
|
||||
self.assertEqual(managed_object, mo._type)
|
||||
self.assertEqual(managed_object, mo.value)
|
||||
raise urllib2.URLError(None)
|
||||
|
||||
svc_obj = service.Service()
|
||||
attr_name = 'powerOn'
|
||||
service_mock = svc_obj.client.service
|
||||
setattr(service_mock, attr_name, side_effect)
|
||||
self.assertRaises(exceptions.VimConnectionException,
|
||||
svc_obj.powerOn,
|
||||
managed_object)
|
||||
|
||||
def test_request_handler_with_http_error(self):
|
||||
managed_object = 'VirtualMachine'
|
||||
|
||||
def side_effect(mo, **kwargs):
|
||||
self.assertEqual(managed_object, mo._type)
|
||||
self.assertEqual(managed_object, mo.value)
|
||||
raise urllib2.HTTPError(None, None, None, None, None)
|
||||
|
||||
svc_obj = service.Service()
|
||||
attr_name = 'powerOn'
|
||||
service_mock = svc_obj.client.service
|
||||
setattr(service_mock, attr_name, side_effect)
|
||||
self.assertRaises(exceptions.VimConnectionException,
|
||||
svc_obj.powerOn,
|
||||
managed_object)
|
||||
|
||||
@mock.patch.object(vim_util, 'get_moref', return_value=None)
|
||||
def test_request_handler_no_value(self, mock_moref):
|
||||
managed_object = 'VirtualMachine'
|
||||
svc_obj = service.Service()
|
||||
ret = svc_obj.UnregisterVM(managed_object)
|
||||
self.assertIsNone(ret)
|
||||
|
||||
def _test_request_handler_with_exception(self, message, exception):
|
||||
managed_object = 'VirtualMachine'
|
||||
|
||||
def side_effect(mo, **kwargs):
|
||||
self.assertEqual(managed_object, mo._type)
|
||||
self.assertEqual(managed_object, mo.value)
|
||||
raise Exception(message)
|
||||
|
||||
svc_obj = service.Service()
|
||||
attr_name = 'powerOn'
|
||||
service_mock = svc_obj.client.service
|
||||
setattr(service_mock, attr_name, side_effect)
|
||||
self.assertRaises(exception, svc_obj.powerOn, managed_object)
|
||||
|
||||
def test_request_handler_with_address_in_use_error(self):
|
||||
self._test_request_handler_with_exception(
|
||||
service.ADDRESS_IN_USE_ERROR,
|
||||
exceptions.VimSessionOverLoadException)
|
||||
|
||||
def test_request_handler_with_conn_abort_error(self):
|
||||
self._test_request_handler_with_exception(
|
||||
service.CONN_ABORT_ERROR, exceptions.VimSessionOverLoadException)
|
||||
|
||||
def test_request_handler_with_resp_not_xml_error(self):
|
||||
self._test_request_handler_with_exception(
|
||||
service.RESP_NOT_XML_ERROR, exceptions.VimSessionOverLoadException)
|
||||
|
||||
def test_request_handler_with_generic_error(self):
|
||||
self._test_request_handler_with_exception(
|
||||
'GENERIC_ERROR', exceptions.VimException)
|
||||
|
||||
def test_get_session_cookie(self):
|
||||
svc_obj = service.Service()
|
||||
cookie_value = 'xyz'
|
||||
cookie = mock.Mock()
|
||||
cookie.name = 'vmware_soap_session'
|
||||
cookie.value = cookie_value
|
||||
svc_obj.client.options.transport.cookiejar = [cookie]
|
||||
self.assertEqual(cookie_value, svc_obj.get_http_cookie())
|
||||
|
||||
def test_get_session_cookie_with_no_cookie(self):
|
||||
svc_obj = service.Service()
|
||||
cookie = mock.Mock()
|
||||
cookie.name = 'cookie'
|
||||
cookie.value = 'xyz'
|
||||
svc_obj.client.options.transport.cookiejar = [cookie]
|
||||
self.assertIsNone(svc_obj.get_http_cookie())
|
||||
@@ -17,37 +17,13 @@
|
||||
Unit tests for classes to invoke VMware VI SOAP calls.
|
||||
"""
|
||||
|
||||
import httplib
|
||||
import urllib2
|
||||
|
||||
import mock
|
||||
import suds
|
||||
|
||||
from oslo.vmware import exceptions
|
||||
from oslo.vmware import vim
|
||||
from oslo.vmware import vim_util
|
||||
from tests import base
|
||||
|
||||
|
||||
class VimMessagePluginTest(base.TestCase):
|
||||
"""Test class for VimMessagePlugin."""
|
||||
|
||||
def test_add_attribute_for_value(self):
|
||||
node = mock.Mock()
|
||||
node.name = 'value'
|
||||
plugin = vim.VimMessagePlugin()
|
||||
plugin.add_attribute_for_value(node)
|
||||
node.set.assert_called_once_with('xsi:type', 'xsd:string')
|
||||
|
||||
def test_marshalled(self):
|
||||
plugin = vim.VimMessagePlugin()
|
||||
context = mock.Mock()
|
||||
plugin.marshalled(context)
|
||||
context.envelope.prune.assert_called_once_with()
|
||||
context.envelope.walk.assert_called_once_with(
|
||||
plugin.add_attribute_for_value)
|
||||
|
||||
|
||||
class VimTest(base.TestCase):
|
||||
"""Test class for Vim."""
|
||||
|
||||
@@ -58,234 +34,16 @@ class VimTest(base.TestCase):
|
||||
self.SudsClientMock = patcher.start()
|
||||
|
||||
@mock.patch.object(vim.Vim, '__getattr__', autospec=True)
|
||||
def test_init(self, getattr_mock):
|
||||
def test_service_content(self, getattr_mock):
|
||||
getattr_ret = mock.Mock()
|
||||
getattr_mock.side_effect = lambda *args: getattr_ret
|
||||
vim_obj = vim.Vim()
|
||||
vim_obj.service_content
|
||||
getattr_mock.assert_called_once_with(vim_obj, 'RetrieveServiceContent')
|
||||
getattr_ret.assert_called_once_with('ServiceInstance')
|
||||
self.assertEqual(self.SudsClientMock.return_value, vim_obj.client)
|
||||
self.assertEqual(getattr_ret.return_value, vim_obj.service_content)
|
||||
|
||||
def test_retrieve_properties_ex_fault_checker_with_empty_response(self):
|
||||
try:
|
||||
vim.Vim._retrieve_properties_ex_fault_checker(None)
|
||||
assert False
|
||||
except exceptions.VimFaultException as ex:
|
||||
self.assertEqual([exceptions.NOT_AUTHENTICATED],
|
||||
ex.fault_list)
|
||||
|
||||
def test_retrieve_properties_ex_fault_checker(self):
|
||||
fault_list = ['FileFault', 'VimFault']
|
||||
missing_set = []
|
||||
for fault in fault_list:
|
||||
missing_elem = mock.Mock()
|
||||
missing_elem.fault.fault.__class__.__name__ = fault
|
||||
missing_set.append(missing_elem)
|
||||
obj_cont = mock.Mock()
|
||||
obj_cont.missingSet = missing_set
|
||||
response = mock.Mock()
|
||||
response.objects = [obj_cont]
|
||||
|
||||
try:
|
||||
vim.Vim._retrieve_properties_ex_fault_checker(response)
|
||||
assert False
|
||||
except exceptions.VimFaultException as ex:
|
||||
self.assertEqual(fault_list, ex.fault_list)
|
||||
|
||||
def test_vim_request_handler(self):
|
||||
managed_object = 'VirtualMachine'
|
||||
resp = mock.Mock()
|
||||
|
||||
def side_effect(mo, **kwargs):
|
||||
self.assertEqual(managed_object, mo._type)
|
||||
self.assertEqual(managed_object, mo.value)
|
||||
return resp
|
||||
|
||||
vim_obj = vim.Vim()
|
||||
attr_name = 'powerOn'
|
||||
service_mock = vim_obj._client.service
|
||||
setattr(service_mock, attr_name, side_effect)
|
||||
ret = vim_obj.powerOn(managed_object)
|
||||
self.assertEqual(resp, ret)
|
||||
|
||||
def test_vim_request_handler_with_retrieve_properties_ex_fault(self):
|
||||
managed_object = 'Datacenter'
|
||||
|
||||
def side_effect(mo, **kwargs):
|
||||
self.assertEqual(managed_object, mo._type)
|
||||
self.assertEqual(managed_object, mo.value)
|
||||
return None
|
||||
|
||||
vim_obj = vim.Vim()
|
||||
attr_name = 'retrievePropertiesEx'
|
||||
service_mock = vim_obj._client.service
|
||||
setattr(service_mock, attr_name, side_effect)
|
||||
self.assertRaises(exceptions.VimFaultException,
|
||||
vim_obj.retrievePropertiesEx,
|
||||
managed_object)
|
||||
|
||||
def test_vim_request_handler_with_web_fault(self):
|
||||
managed_object = 'VirtualMachine'
|
||||
fault_list = ['Fault']
|
||||
|
||||
def side_effect(mo, **kwargs):
|
||||
self.assertEqual(managed_object, mo._type)
|
||||
self.assertEqual(managed_object, mo.value)
|
||||
fault_string = mock.Mock()
|
||||
fault_string.getText.return_value = "MyFault"
|
||||
|
||||
fault_children = mock.Mock()
|
||||
fault_children.name = "name"
|
||||
fault_children.getText.return_value = "value"
|
||||
child = mock.Mock()
|
||||
child.get.return_value = fault_list[0]
|
||||
child.getChildren.return_value = [fault_children]
|
||||
detail = mock.Mock()
|
||||
detail.getChildren.return_value = [child]
|
||||
doc = mock.Mock()
|
||||
doc.childAtPath = mock.Mock(side_effect=[fault_string, detail])
|
||||
raise suds.WebFault(None, doc)
|
||||
|
||||
vim_obj = vim.Vim()
|
||||
attr_name = 'powerOn'
|
||||
service_mock = vim_obj._client.service
|
||||
setattr(service_mock, attr_name, side_effect)
|
||||
|
||||
try:
|
||||
vim_obj.powerOn(managed_object)
|
||||
except exceptions.VimFaultException as ex:
|
||||
self.assertEqual(fault_list, ex.fault_list)
|
||||
self.assertEqual({'name': 'value'}, ex.details)
|
||||
self.assertEqual("MyFault", ex.msg)
|
||||
|
||||
def test_vim_request_handler_with_attribute_error(self):
|
||||
managed_object = 'VirtualMachine'
|
||||
vim_obj = vim.Vim()
|
||||
# no powerOn method in Vim
|
||||
service_mock = mock.Mock(spec=vim.Vim)
|
||||
vim_obj._client.service = service_mock
|
||||
self.assertRaises(exceptions.VimAttributeException,
|
||||
vim_obj.powerOn,
|
||||
managed_object)
|
||||
|
||||
def test_vim_request_handler_with_http_cannot_send_error(self):
|
||||
managed_object = 'VirtualMachine'
|
||||
|
||||
def side_effect(mo, **kwargs):
|
||||
self.assertEqual(managed_object, mo._type)
|
||||
self.assertEqual(managed_object, mo.value)
|
||||
raise httplib.CannotSendRequest()
|
||||
|
||||
vim_obj = vim.Vim()
|
||||
attr_name = 'powerOn'
|
||||
service_mock = vim_obj._client.service
|
||||
setattr(service_mock, attr_name, side_effect)
|
||||
self.assertRaises(exceptions.VimSessionOverLoadException,
|
||||
vim_obj.powerOn,
|
||||
managed_object)
|
||||
|
||||
def test_vim_request_handler_with_http_response_not_ready_error(self):
|
||||
managed_object = 'VirtualMachine'
|
||||
|
||||
def side_effect(mo, **kwargs):
|
||||
self.assertEqual(managed_object, mo._type)
|
||||
self.assertEqual(managed_object, mo.value)
|
||||
raise httplib.ResponseNotReady()
|
||||
|
||||
vim_obj = vim.Vim()
|
||||
attr_name = 'powerOn'
|
||||
service_mock = vim_obj._client.service
|
||||
setattr(service_mock, attr_name, side_effect)
|
||||
self.assertRaises(exceptions.VimSessionOverLoadException,
|
||||
vim_obj.powerOn,
|
||||
managed_object)
|
||||
|
||||
def test_vim_request_handler_with_http_cannot_send_header_error(self):
|
||||
managed_object = 'VirtualMachine'
|
||||
|
||||
def side_effect(mo, **kwargs):
|
||||
self.assertEqual(managed_object, mo._type)
|
||||
self.assertEqual(managed_object, mo.value)
|
||||
raise httplib.CannotSendHeader()
|
||||
|
||||
vim_obj = vim.Vim()
|
||||
attr_name = 'powerOn'
|
||||
service_mock = vim_obj._client.service
|
||||
setattr(service_mock, attr_name, side_effect)
|
||||
self.assertRaises(exceptions.VimSessionOverLoadException,
|
||||
vim_obj.powerOn,
|
||||
managed_object)
|
||||
|
||||
def test_vim_request_handler_with_url_error(self):
|
||||
managed_object = 'VirtualMachine'
|
||||
|
||||
def side_effect(mo, **kwargs):
|
||||
self.assertEqual(managed_object, mo._type)
|
||||
self.assertEqual(managed_object, mo.value)
|
||||
raise urllib2.URLError(None)
|
||||
|
||||
vim_obj = vim.Vim()
|
||||
attr_name = 'powerOn'
|
||||
service_mock = vim_obj._client.service
|
||||
setattr(service_mock, attr_name, side_effect)
|
||||
self.assertRaises(exceptions.VimConnectionException,
|
||||
vim_obj.powerOn,
|
||||
managed_object)
|
||||
|
||||
def test_vim_request_handler_with_http_error(self):
|
||||
managed_object = 'VirtualMachine'
|
||||
|
||||
def side_effect(mo, **kwargs):
|
||||
self.assertEqual(managed_object, mo._type)
|
||||
self.assertEqual(managed_object, mo.value)
|
||||
raise urllib2.HTTPError(None, None, None, None, None)
|
||||
|
||||
vim_obj = vim.Vim()
|
||||
attr_name = 'powerOn'
|
||||
service_mock = vim_obj._client.service
|
||||
setattr(service_mock, attr_name, side_effect)
|
||||
self.assertRaises(exceptions.VimConnectionException,
|
||||
vim_obj.powerOn,
|
||||
managed_object)
|
||||
|
||||
@mock.patch.object(vim_util, 'get_moref', return_value=None)
|
||||
def test_vim_request_handler_no_value(self, mock_moref):
|
||||
managed_object = 'VirtualMachine'
|
||||
vim_obj = vim.Vim()
|
||||
ret = vim_obj.UnregisterVM(managed_object)
|
||||
self.assertIsNone(ret)
|
||||
|
||||
def _test_vim_request_handler_with_exception(self, message, exception):
|
||||
managed_object = 'VirtualMachine'
|
||||
|
||||
def side_effect(mo, **kwargs):
|
||||
self.assertEqual(managed_object, mo._type)
|
||||
self.assertEqual(managed_object, mo.value)
|
||||
raise Exception(message)
|
||||
|
||||
vim_obj = vim.Vim()
|
||||
attr_name = 'powerOn'
|
||||
service_mock = vim_obj._client.service
|
||||
setattr(service_mock, attr_name, side_effect)
|
||||
self.assertRaises(exception, vim_obj.powerOn, managed_object)
|
||||
|
||||
def test_vim_request_handler_with_address_in_use_error(self):
|
||||
self._test_vim_request_handler_with_exception(
|
||||
vim.ADDRESS_IN_USE_ERROR, exceptions.VimSessionOverLoadException)
|
||||
|
||||
def test_vim_request_handler_with_conn_abort_error(self):
|
||||
self._test_vim_request_handler_with_exception(
|
||||
vim.CONN_ABORT_ERROR, exceptions.VimSessionOverLoadException)
|
||||
|
||||
def test_vim_request_handler_with_resp_not_xml_error(self):
|
||||
self._test_vim_request_handler_with_exception(
|
||||
vim.RESP_NOT_XML_ERROR, exceptions.VimSessionOverLoadException)
|
||||
|
||||
def test_vim_request_handler_with_generic_error(self):
|
||||
self._test_vim_request_handler_with_exception(
|
||||
'GENERIC_ERROR', exceptions.VimException)
|
||||
|
||||
def test_exception_summary_exception_as_list(self):
|
||||
# assert that if a list is fed to the VimException object
|
||||
# that it will error.
|
||||
@@ -340,3 +98,10 @@ class VimTest(base.TestCase):
|
||||
vim_obj.wsdl_url)
|
||||
self.assertEqual('https://[::1]:12345/sdk',
|
||||
vim_obj.soap_url)
|
||||
|
||||
def test_configure_with_wsdl_url_override(self):
|
||||
vim_obj = vim.Vim('https', 'www.example.com',
|
||||
wsdl_url='https://test.com/sdk/vimService.wsdl')
|
||||
self.assertEqual('https://test.com/sdk/vimService.wsdl',
|
||||
vim_obj.wsdl_url)
|
||||
self.assertEqual('https://www.example.com/sdk', vim_obj.soap_url)
|
||||
|
||||
@@ -292,18 +292,3 @@ class VimUtilTest(base.TestCase):
|
||||
self.assertEqual(prop.val, val)
|
||||
get_object_properties.assert_called_once_with(
|
||||
vim, moref, [property_name])
|
||||
|
||||
def test_configure_without_wsdl_loc_override(self):
|
||||
wsdl_url = vim_util.get_wsdl_url("https", "www.example.com")
|
||||
url = vim_util.get_soap_url("https", "www.example.com")
|
||||
self.assertEqual("https://www.example.com/sdk/vimService.wsdl",
|
||||
wsdl_url)
|
||||
self.assertEqual("https://www.example.com/sdk", url)
|
||||
|
||||
def test_configure_without_wsdl_loc_override_using_ipv6(self):
|
||||
# Same as above but with ipv6 based host ip
|
||||
wsdl_url = vim_util.get_wsdl_url("https", "::1")
|
||||
url = vim_util.get_soap_url("https", "::1")
|
||||
self.assertEqual("https://[::1]/sdk/vimService.wsdl",
|
||||
wsdl_url)
|
||||
self.assertEqual("https://[::1]/sdk", url)
|
||||
|
||||
Reference in New Issue
Block a user