Function TestAddEngineToWorkflow

pkg/cli/add_source_test.go:184–307

TestAddEngineToWorkflow tests the addEngineToWorkflow function

(t *testing.T)

// TestAddEngineToWorkflow tests the addEngineToWorkflow function
func TestAddEngineToWorkflow(t *testing.T) {
	tests := []struct {
		name            string
		content         string
		engine          string
		expectError     bool
		checkEngine     bool
		expectBlankLine bool // expect blank line after engine declaration
	}{
		{
			name: "add_new_engine_field_has_trailing_blank_line",
			content: `---
on: push
permissions:
  contents: read
---

# Test Workflow`,
			engine:          "claude",
			expectError:     false,
			checkEngine:     true,
			expectBlankLine: true,
		},
		{
			name: "update_existing_engine_field_no_extra_blank_line",
			content: `---
on: push
engine: codex
---

# Test Workflow`,
			engine:          "claude",
			expectError:     false,
			checkEngine:     true,
			expectBlankLine: false, // updating existing field doesn't add a blank line
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			result, err := addEngineToWorkflow(tt.content, tt.engine)

			if tt.expectError && err == nil {
				t.Errorf("addEngineToWorkflow() expected error, got nil")
				return
			}

			if !tt.expectError && err != nil {
				t.Errorf("addEngineToWorkflow() error = %v", err)
				return
			}

			if !tt.expectError && tt.checkEngine {
				// Verify that the engine field is present with the correct value
				if !strings.Contains(result, "engine: "+tt.engine) {
					t.Errorf("addEngineToWorkflow() result does not contain 'engine: %s'. Result:\n%s", tt.engine, result)
				}