diff --git a/tests/unit/test_api_server.py b/tests/unit/test_api_server.py index 2be26f7..43879c6 100644 --- a/tests/unit/test_api_server.py +++ b/tests/unit/test_api_server.py @@ -180,6 +180,8 @@ class ApiServerRoutingTests(unittest.TestCase): "instrument_id": kwargs["instrument_id"], "limit": kwargs["limit"], "filters": {"preferred_history_kind": kwargs["preferred_history_kind"], "query_text": kwargs["query_text"]}, + "long_range_trend_summary": {"status": "active"}, + "baseline_pin_summary": {"status": "active"}, } }, ) @@ -193,6 +195,8 @@ class ApiServerRoutingTests(unittest.TestCase): "query_text": kwargs["query_text"], "left_id": kwargs["left_id"], "right_id": kwargs["right_id"], + "long_range_trend_summary": {"status": "active"}, + "baseline_pin_summary": {"status": "active"}, } }, ) @@ -216,6 +220,8 @@ class ApiServerRoutingTests(unittest.TestCase): "status": "active", "instrument_id": kwargs["instrument_id"], "filters": {"current_session": kwargs["current_session"], "next_phase": kwargs["next_phase"], "query_text": kwargs["query_text"]}, + "long_range_trend_summary": {"status": "active"}, + "baseline_pin_summary": {"status": "active"}, } }, ) @@ -229,6 +235,8 @@ class ApiServerRoutingTests(unittest.TestCase): "query_text": kwargs["query_text"], "left_id": kwargs["left_id"], "right_id": kwargs["right_id"], + "long_range_trend_summary": {"status": "active"}, + "baseline_pin_summary": {"status": "active"}, "handoff_bundle_summary": {"status": "active"}, } }, @@ -278,17 +286,21 @@ class ApiServerRoutingTests(unittest.TestCase): self.assertEqual(digest_archive_payload["limit"], 5) self.assertEqual(digest_archive_payload["filters"]["preferred_history_kind"], "reminder") self.assertEqual(digest_archive_payload["filters"]["query_text"], "candidate") + self.assertEqual(digest_archive_payload["long_range_trend_summary"]["status"], "active") self.assertEqual(digest_export_status, 200) self.assertEqual(digest_export_payload["export_kind"], "digest_archive_bundle") self.assertEqual(digest_export_payload["left_id"], 4) + self.assertEqual(digest_export_payload["baseline_pin_summary"]["status"], "active") self.assertEqual(digest_compare_status, 200) self.assertEqual(digest_compare_payload["left_id"], 4) self.assertEqual(digest_compare_payload["query_text"], "candidate") self.assertEqual(workflow_archive_status, 200) self.assertEqual(workflow_archive_payload["filters"]["next_phase"], "intraday") self.assertEqual(workflow_archive_payload["filters"]["query_text"], "handoff") + self.assertEqual(workflow_archive_payload["baseline_pin_summary"]["status"], "active") self.assertEqual(workflow_export_status, 200) self.assertEqual(workflow_export_payload["export_kind"], "workflow_archive_bundle") + self.assertEqual(workflow_export_payload["long_range_trend_summary"]["status"], "active") self.assertEqual(workflow_export_payload["handoff_bundle_summary"]["status"], "active") self.assertEqual(workflow_compare_status, 200) self.assertEqual(workflow_compare_payload["right_id"], 8)