feat: distinguish full-day traffic rollups

This commit is contained in:
2026-08-04 03:26:59 +08:00
parent 1db6b20dcc
commit 05b17face8
5 changed files with 31 additions and 14 deletions
+8
View File
@@ -89,6 +89,7 @@ def workspace_payload_fixture() -> dict[str, object]:
"ends_at": local_end.isoformat(),
"sample_limit": 1000,
"sample_limited": False,
"usage_rollups_full_day": True,
"partial": False,
"instances": [
{"instance": "server6", "ok": True, "error_total": 4},
@@ -810,6 +811,7 @@ class WorkspaceTests(unittest.TestCase):
self.assertEqual(keys["period_kind"], "calendar_day")
self.assertEqual(keys["date"], payload["traffic"]["date"])
self.assertEqual(keys["timezone"], "Asia/Shanghai")
self.assertTrue(keys["usage_rollups_full_day"])
self.assertIn(f"today {keys['date']} (Asia/Shanghai)", mod.usage_period_label(keys))
self.assertEqual(errors["time_range"], f"today {keys['date']} (Asia/Shanghai)")
self.assertEqual({item["_node"] for item in errors["items"]}, {"server6", "server4"})
@@ -876,6 +878,12 @@ class WorkspaceTests(unittest.TestCase):
with self.assertRaisesRegex(RuntimeError, "invalid workspace traffic period"):
mod.fetch_workspace_payload("https://workspace.example.test/data", 3)
incomplete_usage = workspace_payload_fixture()
incomplete_usage["traffic"]["usage_rollups_full_day"] = False
with mock.patch.object(mod, "fetch_payload", return_value=incomplete_usage):
with self.assertRaisesRegex(RuntimeError, "incomplete workspace traffic usage"):
mod.fetch_workspace_payload("https://workspace.example.test/data", 3)
noncanonical_request = workspace_payload_fixture()
noncanonical_request["traffic"]["requests"][0]["instance"] = "server4"
with mock.patch.object(mod, "fetch_payload", return_value=noncanonical_request):