From 68017dd0d81d52543378c62971cb1c41b245a601 Mon Sep 17 00:00:00 2001 From: Nicholas Blaskey Date: Wed, 6 Mar 2024 12:02:12 -0500 Subject: [PATCH] ci: update performance test script for breaking Determined change (#8961) --- .../wait_for_perf_migration_upload_results.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/.circleci/scripts/wait_for_perf_migration_upload_results.py b/.circleci/scripts/wait_for_perf_migration_upload_results.py index 1e6bebc5084..7e705c14221 100644 --- a/.circleci/scripts/wait_for_perf_migration_upload_results.py +++ b/.circleci/scripts/wait_for_perf_migration_upload_results.py @@ -14,12 +14,15 @@ def _wait_for_master() -> None: - print("Checking for master at 127.0.0.1") + print("Checking for master") + cert = certs.Cert(noverify=True) + sess = api.UnauthSession("http://127.0.0.1:8080", cert) + # 2 hours is the most a migration can take, with this setup. # If a migration takes longer than that we have hit an issue a customer will likely hit too. for i in range(2 * 60 * 60): try: - r = api.get("127.0.0.1", "info", authenticated=False) + r = sess.get("info") if r.status_code == requests.codes.ok: return except api.errors.MasterNotFoundException: @@ -31,12 +34,9 @@ def _wait_for_master() -> None: def _upload_migration_length(conn: extensions.connection) -> None: - authentication.cli_auth = authentication.Authentication( - "http://127.0.0.1:8080", - requested_user="admin", - password="", - ) - sess = api.Session("http://127.0.0.1:8080", "admin", authentication.cli_auth, certs.cli_cert) + cert = certs.Cert(noverify=True) + utp = authentication.login("http://127.0.0.1:8080", "admin", "", cert) + sess = api.Session("http://127.0.0.1:8080", utp, cert) migration_start_log = None migration_end_log = None