Function TestModelEnvVarInjectionForAgentJob

pkg/workflow/model_env_vars_test.go:14–97

TestModelEnvVarInjectionForAgentJob tests that agent jobs get the correct model environment variable

(t *testing.T)

// TestModelEnvVarInjectionForAgentJob tests that agent jobs get the correct model environment variable
func TestModelEnvVarInjectionForAgentJob(t *testing.T) {
	tests := []struct {
		name                    string
		engine                  string
		expectedEnvVar          string
		expectedCommand         string
		expectedDefault         string
		expectedDefaultOverride string
	}{
		{
			name:                    "Claude agent uses GH_AW_MODEL_AGENT_CLAUDE",
			engine:                  "claude",
			expectedEnvVar:          constants.EnvVarModelAgentClaude,
			expectedCommand:         "${" + constants.EnvVarModelAgentClaude + ":+ --model",
			expectedDefault:         "", // Claude has no default model
			expectedDefaultOverride: compilerenv.DefaultModelClaude,
		},
		{
			name:                    "Codex agent uses GH_AW_MODEL_AGENT_CODEX",
			engine:                  "codex",
			expectedEnvVar:          constants.EnvVarModelAgentCodex,
			expectedCommand:         "${" + constants.EnvVarModelAgentCodex + `:+ --model "`,
			expectedDefault:         constants.CodexDefaultModel,
			expectedDefaultOverride: compilerenv.DefaultModelCodex,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			// Create a simple workflow with the specified engine
			// Add SafeOutputs to distinguish from detection jobs
			workflowData := &WorkflowData{
				Name: "test-workflow",
				AI:   tt.engine,
				Tools: map[string]any{
					"bash": []any{"echo"},
				},
				SafeOutputs: &SafeOutputsConfig{
					// Just enough to make it an agent job
				},
			}

			// Get the engine
			engine, err := GetGlobalEngineRegistry().GetEngine(tt.engine)
			if err != nil {
				t.Fatalf("Failed to get engine: %v", err)
			}

			// Get execution steps
			steps := engine.GetExecutionSteps(workflowData, "/tmp/test.log")

			// Convert steps to string for analysis
			var stepsStr strings.Builder
			for _, step := range steps {
				for _, line := range step {
					stepsStr.WriteString(line)
					stepsStr.WriteString("\n")
				}