from oslo_vmware import vim_util, exceptions
from networking_dvs.common import config as dvs_config, util as c_util
from networking_dvs.utils import dvs_util
from networking_dvs.utils import spec_builder
CONF = dvs_config.CONF
LOG = log.getLogger(__name__)
SECURITY_GROUPS_ATTRIBUTE = "NeutronSecurityGroups"
class RequestCanceledException(exceptions.VimException):
msg_fmt = _("The task was canceled by a user.")
code = 200
exceptions.register_fault_class('RequestCanceled', RequestCanceledException)
def _create_session(config):
"""Create Vcenter Session for API Calling."""
kwargs = {'create_session': True}
connection = dvs_util.connect(config, **kwargs)
atexit.register(connection.logout)
return connection
def _cast(value, _type=str):
if value is None:
return None
return _type(value)
def test_register_fault_class_override(self):
exc = self._create_subclass_exception()
exceptions.register_fault_class(exceptions.ALREADY_EXISTS, exc)
self.assertEqual(exc,
exceptions.get_fault_class(exceptions.ALREADY_EXISTS))
def test_register_fault_class(self):
exc = self._create_subclass_exception()
exceptions.register_fault_class('ValueError', exc)
self.assertEqual(exc, exceptions.get_fault_class('ValueError'))
def test_register_fault_class_override(self):
exc = self._create_subclass_exception()
exceptions.register_fault_class(exceptions.ALREADY_EXISTS, exc)
self.assertEqual(exc,
exceptions.get_fault_class(exceptions.ALREADY_EXISTS))
def test_register_fault_class(self):
exc = self._create_subclass_exception()
exceptions.register_fault_class('ValueError', exc)
self.assertEqual(exc, exceptions.get_fault_class('ValueError'))
def dvs_register_exceptions():
oslo_vmware_exc.register_fault_class('DvsOperationBulkFault',
DvsOperationBulkFault)
