diff --git a/smoke-test/tests/utils.py b/smoke-test/tests/utils.py index cf744c3b1d..5fd3d49a1a 100644 --- a/smoke-test/tests/utils.py +++ b/smoke-test/tests/utils.py @@ -8,6 +8,7 @@ from joblib import Parallel, delayed import requests_wrapper as requests from datahub.cli import cli_utils +from datahub.cli.cli_utils import get_system_auth from datahub.ingestion.run.pipeline import Pipeline TIME: int = 1581407189000 @@ -19,10 +20,16 @@ def get_frontend_session(): headers = { "Content-Type": "application/json", } - username, password = get_admin_credentials() - data = '{"username":"' + username + '", "password":"' + password + '"}' - response = session.post(f"{get_frontend_url()}/logIn", headers=headers, data=data) - response.raise_for_status() + system_auth = get_system_auth() + if system_auth is not None: + session.headers.update({"Authorization": system_auth}) + else: + username, password = get_admin_credentials() + data = '{"username":"' + username + '", "password":"' + password + '"}' + response = session.post( + f"{get_frontend_url()}/logIn", headers=headers, data=data + ) + response.raise_for_status() return session