From af5e79af662f1a7dc2432ebbea0af8d02e2965f2 Mon Sep 17 00:00:00 2001 From: Vincent Koc Date: Wed, 5 Oct 2022 21:31:15 +1100 Subject: [PATCH] Update source.py --- .../connectors/source-zuora/source_zuora/source.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/airbyte-integrations/connectors/source-zuora/source_zuora/source.py b/airbyte-integrations/connectors/source-zuora/source_zuora/source.py index 6e620d698411b..7eb00bc8ff739 100644 --- a/airbyte-integrations/connectors/source-zuora/source_zuora/source.py +++ b/airbyte-integrations/connectors/source-zuora/source_zuora/source.py @@ -503,9 +503,14 @@ def check_connection(self, logger: AirbyteLogger, config: Mapping[str, Any]) -> """ Testing connection availability for the connector by granting the token. """ - auth = ZuoraAuthenticator(config).get_auth() try: - auth.get_auth_header() + url = f"{config['domain_url']}/query/jobs" + + authenticator = ZuoraAuthenticator(config) + + session = requests.get(url, headers=authenticator.get_auth_header()) + session.raise_for_status() + return True, None except Exception as e: return False, e @@ -516,7 +521,7 @@ def streams(self, config: Mapping[str, Any]) -> List[ZuoraStream]: Defining streams to run by building stream classes dynamically. """ auth = ZuoraAuthenticator(config) - config["authenticator"] = auth.get_auth() + config["authenticator"] = auth config["url_base"] = auth.url_base # List available objects (streams) names from Zuora