chore: handle noisy log messages
This commit is contained in:
parent
858d127246
commit
2611d1bb6d
4 changed files with 71 additions and 27 deletions
|
|
@ -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")))
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue