mirror/nanobot
1
0
Fork 0
mirror of https://github.com/HKUDS/nanobot.git synced 2026-05-01 15:25:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(tests): update Tavily usage tests to match actual API response shape

Browse Source 
The _parse_tavily_usage implementation was updated to use the real
{account: {plan_usage, plan_limit, ...}} structure, but the tests
still used the old flat {used, limit, breakdown} format.

Made-with: Cursor
This commit is contained in:
Xubin Ren 2026-04-06 11:17:00 +00:00 committed by Xubin Ren
parent e528e6dd96
commit dad9c07843
1 changed files with 25 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
tests/utils/test_searchusage.py
View File

@ -79,11 +79,18 @@ class TestSearchUsageInfoFormat:
class TestParseTavilyUsage: class TestParseTavilyUsage:
def test_full_response(self): def test_full_response(self):
data = { data = {
"used": 142, "account": {
"limit": 1000, "current_plan": "Researcher",
"remaining": 858, "plan_usage": 142,
"reset_date": "2026-05-01", "plan_limit": 1000,
"breakdown": {"search": 120, "extract": 15, "crawl": 7}, "search_usage": 120,
"extract_usage": 15,
"crawl_usage": 7,
"map_usage": 0,
"research_usage": 0,
"paygo_usage": 0,
"paygo_limit": None,
},
} }
info = _parse_tavily_usage(data) info = _parse_tavily_usage(data)
assert info.provider == "tavily" assert info.provider == "tavily"
@ -91,26 +98,20 @@ class TestParseTavilyUsage:
assert info.used == 142 assert info.used == 142
assert info.limit == 1000 assert info.limit == 1000
assert info.remaining == 858 assert info.remaining == 858
assert info.reset_date == "2026-05-01"
assert info.search_used == 120 assert info.search_used == 120
assert info.extract_used == 15 assert info.extract_used == 15
assert info.crawl_used == 7 assert info.crawl_used == 7
def test_remaining_computed_when_missing(self): def test_remaining_computed(self):
data = {"used": 300, "limit": 1000} data = {"account": {"plan_usage": 300, "plan_limit": 1000}}
info = _parse_tavily_usage(data) info = _parse_tavily_usage(data)
assert info.remaining == 700 assert info.remaining == 700
def test_remaining_not_negative(self): def test_remaining_not_negative(self):
data = {"used": 1100, "limit": 1000} data = {"account": {"plan_usage": 1100, "plan_limit": 1000}}
info = _parse_tavily_usage(data) info = _parse_tavily_usage(data)
assert info.remaining == 0 assert info.remaining == 0
def test_camel_case_reset_date(self):
data = {"used": 10, "limit": 100, "resetDate": "2026-06-01"}
info = _parse_tavily_usage(data)
assert info.reset_date == "2026-06-01"
def test_empty_response(self): def test_empty_response(self):
info = _parse_tavily_usage({}) info = _parse_tavily_usage({})
assert info.provider == "tavily" assert info.provider == "tavily"
@ -118,8 +119,8 @@ class TestParseTavilyUsage:
assert info.used is None assert info.used is None
assert info.limit is None assert info.limit is None
def test_no_breakdown_key(self): def test_no_breakdown_fields(self):
data = {"used": 5, "limit": 50} data = {"account": {"plan_usage": 5, "plan_limit": 50}}
info = _parse_tavily_usage(data) info = _parse_tavily_usage(data)
assert info.search_used is None assert info.search_used is None
assert info.extract_used is None assert info.extract_used is None
@ -175,11 +176,14 @@ class TestFetchSearchUsageRouting:
mock_response = MagicMock() mock_response = MagicMock()
mock_response.status_code = 200 mock_response.status_code = 200
mock_response.json.return_value = { mock_response.json.return_value = {
"used": 142, "account": {
"limit": 1000, "current_plan": "Researcher",
"remaining": 858, "plan_usage": 142,
"reset_date": "2026-05-01", "plan_limit": 1000,
"breakdown": {"search": 120, "extract": 15, "crawl": 7}, "search_usage": 120,
"extract_usage": 15,
"crawl_usage": 7,
},
} }
mock_response.raise_for_status = MagicMock() mock_response.raise_for_status = MagicMock()
@ -197,7 +201,6 @@ class TestFetchSearchUsageRouting:
assert info.used == 142 assert info.used == 142
assert info.limit == 1000 assert info.limit == 1000
assert info.remaining == 858 assert info.remaining == 858
assert info.reset_date == "2026-05-01"
assert info.search_used == 120 assert info.search_used == 120
@pytest.mark.asyncio @pytest.mark.asyncio