add_gitattributes_test.go

//go:build !integration

package cli

import (
	"context"
	"os"
	"os/exec"
	"path/filepath"
	"strings"
	"testing"
)

// TestAddCommandUpdatesGitAttributes tests that the add command updates .gitattributes by default
func TestAddCommandUpdatesGitAttributes(t *testing.T) {
	// Create a temporary directory for testing
	tmpDir, err := os.MkdirTemp("", "gh-aw-add-gitattributes-test")
	if err != nil {
		t.Fatalf("Failed to create temp dir: %v", err)
	}
	defer os.RemoveAll(tmpDir)

	// Change to temp directory
	originalDir, err := os.Getwd()
	if err != nil {
		t.Fatalf("Failed to get current directory: %v", err)
	}
	defer func() {
		_ = os.Chdir(originalDir)
	}()

	if err := os.Chdir(tmpDir); err != nil {
		t.Fatalf("Failed to change to temp directory: %v", err)
	}

	// Initialize a git repository
	if err := os.WriteFile("test.txt", []byte("test"), 0644); err != nil {
		t.Fatalf("Failed to create test file: %v", err)
	}
	if cmd := exec.Command("git", "init"); cmd.Run() != nil {
		t.Skip("Skipping test - git not available")
	}
	if cmd := exec.Command("git", "config", "user.email", "test@example.com"); cmd.Run() != nil {
		t.Skip("Skipping test - git config failed")
	}
	if cmd := exec.Command("git", "config", "user.name", "Test User"); cmd.Run() != nil {
		t.Skip("Skipping test - git config failed")
	}

	// Create .github/workflows directory
	workflowsDir := filepath.Join(tmpDir, ".github", "workflows")
	if err := os.MkdirAll(workflowsDir, 0755); err != nil {
		t.Fatalf("Failed to create .github/workflows directory: %v", err)
	}

	// Create a minimal workflow content for testing
	workflowContent := `---
on: push
permissions:
  contents: read
engine: copilot
---

# Test Workflow

This is a test workflow.`

	// Create a ResolvedWorkflow for testing
	resolved := &ResolvedWorkflow{
		Spec: &WorkflowSpec{
			RepoSpec: RepoSpec{
				RepoSlug: "test/repo",
				Version:  "",
			},
			WorkflowPath: "./test.md",
			WorkflowName: "test",
		},
		Content: []byte(workflowContent),
		SourceInfo: &FetchedWorkflow{
			Content:    []byte(workflowContent),
			IsLocal:    true,
			SourcePath: "./test.md",
			CommitSHA:  "",
		},
	}

	t.Run("gitattributes_updated_by_default", func(t *testing.T) {
		// Remove any existing .gitattributes
		os.Remove(".gitattributes")

		// Call addWorkflows with noGitattributes=false
		opts := AddOptions{}
		err := addWorkflows(context.Background(), []*ResolvedWorkflow{resolved}, opts)
		if err != nil {
			// Log any error but don't fail - we're testing gitattributes behavior
			t.Logf("Note: workflow addition returned: %v", err)
		}

		// Check that .gitattributes was created
		gitAttributesPath := filepath.Join(tmpDir, ".gitattributes")
		if _, err := os.Stat(gitAttributesPath); os.IsNotExist(err) {
			t.Errorf("Expected .gitattributes file to be created")
			return
		}

		// Check content
		content, err := os.ReadFile(gitAttributesPath)
		if err != nil {
			t.Fatalf("Failed to read .gitattributes: %v", err)
		}

		if !strings.Contains(string(content), ".github/workflows/*.lock.yml linguist-generated=true") {
			t.Errorf("Expected .gitattributes to contain '.github/workflows/*.lock.yml linguist-generated=true'")
		}
	})

	t.Run("gitattributes_not_updated_with_flag", func(t *testing.T) {
		// Remove any existing .gitattributes
		os.Remove(".gitattributes")

		opts := AddOptions{NoGitattributes: true}
		// Call addWorkflows with noGitattributes=true
		err := addWorkflows(context.Background(), []*ResolvedWorkflow{resolved}, opts)
		if err != nil {
			// Log any error but don't fail - we're testing gitattributes behavior
			t.Logf("Note: workflow addition returned: %v", err)
		}

		// Check that .gitattributes was NOT created
		gitAttributesPath := filepath.Join(tmpDir, ".gitattributes")
		if _, err := os.Stat(gitAttributesPath); !os.IsNotExist(err) {
			t.Errorf("Expected .gitattributes file NOT to be created when --no-gitattributes flag is set")
		}
	})

	t.Run("existing_gitattributes_not_modified_with_flag", func(t *testing.T) {
		// Create a .gitattributes file with existing content
		existingContent := "*.txt linguist-vendored=true\n"
		gitAttributesPath := filepath.Join(tmpDir, ".gitattributes")
		if err := os.WriteFile(gitAttributesPath, []byte(existingContent), 0644); err != nil {
			t.Fatalf("Failed to create .gitattributes: %v", err)
		}

		opts := AddOptions{NoGitattributes: true}
		// Call addWorkflows with noGitattributes=true
		err := addWorkflows(context.Background(), []*ResolvedWorkflow{resolved}, opts)
		if err != nil {
			// Log any error but don't fail - we're testing gitattributes behavior
			t.Logf("Note: workflow addition returned: %v", err)
		}

		// Check that .gitattributes was NOT modified
		content, err := os.ReadFile(gitAttributesPath)
		if err != nil {
			t.Fatalf("Failed to read .gitattributes: %v", err)
		}

		if string(content) != existingContent {
			t.Errorf("Expected .gitattributes to remain unchanged when --no-gitattributes flag is set.\nExpected:\n%s\nGot:\n%s",
				existingContent, string(content))
		}
	})
}