1
0
Fork 0

chore: handle noisy log messages

This commit is contained in:
Arthur K. 2026-03-02 20:15:33 +03:00
parent 858d127246
commit 2611d1bb6d
Signed by: wzray
GPG key ID: B97F30FDC4636357
4 changed files with 71 additions and 27 deletions

View file

@ -26,9 +26,17 @@ class FakeProvider(Provider):
"secondary_window": None,
}
self._rotate = rotate
self._prepare_threshold = 80
self.get_token_calls = 0
self.standby_calls = 0
@property
def prepare_threshold(self) -> int:
return self._prepare_threshold
def should_prepare_standby(self, usage_percent: int) -> bool:
return usage_percent >= self.prepare_threshold
@property
def name(self) -> str:
return "fake"
@ -55,9 +63,10 @@ class FakeProvider(Provider):
return self._rotate
async def ensure_standby_account(
self, usage_percent: int, prepare_threshold: int
self,
usage_percent: int,
) -> None:
_ = usage_percent, prepare_threshold
_ = usage_percent
self.standby_calls += 1
@ -91,7 +100,10 @@ def test_build_limit_fields():
def test_get_prepare_threshold():
assert server.get_prepare_threshold("chatgpt") == server.CHATGPT_PREPARE_THRESHOLD
assert (
server.get_prepare_threshold("chatgpt")
== server.PROVIDERS["chatgpt"].prepare_threshold
)
assert server.get_prepare_threshold("unknown") == 100
@ -105,8 +117,6 @@ def test_token_handler_success(monkeypatch):
provider = FakeProvider()
monkeypatch.setattr(server, "PROVIDERS", {"fake": provider})
monkeypatch.setattr(server, "background_tasks", {"fake": None})
monkeypatch.setattr(server, "get_prepare_threshold", lambda _: 80)
resp = asyncio.run(server.token_handler(FakeRequest("fake")))
data = _response_json(resp)
@ -125,10 +135,9 @@ def test_token_handler_triggers_standby(monkeypatch):
def fake_trigger(name, usage_percent, reason):
assert name == "fake"
assert usage_percent == 90
assert "threshold" in reason
assert "standby policy" in reason
called["value"] = True
monkeypatch.setattr(server, "get_prepare_threshold", lambda _: 80)
monkeypatch.setattr(server, "trigger_standby_prepare", fake_trigger)
resp = asyncio.run(server.token_handler(FakeRequest("fake")))
@ -143,7 +152,6 @@ def test_token_handler_rotation_path(monkeypatch):
)
monkeypatch.setattr(server, "PROVIDERS", {"fake": provider})
monkeypatch.setattr(server, "background_tasks", {"fake": None})
monkeypatch.setattr(server, "get_prepare_threshold", lambda _: 80)
monkeypatch.setattr(server, "trigger_standby_prepare", lambda *_: None)
resp = asyncio.run(server.token_handler(FakeRequest("fake")))