Function TestAdditionalClaudeToolsForSafeOutputs

pkg/workflow/git_commands_test.go:134–250

(t *testing.T)

}

func TestAdditionalClaudeToolsForSafeOutputs(t *testing.T) {
	compiler := NewCompiler()
	engine := NewClaudeEngine()

	tests := []struct {
		name               string
		tools              map[string]any
		safeOutputs        *SafeOutputsConfig
		expectEditingTools bool
	}{
		{
			name:               "no safe outputs - no editing tools",
			tools:              map[string]any{},
			safeOutputs:        nil,
			expectEditingTools: false,
		},
		{
			name:  "create-pull-request enabled - should add editing tools",
			tools: map[string]any{},
			safeOutputs: &SafeOutputsConfig{
				CreatePullRequests: &CreatePullRequestsConfig{},
			},
			expectEditingTools: true,
		},
		{
			name:  "push-to-pull-request-branch enabled - should add editing tools",
			tools: map[string]any{},
			safeOutputs: &SafeOutputsConfig{
				PushToPullRequestBranch: &PushToPullRequestBranchConfig{},
			},
			expectEditingTools: true,
		},
		{
			name:  "only create-issue enabled - no editing tools",
			tools: map[string]any{},
			safeOutputs: &SafeOutputsConfig{
				CreateIssues: &CreateIssuesConfig{},
			},
			expectEditingTools: false,
		},
		{
			name: "existing editing tools should be preserved",
			tools: map[string]any{
				"edit": nil,
			},
			safeOutputs: &SafeOutputsConfig{
				CreatePullRequests: &CreatePullRequestsConfig{},
			},
			expectEditingTools: true,
		},
	}

	expectedEditingTools := []string{"Edit", "MultiEdit", "NotebookEdit"}
	expectedWriteTool := "Write"

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			// Create a copy of input tools to avoid modifying test data