1
0
Fork 0

i think that this works now

This commit is contained in:
Arthur K. 2026-03-08 09:44:33 +03:00
parent a3c843d63c
commit fc555244a8
Signed by: wzray
GPG key ID: B97F30FDC4636357
13 changed files with 715 additions and 505 deletions

View file

@ -1,72 +1,23 @@
import json
from tokens import (
ProviderTokens,
load_state,
save_state,
save_tokens,
promote_next_tokens,
clear_next_tokens,
)
import tokens as t
from tokens import ProviderTokens
def test_save_and_load_state(tmp_path, monkeypatch):
monkeypatch.setattr(t, "TOKENS_FILE", tmp_path / "tokens.json")
active = ProviderTokens("key1", None, 0)
nxt = ProviderTokens("key2", None, 0)
save_state(active, nxt)
a, n = load_state()
assert a is not None and a.access_token == "key1"
assert n is not None and n.access_token == "key2"
def test_provider_tokens_basic():
t = ProviderTokens(access_token="abc123", refresh_token=None, expires_at=0)
assert t.access_token == "abc123"
assert t.refresh_token is None
assert t.expires_at == 0
def test_promote_next_tokens(tmp_path, monkeypatch):
monkeypatch.setattr(t, "TOKENS_FILE", tmp_path / "tokens.json")
save_state(ProviderTokens("key1", None, 0), ProviderTokens("key2", None, 0))
assert promote_next_tokens() is True
a, n = load_state()
assert a is not None and a.access_token == "key2"
assert n is None
def test_provider_tokens_with_metadata():
t = ProviderTokens(
access_token="key",
refresh_token="refresh",
expires_at=12345.0,
metadata={"foo": "bar"},
)
assert t.metadata == {"foo": "bar"}
def test_clear_next_tokens(tmp_path, monkeypatch):
monkeypatch.setattr(t, "TOKENS_FILE", tmp_path / "tokens.json")
save_state(ProviderTokens("key1", None, 0), ProviderTokens("key2", None, 0))
clear_next_tokens()
a, n = load_state()
assert a is not None and a.access_token == "key1"
assert n is None
def test_save_tokens_preserves_next(tmp_path, monkeypatch):
monkeypatch.setattr(t, "TOKENS_FILE", tmp_path / "tokens.json")
save_state(ProviderTokens("key1", None, 0), ProviderTokens("key2", None, 0))
save_tokens(ProviderTokens("key3", None, 0))
a, n = load_state()
assert a is not None and a.access_token == "key3"
assert n is not None and n.access_token == "key2"
def test_load_missing_file(tmp_path, monkeypatch):
monkeypatch.setattr(t, "TOKENS_FILE", tmp_path / "missing.json")
a, n = load_state()
assert a is None and n is None
def test_atomic_write(tmp_path, monkeypatch):
f = tmp_path / "tokens.json"
monkeypatch.setattr(t, "TOKENS_FILE", f)
save_state(ProviderTokens("x", None, 0), None)
with open(f) as fp:
data = json.load(fp)
assert data["active"]["access_token"] == "x"
def test_provider_tokens_default_metadata():
t = ProviderTokens(access_token="x", refresh_token=None, expires_at=0)
assert t.metadata is None