import time import os import sys sys.path.insert(0, os.environ["SWAGGER_CLIENT_PATH"]) import v2_swagger_client from swagger_client.rest import ApiException import swagger_client.models from pprint import pprint admin_user = "admin" admin_pwd = "Harbor12345" harbor_server = os.environ["HARBOR_HOST"] #CLIENT=dict(endpoint="https://"+harbor_server+"/api") ADMIN_CLIENT=dict(endpoint = os.environ.get("HARBOR_HOST_SCHEMA", "https")+ "://"+harbor_server+"/api/v2.0", username = admin_user, password = admin_pwd) CHART_API_CLIENT=dict(endpoint = os.environ.get("HARBOR_HOST_SCHEMA", "https")+ "://"+harbor_server+"/api", username = admin_user, password = admin_pwd) USER_ROLE=dict(admin=0,normal=1) TEARDOWN = os.environ.get('TEARDOWN', 'true').lower() in ('true', 'yes') def GetProductApi(username, password, harbor_server= os.environ["HARBOR_HOST"]): cfg = swagger_client.Configuration() cfg.host = "https://"+harbor_server+"/api/v2.0" cfg.username = username cfg.password = password cfg.verify_ssl = False cfg.debug = True api_client = swagger_client.ApiClient(cfg) api_instance = swagger_client.ProductsApi(api_client) return api_instance def GetRepositoryApi(username, password, harbor_server= os.environ["HARBOR_HOST"]): cfg = v2_swagger_client.Configuration() cfg.host = "https://"+harbor_server+"/api/v2.0" cfg.username = username cfg.password = password cfg.verify_ssl = False cfg.debug = True api_client = v2_swagger_client.ApiClient(cfg) api_instance = v2_swagger_client.RepositoryApi(api_client) return api_instance class TestResult(object): def __init__(self): self.num_errors = 0 self.error_message = [] def add_test_result(self, error_message): self.num_errors = self.num_errors + 1 self.error_message.append(error_message) def get_final_result(self): if self.num_errors > 0: for each_err_msg in self.error_message: print "Error message:", each_err_msg raise Exception(r"Test case failed with {} errors.".format(self.num_errors)) from contextlib import contextmanager @contextmanager def created_user(password): from library.user import User api = User() user_id, user_name = api.create_user(user_password=password, **ADMIN_CLIENT) try: yield (user_id, user_name) finally: if TEARDOWN: api.delete_user(user_id, **ADMIN_CLIENT) @contextmanager def created_project(name=None, metadata=None, user_id=None, member_role_id=None): from library.project import Project api = Project() project_id, project_name = api.create_project(name=None, metadata=None, **ADMIN_CLIENT) if user_id: api.add_project_members(project_id, user_id, member_role_id=member_role_id, **ADMIN_CLIENT) try: yield (project_id, project_name) finally: if TEARDOWN: api.delete_project(project_id, **ADMIN_CLIENT)