From 1dca8c7f25bcbf97d7f3e263deef64e8a736a0bd Mon Sep 17 00:00:00 2001 From: kami Date: Fri, 17 Jul 2026 12:15:33 +0400 Subject: [PATCH] test(workflow): update shipped-workflow tool-grant expectations for codebase-memory MCP tools Follow-up to 3e31ebcc, which granted the MCP read-only query tools to review_loop and role_pipeline stages but left the exact-set assertions stale. Co-Authored-By: Claude Fable 5 --- .../workflow/ReviewLoopWorkflowTest.kt | 13 +++++++++++-- .../workflow/RolePipelineWorkflowTest.kt | 7 ++++++- 2 files changed, 17 insertions(+), 3 deletions(-) diff --git a/infrastructure/workflow/src/test/kotlin/com/correx/infrastructure/workflow/ReviewLoopWorkflowTest.kt b/infrastructure/workflow/src/test/kotlin/com/correx/infrastructure/workflow/ReviewLoopWorkflowTest.kt index b2d4c577..81d97ba3 100644 --- a/infrastructure/workflow/src/test/kotlin/com/correx/infrastructure/workflow/ReviewLoopWorkflowTest.kt +++ b/infrastructure/workflow/src/test/kotlin/com/correx/infrastructure/workflow/ReviewLoopWorkflowTest.kt @@ -115,12 +115,21 @@ class ReviewLoopWorkflowTest { // implement claims/submits and may create a task, alongside its file write. assertEquals( - setOf("file_write", "task_create", "task_update", "task_context", "task_search"), + setOf( + "file_write", "task_create", "task_update", "task_context", "task_search", + "mcp__codebase-memory__search_code", + "mcp__codebase-memory__get_code_snippet", + "mcp__codebase-memory__trace_path", + ), graph.stages[StageId("implement")]!!.allowedTools, ) // review reads the task and completes it on an approved verdict. assertEquals( - setOf("task_context", "task_update"), + setOf( + "task_context", "task_update", + "mcp__codebase-memory__search_code", + "mcp__codebase-memory__trace_path", + ), graph.stages[StageId("review")]!!.allowedTools, ) } diff --git a/infrastructure/workflow/src/test/kotlin/com/correx/infrastructure/workflow/RolePipelineWorkflowTest.kt b/infrastructure/workflow/src/test/kotlin/com/correx/infrastructure/workflow/RolePipelineWorkflowTest.kt index 684f4744..ab048c13 100644 --- a/infrastructure/workflow/src/test/kotlin/com/correx/infrastructure/workflow/RolePipelineWorkflowTest.kt +++ b/infrastructure/workflow/src/test/kotlin/com/correx/infrastructure/workflow/RolePipelineWorkflowTest.kt @@ -114,7 +114,12 @@ class RolePipelineWorkflowTest { assertTrue(implTools.containsAll(setOf("file_write", "file_edit", "task_update"))) // reviewer reads the task and completes it on an approved verdict. assertEquals( - setOf("task_context", "task_update"), + setOf( + "task_context", "task_update", + "mcp__codebase-memory__search_code", + "mcp__codebase-memory__trace_path", + "mcp__codebase-memory__get_code_snippet", + ), graph.stages[StageId("reviewer")]!!.allowedTools, ) }