def __init__(self, registry_url, proxies=None, user_agent=None, backoff_strategy=None, adapters=None, request_timeout=None, model_resolution=None, correlation_id=None, user_id=None, application_id=None, content_type=None, **auth_kwargs): logging.debug('Client initialising with registry_url=%s ', registry_url) self._registry_url = registry_url self._request_timeout = request_timeout or env.DEFAULT_REQUEST_TIMEOUT_SECONDS self._proxies = proxies self._user_agent = user_agent self._correlation_id = correlation_id.strip() if correlation_id else None self.user_id = user_id.strip() if user_id else None self.application_id = application_id.strip() if application_id else None self._model_resolution = model_resolution self._registry = self._initialize_registry(adapters, backoff_strategy, content_type, **auth_kwargs) self._auth = AuthFactory.create(token_url=self._get_token_url(auth_kwargs.get("auth_type", None)), request_timeout=self._request_timeout, **auth_kwargs) self._auth.register_adapters(adapters) self._auth.init_session() self._http = http.HttpExecutor(self._auth, proxies=self._proxies, user_agent=self._user_agent, session=self._auth.session, request_timeout=self._request_timeout, correlation_id=self._correlation_id, user_id=self.user_id, application_id=self.application_id, backoff_strategy=backoff_strategy, content_type=content_type)
def test_should_create_auth_with_certificates_on_session(self): auth = AuthFactory.create(auth_type=AuthType.MUTUAL, client_cert='/certs/client_cert.pem', client_key='/certs/client_key.pem', server_cert='/certs/server_cert.pem') assert_that( auth.session.cert, equal_to(('/certs/client_cert.pem', '/certs/client_key.pem'))) assert_that(auth.session.verify, equal_to('/certs/server_cert.pem'))
def _initialize_registry(self, adapters, backoff_strategy, content_type, **auth_kwargs): auth = AuthFactory.create(**auth_kwargs) if auth_kwargs.get( "auth_type", None) == AuthType.MUTUAL else AuthFactory.create( auth_type=AuthType.NO_AUTH) auth.register_adapters(adapters) http_executor = http.HttpExecutor( auth, proxies=self._proxies, user_agent=self._user_agent, session=auth.session, request_timeout=self._request_timeout, correlation_id=self._correlation_id, user_id=self.user_id, application_id=self.application_id, backoff_strategy=backoff_strategy, content_type=content_type) return registry.Registry(self._registry_url, http_executor)
def _initialize_registry(self, adapters, backoff_strategy): auth = AuthFactory.create(auth_type=AuthType.NO_AUTH) auth.register_adapters(adapters) http_executor = http.HttpExecutor( auth, proxies=self._proxies, user_agent=self._user_agent, session=auth.session, request_timeout=self._request_timeout, correlation_id=self._correlation_id, backoff_strategy=backoff_strategy) return registry.Registry(self._registry_url, http_executor)