Function TestMCPScriptsTimeoutInJSON

pkg/workflow/mcp_scripts_timeout_test.go:235–296

TestMCPScriptsTimeoutInJSON tests that timeout is included in the generated tools.json

(t *testing.T)

// TestMCPScriptsTimeoutInJSON tests that timeout is included in the generated tools.json
func TestMCPScriptsTimeoutInJSON(t *testing.T) {
	config := &MCPScriptsConfig{
		Tools: map[string]*MCPScriptToolConfig{
			"fast-tool": {
				Name:        "fast-tool",
				Description: "Fast tool",
				Script:      "return 'fast';",
				Timeout:     30,
			},
			"slow-tool": {
				Name:        "slow-tool",
				Description: "Slow tool",
				Run:         "echo 'slow'",
				Timeout:     120,
			},
			"default-tool": {
				Name:        "default-tool",
				Description: "Default timeout tool",
				Py:          "print('default')",
				Timeout:     60,
			},
			"go-tool": {
				Name:        "go-tool",
				Description: "Go timeout tool",
				Go:          "fmt.Println(\"hello\")",
				Timeout:     180,
			},
		},
	}

	jsonStr := GenerateMCPScriptsToolsConfig(config)

	// Parse the JSON to verify structure
	var parsedConfig MCPScriptsConfigJSON
	if err := json.Unmarshal([]byte(jsonStr), &parsedConfig); err != nil {
		t.Fatalf("Failed to parse generated JSON: %v", err)
	}

	// Verify timeouts are present
	toolTimeouts := make(map[string]int)
	for _, tool := range parsedConfig.Tools {
		toolTimeouts[tool.Name] = tool.Timeout
	}

	expected := map[string]int{
		"fast-tool":    30,
		"slow-tool":    120,
		"default-tool": 60,
		"go-tool":      180,
	}

	for toolName, expectedTimeout := range expected {
		actualTimeout, exists := toolTimeouts[toolName]
		if !exists {
			t.Errorf("Tool %s not found in generated JSON", toolName)
			continue
		}
		if actualTimeout != expectedTimeout {