From 430b82024fa88575de6d19399e8da6047962b77d Mon Sep 17 00:00:00 2001 From: "Xiang (Sean) Zhou" Date: Wed, 23 Jul 2025 10:44:23 -0700 Subject: [PATCH] chore: Fixed flaky test_update_credential_with_tokens unittest PiperOrigin-RevId: 786340983 --- tests/unittests/auth/test_oauth2_credential_util.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/unittests/auth/test_oauth2_credential_util.py b/tests/unittests/auth/test_oauth2_credential_util.py index aba6a992..f1fd607f 100644 --- a/tests/unittests/auth/test_oauth2_credential_util.py +++ b/tests/unittests/auth/test_oauth2_credential_util.py @@ -132,10 +132,12 @@ class TestOAuth2CredentialUtil: ), ) + # Store the expected expiry time to avoid timing issues + expected_expires_at = int(time.time()) + 3600 tokens = OAuth2Token({ "access_token": "new_access_token", "refresh_token": "new_refresh_token", - "expires_at": int(time.time()) + 3600, + "expires_at": expected_expires_at, "expires_in": 3600, }) @@ -143,5 +145,5 @@ class TestOAuth2CredentialUtil: assert credential.oauth2.access_token == "new_access_token" assert credential.oauth2.refresh_token == "new_refresh_token" - assert credential.oauth2.expires_at == int(time.time()) + 3600 + assert credential.oauth2.expires_at == expected_expires_at assert credential.oauth2.expires_in == 3600