magnum-tempest-plugin/magnum_tempest_plugin/common/manager.py
Diogo Guerra dc7f0ba4aa Drop bay and baymodel
Change-Id: Ib85e4fda8e4ac467bd49590dc72ba5913bb9a19d
Signed-off-by: Diogo Guerra <diogo.filipe.tomas.guerra@cern.ch>
2023-02-05 11:17:48 +00:00

64 lines
2.6 KiB
Python

# 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.
from tempest import clients
from tempest.common import credentials_factory as common_creds
from magnum_tempest_plugin.common import client
from magnum_tempest_plugin.common import config
from magnum_tempest_plugin.tests.api.v1.clients import cert_client
from magnum_tempest_plugin.tests.api.v1.clients import cluster_client
from magnum_tempest_plugin.tests.api.v1.clients import cluster_template_client
from magnum_tempest_plugin.tests.api.v1.clients import magnum_service_client
class Manager(clients.Manager):
def __init__(self, credentials=None, request_type=None):
if not credentials:
credentials = common_creds.get_configured_credentials(
'identity_admin')
super(Manager, self).__init__(credentials)
self.auth_provider.orig_base_url = self.auth_provider.base_url
self.auth_provider.base_url = self.bypassed_base_url
auth = self.auth_provider
if request_type == 'cert':
self.client = cert_client.CertClient(auth)
elif request_type == 'cluster_template':
self.client = cluster_template_client.ClusterTemplateClient(auth)
elif request_type == 'cluster':
self.client = cluster_client.ClusterClient(auth)
elif request_type == 'service':
self.client = magnum_service_client.MagnumServiceClient(auth)
else:
self.client = client.MagnumClient(auth)
def bypassed_base_url(self, filters, auth_data=None):
if (config.Config.magnum_url
and filters['service'] == 'container-infra'):
return config.Config.magnum_url
return self.auth_provider.orig_base_url(filters, auth_data=auth_data)
class DefaultManager(Manager):
def __init__(self, credentials, request_type=None):
super(DefaultManager, self).__init__(credentials, request_type)
class AltManager(Manager):
def __init__(self, credentials, request_type=None):
super(AltManager, self).__init__(credentials, request_type)
class AdminManager(Manager):
def __init__(self, credentials, request_type=None):
super(AdminManager, self).__init__(credentials, request_type)