From 0bb01c2d6af09385d4adc702ecd75e0584d4a2f1 Mon Sep 17 00:00:00 2001 From: Hiten Shah Date: Mon, 18 May 2026 07:43:10 -0700 Subject: [PATCH] test: cover requested sources in fallback quick plans --- tests/test_planner_v3.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/tests/test_planner_v3.py b/tests/test_planner_v3.py index 20ff29f..77881bb 100644 --- a/tests/test_planner_v3.py +++ b/tests/test_planner_v3.py @@ -120,6 +120,17 @@ class PlannerV3Tests(unittest.TestCase): ) self.assertIn("digg", plan.subqueries[0].sources) + def test_quick_mode_preserves_explicit_requested_sources_in_fallback_plan(self): + plan = planner.plan_query( + topic="AI coding agents", + available_sources=["reddit", "youtube", "github"], + requested_sources=["reddit", "github"], + depth="quick", + provider=None, + model=None, + ) + self.assertIn("github", plan.subqueries[0].sources) + def test_default_comparison_uses_all_capable_sources(self): plan = planner.plan_query( topic="codex vs claude code",