fix: show request timing and nested usage

This commit is contained in:
2026-06-30 17:00:33 +08:00
parent 979dd81c8f
commit 9ce90ff01e
2 changed files with 59 additions and 19 deletions
+28
View File
@@ -109,6 +109,34 @@ class CodexRetryGatewayTUITests(unittest.TestCase):
}
self.assertEqual(mod.request_usage_summary(row), "in 75 | out 40 | cached 25 (25%)")
def test_request_rows_read_usage_from_nested_usage_object(self) -> None:
mod = load_module()
payload = {
"entries": [
{
"seq": 1,
"request_id": "req_nested",
"path": "/responses",
"usage": {
"input_tokens": 120,
"output_tokens": 33,
"total_tokens": 153,
"cached_tokens": 20,
},
}
]
}
rows = mod.normalize_request_rows(payload)
self.assertEqual(rows[0]["input_tokens"], 120)
self.assertEqual(rows[0]["output_tokens"], 33)
self.assertEqual(rows[0]["total_tokens"], 153)
self.assertEqual(rows[0]["cached_tokens"], 20)
def test_format_duration_ms_as_seconds(self) -> None:
mod = load_module()
self.assertEqual(mod.format_duration_ms_as_seconds(1532), "1.5s")
self.assertEqual(mod.format_duration_ms_as_seconds(120000), "120s")
def test_profile_editor_payload_round_trip(self) -> None:
mod = load_module()
row = {