58 lines
1.5 KiB
Python
58 lines
1.5 KiB
Python
from __future__ import annotations
|
|
|
|
import base64
|
|
import json
|
|
import time
|
|
|
|
from hermes_cli import auth
|
|
|
|
|
|
def _jwt_with_exp(exp: int) -> str:
|
|
header = (
|
|
base64.urlsafe_b64encode(json.dumps({"alg": "none"}).encode())
|
|
.decode()
|
|
.rstrip("=")
|
|
)
|
|
payload = (
|
|
base64.urlsafe_b64encode(json.dumps({"exp": exp}).encode())
|
|
.decode()
|
|
.rstrip("=")
|
|
)
|
|
return f"{header}.{payload}.sig"
|
|
|
|
|
|
def test_xai_oauth_refresh_skew_is_one_hour() -> None:
|
|
assert auth.XAI_ACCESS_TOKEN_REFRESH_SKEW_SECONDS == 3600
|
|
|
|
|
|
def test_xai_oauth_token_expiring_uses_one_hour_skew() -> None:
|
|
token = _jwt_with_exp(int(time.time()) + 30 * 60)
|
|
|
|
assert auth._xai_access_token_is_expiring(
|
|
token,
|
|
auth.XAI_ACCESS_TOKEN_REFRESH_SKEW_SECONDS,
|
|
)
|
|
|
|
|
|
def test_xai_oauth_token_not_expiring_beyond_one_hour_skew() -> None:
|
|
token = _jwt_with_exp(int(time.time()) + 90 * 60)
|
|
|
|
assert not auth._xai_access_token_is_expiring(
|
|
token,
|
|
auth.XAI_ACCESS_TOKEN_REFRESH_SKEW_SECONDS,
|
|
)
|
|
|
|
|
|
def test_xai_proactive_refresh_skew_short_lived_token() -> None:
|
|
token = _jwt_with_exp(int(time.time()) + 15 * 60)
|
|
skew = auth._xai_proactive_refresh_skew_seconds(token)
|
|
|
|
assert skew == 120
|
|
assert not auth._xai_access_token_is_expiring(token, skew)
|
|
|
|
|
|
def test_xai_proactive_refresh_skew_long_lived_token() -> None:
|
|
token = _jwt_with_exp(int(time.time()) + 5 * 60 * 60)
|
|
|
|
assert auth._xai_proactive_refresh_skew_seconds(token) == auth.XAI_ACCESS_TOKEN_REFRESH_SKEW_SECONDS
|