Skip to content

Commit

Permalink
ci: update performance test script for breaking Determined change (#8961
Browse files Browse the repository at this point in the history
)
  • Loading branch information
NicholasBlaskey authored Mar 6, 2024
1 parent b2b85d7 commit 68017dd
Showing 1 changed file with 8 additions and 8 deletions.
16 changes: 8 additions & 8 deletions .circleci/scripts/wait_for_perf_migration_upload_results.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand All @@ -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
Expand Down

0 comments on commit 68017dd

Please sign in to comment.