akadmin/headroom
1
0
Fork 0
mirror of https://github.com/headroomlabs-ai/headroom.git synced 2026-08-27 14:17:10 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

style: apply ruff format to copilot_quota and test_copilot_quota

Browse source 
Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
This commit is contained in:
JerrettDavis 2026-04-12 00:08:56 -05:00
parent d11feea511
commit a9f53cc104
2 changed files with 34 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

4
headroom/subscription/copilot_quota.py
View file

@ -276,7 +276,9 @@ class _CopilotQuotaTracker:
try:
async with aiohttp.ClientSession() as session:
async with session.get(url, headers=headers, timeout=aiohttp.ClientTimeout(total=10)) as resp:
async with session.get(
url, headers=headers, timeout=aiohttp.ClientTimeout(total=10)
) as resp:
if resp.status == 401:
with self._lock:
self._state.last_error = "unauthorized — check GITHUB_TOKEN"

41
tests/test_copilot_quota.py
View file

@ -150,11 +150,7 @@ class TestParseCopilotQuota:
assert snap.categories["chat"].remaining == 75
def test_unlimited_category(self):
data = {
"quota_snapshots": {
"completions": {"unlimited": True}
}
}
data = {"quota_snapshots": {"completions": {"unlimited": True}}}
snap = parse_copilot_quota(data)
assert snap.categories["completions"].unlimited is True
@ -207,31 +203,56 @@ class TestParseCopilotQuota:
class TestDiscoverGithubToken:
def test_returns_none_when_no_env_vars(self, monkeypatch):
for var in ["GITHUB_COPILOT_GITHUB_TOKEN", "GITHUB_TOKEN", "COPILOT_GITHUB_TOKEN", "GITHUB_COPILOT_API_TOKEN"]:
for var in [
"GITHUB_COPILOT_GITHUB_TOKEN",
"GITHUB_TOKEN",
"COPILOT_GITHUB_TOKEN",
"GITHUB_COPILOT_API_TOKEN",
]:
monkeypatch.delenv(var, raising=False)
assert discover_github_token() is None
def test_picks_up_github_token(self, monkeypatch):
for var in ["GITHUB_COPILOT_GITHUB_TOKEN", "GITHUB_TOKEN", "COPILOT_GITHUB_TOKEN", "GITHUB_COPILOT_API_TOKEN"]:
for var in [
"GITHUB_COPILOT_GITHUB_TOKEN",
"GITHUB_TOKEN",
"COPILOT_GITHUB_TOKEN",
"GITHUB_COPILOT_API_TOKEN",
]:
monkeypatch.delenv(var, raising=False)
monkeypatch.setenv("GITHUB_TOKEN", "ghp_testtoken123")
assert discover_github_token() == "ghp_testtoken123"
def test_prefers_copilot_specific_token(self, monkeypatch):
for var in ["GITHUB_COPILOT_GITHUB_TOKEN", "GITHUB_TOKEN", "COPILOT_GITHUB_TOKEN", "GITHUB_COPILOT_API_TOKEN"]:
for var in [
"GITHUB_COPILOT_GITHUB_TOKEN",
"GITHUB_TOKEN",
"COPILOT_GITHUB_TOKEN",
"GITHUB_COPILOT_API_TOKEN",
]:
monkeypatch.delenv(var, raising=False)
monkeypatch.setenv("GITHUB_COPILOT_GITHUB_TOKEN", "ghp_copilot_specific")
monkeypatch.setenv("GITHUB_TOKEN", "ghp_generic")
assert discover_github_token() == "ghp_copilot_specific"
def test_falls_through_to_next_env_var(self, monkeypatch):
for var in ["GITHUB_COPILOT_GITHUB_TOKEN", "GITHUB_TOKEN", "COPILOT_GITHUB_TOKEN", "GITHUB_COPILOT_API_TOKEN"]:
for var in [
"GITHUB_COPILOT_GITHUB_TOKEN",
"GITHUB_TOKEN",
"COPILOT_GITHUB_TOKEN",
"GITHUB_COPILOT_API_TOKEN",
]:
monkeypatch.delenv(var, raising=False)
monkeypatch.setenv("COPILOT_GITHUB_TOKEN", "ghp_copilot")
assert discover_github_token() == "ghp_copilot"
def test_ignores_empty_strings(self, monkeypatch):
for var in ["GITHUB_COPILOT_GITHUB_TOKEN", "GITHUB_TOKEN", "COPILOT_GITHUB_TOKEN", "GITHUB_COPILOT_API_TOKEN"]:
for var in [
"GITHUB_COPILOT_GITHUB_TOKEN",
"GITHUB_TOKEN",
"COPILOT_GITHUB_TOKEN",
"GITHUB_COPILOT_API_TOKEN",
]:
monkeypatch.delenv(var, raising=False)
monkeypatch.setenv("GITHUB_COPILOT_GITHUB_TOKEN", "")
monkeypatch.setenv("GITHUB_TOKEN", "ghp_valid")