Function TestTokenWeightsSingleQuoteEscapingInYAML

pkg/workflow/engine_test.go:518–555

(t *testing.T)

}

func TestTokenWeightsSingleQuoteEscapingInYAML(t *testing.T) {
	compiler := NewCompiler()
	registry := GetGlobalEngineRegistry()
	engine, err := registry.GetEngine("claude")
	if err != nil {
		t.Fatalf("Failed to get claude engine: %v", err)
	}

	// Model name containing a single quote — must not break YAML single-quoted scalar
	workflowData := &WorkflowData{
		Name: "Test Workflow",
		EngineConfig: &EngineConfig{
			ID: "claude",
			TokenWeights: &types.TokenWeights{
				Multipliers: map[string]float64{
					"bob's-model": 2.0, // Single quote in key
				},
			},
		},
	}

	var out strings.Builder
	compiler.generateCreateAwInfo(&out, workflowData, engine)
	output := out.String()

	// The generated YAML must not contain an un-escaped single quote inside a single-quoted value.
	// In YAML, a single quote inside a single-quoted scalar is represented as ”.
	if !strings.Contains(output, "GH_AW_INFO_TOKEN_WEIGHTS") {
		t.Errorf("Expected GH_AW_INFO_TOKEN_WEIGHTS in YAML output when multipliers are configured, got:\n%s", output)
	}
	if !strings.Contains(output, "bob''s-model") {
		t.Errorf("Expected single quote to be escaped as '' in YAML output, got:\n%s", output)
	}
	// There must be no dangling unescaped single quote inside the GH_AW_INFO_TOKEN_WEIGHTS value
	if strings.Contains(output, "bob's-model") {
		t.Errorf("Unescaped single quote found in YAML output:\n%s", output)
	}
}

func TestTokenWeightsEnvOmittedWhenOnlyTokenClassWeightsConfigured(t *testing.T) {
	compiler := NewCompiler()