feat: assert extracted tool calls

Author: 35C4n0r

lib/msgfmt/msgfmt_test.go

@@ -233,8 +233,16 @@ func TestFormatAgentMessage(t *testing.T) {
 					assert.NoError(t, err)
 					expected, err := testdataDir.ReadFile(path.Join(dir, string(agentType), c.Name(), "expected.txt"))
 					assert.NoError(t, err)
-					output, _ := FormatToolCall(agentType, FormatAgentMessage(agentType, string(msg), string(userInput)))
+					output, toolCalls := FormatToolCall(agentType, FormatAgentMessage(agentType, string(msg), string(userInput)))
 					assert.Equal(t, string(expected), output)
+
+					// Assert on the tool calls if there's an expected file
+					expectedToolCallsPath := path.Join(dir, string(agentType), c.Name(), "expected_tool_calls.txt")
+					if expectedToolCallsData, err := testdataDir.ReadFile(expectedToolCallsPath); err == nil {
+						expectedToolCalls := string(expectedToolCallsData)
+						actualToolCalls := strings.Join(toolCalls, "\n---\n")
+						assert.Equal(t, expectedToolCalls, actualToolCalls)
+					}
 				})
 			}
 		})

lib/msgfmt/testdata/format/claude/remove-task-tool-call/expected_tool_calls.txt

@@ -0,0 +1,30 @@
+● coder - coder_report_task (MCP)(summary: "Snake game created
+                                 successfully at snake-game.html",
+                                 link: "file:///home/coder/snake-ga
+                                 me.html", state: "working")
+  ⎿  {
+       "message": "Thanks for reporting!"
+     }
+---
+● coder - coder_report_task (MCP)(summary: "Snake game created
+                                 successfully at snake-game.html",
+                                 link: "file:///home/coder/snake-ga
+                                 me.html", state: "working")
+  ⎿  {
+       "message": "Thanks for reporting!"
+     }
+---
+● coder - coder_report_task (MCP)(summary: "Building a snake game
+                                 with HTML/CSS/JavaScript", link:
+                                 "", state: "working")
+  ⎿  {
+       "message": "Thanks for reporting!"
+     }
+---
+● coder - coder_report_task (MCP)(summary: "Snake game created
+                                 successfully at snake-game.html",
+                                 link: "file:///home/coder/snake-ga
+                                 me.html", state: "working")
+  ⎿  {
+       "message": "Thanks for reporting!"
+     }