feat: distinguish full-day traffic rollups
This commit is contained in:
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user