fix: test

Author: stillmatic

integration/test_auth.py

@@ -164,9 +164,9 @@ def test_session_cookies_with_tolerance(api_key):
     dev_claims = {'premium' : True, 'subscription' : 'silver'}
     custom_token = auth.create_custom_token('user3', dev_claims)
     id_token = _sign_in(custom_token, api_key)
-    expires_in = datetime.timedelta(seconds=1)
+    expires_in = datetime.timedelta(seconds=300)
     session_cookie = auth.create_session_cookie(id_token, expires_in=expires_in)
-    time.sleep(2)
+    time.sleep(300)
     # expect this to fail because the cookie is expired
     with pytest.raises(auth.ExpiredSessionCookieError):
         auth.verify_session_cookie(session_cookie)
@@ -596,20 +596,17 @@ def test_verify_id_token_revoked(new_user, api_key):
     assert claims['iat'] * 1000 >= user.tokens_valid_after_timestamp
 
 def test_verify_id_token_tolerance(new_user, api_key):
-    # create token that expires in 1 second.
-    custom_token = auth.create_custom_token(new_user.uid, expires_in=datetime.timedelta(seconds=1))
-    id_token = _sign_in(custom_token, api_key)
+    expired_id_token = _sign_in_with_password(new_user.email, 'password', api_key)
     time.sleep(1)
+    # Verify the ID token with a tolerance of 0 seconds. This should
+    # raise an exception because the token is expired.
+    with pytest.raises(auth.InvalidIdTokenError) as excinfo:
+        auth.verify_id_token(expired_id_token, check_revoked=False, clock_skew_in_seconds=0)
+    assert str(excinfo.value) == 'The Firebase ID token is expired.'
 
-    # verify with tolerance of 0 seconds. This should fail.
-    with pytest.raises(auth.ExpiredIdTokenError) as excinfo:
-        auth.verify_id_token(id_token, check_revoked=False, max_age=datetime.timedelta(seconds=0))
-    assert str(excinfo.value) == 'The Firebase ID token has expired.'
-
-    # verify with tolerance of 2 seconds. This should succeed.
-    claims = auth.verify_id_token(id_token, check_revoked=False, max_age=datetime.timedelta(seconds=2))
-    assert claims['sub'] == new_user.uid
-
+    # Verify the ID token with a tolerance of 2 seconds. This should
+    # not raise an exception because the token is within the tolerance.
+    auth.verify_id_token(expired_id_token, check_revoked=False, clock_skew_in_seconds=2)
 
 def test_verify_id_token_disabled(new_user, api_key):
     custom_token = auth.create_custom_token(new_user.uid)
@@ -652,8 +649,8 @@ def test_verify_session_cookie_revoked(new_user, api_key):
     assert claims['iat'] * 1000 >= user.tokens_valid_after_timestamp
 
 def test_verify_session_cookie_tolerance(new_user, api_key):
-    expired_session_cookie = auth.create_session_cookie(_sign_in(auth.create_custom_token(new_user.uid), api_key), expires_in=datetime.timedelta(seconds=1))
-    time.sleep(1)
+    expired_session_cookie = auth.create_session_cookie(_sign_in(auth.create_custom_token(new_user.uid), api_key), expires_in=datetime.timedelta(seconds=300))
+    time.sleep(300)
     # Verify the session cookie with a tolerance of 0 seconds. This should
     # raise an exception because the cookie is expired.
     with pytest.raises(auth.InvalidSessionCookieError) as excinfo: