Function TestCodexEngineWithToolsTimeout

pkg/workflow/tools_timeout_test.go:103–179

(t *testing.T)

}

func TestCodexEngineWithToolsTimeout(t *testing.T) {
	engine := NewCodexEngine()

	tests := []struct {
		name            string
		toolsTimeout    string
		expectedTimeout string
		expectedEnvVar  string
	}{
		{
			name:            "default timeout when not specified",
			toolsTimeout:    "",
			expectedTimeout: "tool_timeout_sec = 60", // 60 seconds default
			expectedEnvVar:  "",                      // GH_AW_TOOL_TIMEOUT not set when empty
		},
		{
			name:            "custom timeout of 30 seconds",
			toolsTimeout:    "30",
			expectedTimeout: "tool_timeout_sec = 30",
			expectedEnvVar:  "GH_AW_TOOL_TIMEOUT: 30", // env var in seconds
		},
		{
			name:            "custom timeout of 180 seconds",
			toolsTimeout:    "180",
			expectedTimeout: "tool_timeout_sec = 180",
			expectedEnvVar:  "GH_AW_TOOL_TIMEOUT: 180", // env var in seconds
		},
		{
			name:            "expression timeout uses default in TOML",
			toolsTimeout:    "${{ inputs.tool-timeout }}",
			expectedTimeout: "tool_timeout_sec = 60", // falls back to default in TOML
			expectedEnvVar:  "GH_AW_TOOL_TIMEOUT: ${{ inputs.tool-timeout }}",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			workflowData := &WorkflowData{
				ToolsTimeout: tt.toolsTimeout,
				Name:         "test-workflow",
				Tools: map[string]any{
					"github": map[string]any{},
				},
			}

			// Render MCP config
			var configBuilder strings.Builder
			mcpTools := []string{"github"}
			if err := engine.RenderMCPConfig(&configBuilder, workflowData.Tools, mcpTools, workflowData); err != nil {
				t.Fatalf("RenderMCPConfig returned unexpected error: %v", err)
			}
			configContent := configBuilder.String()

			if !strings.Contains(configContent, tt.expectedTimeout) {
				t.Errorf("Expected '%s' in MCP config, got: %s", tt.expectedTimeout, configContent)
			}

			// Check for GH_AW_TOOL_TIMEOUT in execution steps