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)