diff --git a/octaviaclient/api/v2/octavia.py b/octaviaclient/api/v2/octavia.py index 297b635..1d883c5 100644 --- a/octaviaclient/api/v2/octavia.py +++ b/octaviaclient/api/v2/octavia.py @@ -20,6 +20,8 @@ from osc_lib import exceptions as osc_exc from octaviaclient.api import constants as const from octaviaclient.api import exceptions +OctaviaClientException = exceptions.OctaviaClientException + def correct_return_codes(func): _status_dict = {400: 'Bad Request', 401: 'Unauthorized', @@ -46,7 +48,7 @@ def correct_return_codes(func): else: raise - raise exceptions.OctaviaClientException( + raise OctaviaClientException( code=code, message=message, request_id=request_id) diff --git a/octaviaclient/tests/unit/api/test_octavia.py b/octaviaclient/tests/unit/api/test_octavia.py index 59d7865..19562f8 100644 --- a/octaviaclient/tests/unit/api/test_octavia.py +++ b/octaviaclient/tests/unit/api/test_octavia.py @@ -155,6 +155,12 @@ SINGLE_FVPF_RESP = {'flavorprofile': {'id': FAKE_FVPF, 'name': 'fvpf1'}} SINGLE_FVPF_UPDATE = {'flavorprofile': {'provider_name': 'fake_provider'}} +class TestAPI(utils.TestCase): + def test_client_exception(self): + self.assertIs(octavia.OctaviaClientException, + exceptions.OctaviaClientException) + + class TestOctaviaClient(utils.TestCase): def setUp(self):