Function TestAuditTool_MultiRunDiffMode

pkg/cli/mcp_tools_privileged_test.go:539–573

TestAuditTool_MultiRunDiffMode verifies that when run_ids_or_urls contains multiple entries the audit tool passes all of them as positional arguments to the audit command (which then runs in diff mode).

(t *testing.T)

// multiple entries the audit tool passes all of them as positional arguments
// to the audit command (which then runs in diff mode).
func TestAuditTool_MultiRunDiffMode(t *testing.T) {
	const expectedStdout = `[{"base_run_id":111,"compare_run_id":222}]`

	var capturedArgs []string
	mockExecCmd := func(ctx context.Context, args ...string) *exec.Cmd {
		capturedArgs = slices.Clone(args)
		return exec.CommandContext(ctx, "sh", "-c", `printf '%s' "$1"`, "sh", expectedStdout)
	}

	server := mcp.NewServer(&mcp.Implementation{Name: "test", Version: "1.0"}, nil)
	err := registerAuditTool(server, mockExecCmd, "", false)
	require.NoError(t, err, "registerAuditTool should succeed")

	session := connectInMemory(t, server)
	result, err := session.CallTool(context.Background(), &mcp.CallToolParams{
		Name: "audit",
		Arguments: map[string]any{
			"run_ids_or_urls": []string{"111", "222", "333"},
		},
	})
	require.NoError(t, err, "audit tool should succeed with multiple run IDs")
	require.NotNil(t, result, "result should not be nil")
	assert.False(t, result.IsError, "result should not be an error")

	textContent, ok := result.Content[0].(*mcp.TextContent)
	require.True(t, ok, "expected text content in audit response")
	assert.JSONEq(t, expectedStdout, textContent.Text, "audit tool should return subprocess stdout")

	// All three run IDs must appear as positional args immediately after "audit"
	require.GreaterOrEqual(t, len(capturedArgs), 4, "captured args should include audit + 3 run IDs: %v", capturedArgs)
	assert.Equal(t, "audit", capturedArgs[0], "first arg should be 'audit'")
	assert.Equal(t, "111", capturedArgs[1], "second arg should be first run ID")
	assert.Equal(t, "222", capturedArgs[2], "third arg should be second run ID")
	assert.Equal(t, "333", capturedArgs[3], "fourth arg should be third run ID")
}

// TestAuditTool_FailsWhenNoRunIDProvided verifies that the audit tool
// returns an error when neither run_id_or_url nor run_ids_or_urls is provided.