1111if str (REPO_ROOT ) not in sys .path :
1212 sys .path .insert (0 , str (REPO_ROOT ))
1313
14- from loopx .control_plane .todos .deferred_resume import ( # noqa: E402
15- TODO_DEFERRED_RESUME_SELECTION_POLICY ,
16- TODO_MONITOR_BLOCKED_RESUME_SELECTION_POLICY ,
17- build_todo_deferred_visibility_lanes ,
18- build_todo_resume_blocked_visibility_lanes ,
19- todo_summary_monitor_blocked_resume_items ,
20- todo_summary_resume_blocked_items ,
21- )
14+ from loopx .control_plane .todos .resume_planning import project_todo_resume_planning # noqa: E402
2215
2316
2417CURRENT_AGENT = "codex-product-capability"
@@ -116,11 +109,11 @@ def assert_deferred_resume_lanes_filter_current_unclaimed_and_other_agents() ->
116109 ],
117110 }
118111
119- lanes = build_todo_deferred_visibility_lanes (
112+ lanes = project_todo_resume_planning (
120113 summary ,
121- agent_identity = { " agent_id" : CURRENT_AGENT } ,
114+ agent_id = CURRENT_AGENT ,
122115 item_limit = 10 ,
123- )
116+ )[ "deferred_lanes" ]
124117 assert lanes ["deferred_count" ] == 1 , lanes
125118 assert lanes ["deferred_visibility_limit" ] == 10 , lanes
126119 assert lanes ["deferred_items" ][0 ]["todo_id" ] == "todo_deferred_backlog" , lanes
@@ -140,9 +133,6 @@ def assert_deferred_resume_lanes_filter_current_unclaimed_and_other_agents() ->
140133 current = lanes ["current_agent_deferred_resume_candidates" ][0 ]
141134 assert current ["required_write_scopes" ] == ["loopx/**" ], current
142135 assert current ["decision_scope" ]["scope_key" ] == "resume" , current
143- assert lanes ["deferred_resume_selection_policy" ] == (
144- TODO_DEFERRED_RESUME_SELECTION_POLICY
145- ), lanes
146136
147137
148138def assert_monitor_blocked_resume_lanes_filter_by_claim_and_monitor_target () -> None :
@@ -181,15 +171,17 @@ def assert_monitor_blocked_resume_lanes_filter_by_claim_and_monitor_target() ->
181171 "backlog_items" : [current ],
182172 }
183173
184- resume_blocked = todo_summary_resume_blocked_items (summary )
174+ projection = project_todo_resume_planning (summary , agent_id = CURRENT_AGENT , item_limit = 10 )
175+ lanes = projection ["resume_blocked_lanes" ]
176+ resume_blocked = lanes ["resume_blocked_items" ]
185177 assert [item ["todo_id" ] for item in resume_blocked ] == [
186178 "todo_current_blocked" ,
187179 "todo_unclaimed_blocked" ,
188180 "todo_other_blocked" ,
189181 "todo_non_monitor_blocked" ,
190182 "todo_excluded_blocked" ,
191183 ], resume_blocked
192- monitor_blocked = todo_summary_monitor_blocked_resume_items ( summary )
184+ monitor_blocked = projection [ "monitor_blocked_items" ]
193185 assert [item ["todo_id" ] for item in monitor_blocked ] == [
194186 "todo_current_blocked" ,
195187 "todo_unclaimed_blocked" ,
@@ -201,20 +193,12 @@ def assert_monitor_blocked_resume_lanes_filter_by_claim_and_monitor_target() ->
201193 for item in monitor_blocked
202194 ), monitor_blocked
203195
204- lanes = build_todo_resume_blocked_visibility_lanes (
205- summary ,
206- agent_identity = {"agent_id" : CURRENT_AGENT },
207- item_limit = 10 ,
208- )
209196 assert lanes ["resume_blocked_count" ] == 5 , lanes
210197 assert lanes ["monitor_blocked_resume_count" ] == 4 , lanes
211198 assert lanes ["current_agent_monitor_blocked_resume_count" ] == 1 , lanes
212199 assert lanes ["unclaimed_monitor_blocked_resume_count" ] == 1 , lanes
213200 assert lanes ["other_agent_monitor_blocked_resume_count" ] == 1 , lanes
214201 assert lanes ["executor_excluded_self_monitor_blocked_resume_count" ] == 1 , lanes
215- assert lanes ["monitor_blocked_resume_selection_policy" ] == (
216- TODO_MONITOR_BLOCKED_RESUME_SELECTION_POLICY
217- ), lanes
218202
219203
220204def main () -> int :
0 commit comments