Function TestPiEngine_GetExecutionSteps_WithModel

pkg/workflow/pi_engine_test.go:199–217

(t *testing.T)

}

func TestPiEngine_GetExecutionSteps_WithModel(t *testing.T) {
	engine := NewPiEngine()
	workflowData := &WorkflowData{
		Name:         "test-workflow",
		EngineConfig: &EngineConfig{ID: "pi", Model: "copilot/claude-sonnet-4"},
		ParsedTools:  NewTools(map[string]any{}),
	}
	steps := engine.GetExecutionSteps(workflowData, "/tmp/gh-aw/agent-stdio.log")
	require.NotEmpty(t, steps, "Steps should not be empty")

	stepText := strings.Join(steps[0], "\n")
	// When firewall is not enabled, Pi is invoked with the --model flag using the
	// native github-copilot provider (Pi's built-in provider for GitHub Copilot).
	assert.Contains(t, stepText, "--model", "Step should pass --model flag to Pi CLI")
	assert.Contains(t, stepText, "github-copilot", "Non-firewall copilot model should use github-copilot/ provider prefix")
	assert.Contains(t, stepText, "claude-sonnet-4", "Step should include the model ID portion")
	assert.Contains(t, stepText, "GH_AW_PI_MODEL", "Step should expose the original workflow model to Pi extensions")
	assert.NotContains(t, stepText, "\n          PI_MODEL:", "Step should not set PI_MODEL in the environment when the CLI model is passed via --model")
}

func TestPiEngine_GetExecutionSteps_IgnoresRedundantYoloArg(t *testing.T) {
	engine := NewPiEngine()