Add more tests

Author: lukaszcl

tools/flakeguard/git/git_test.go

@@ -0,0 +1,96 @@
+package git
+
+import (
+	"bytes"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestGetChangedGoPackagesFromDiff(t *testing.T) {
+	tests := []struct {
+		name        string
+		out         string
+		projectPath string
+		excludes    []string
+		fileMap     map[string][]string
+		expected    []string
+		expectError bool
+	}{
+		{
+			name:     "Basic Case",
+			out:      "pkg1/file1.go\npkg2/file2.go\n",
+			excludes: []string{},
+			fileMap: map[string][]string{
+				"pkg1/file1.go": {"pkg1"},
+				"pkg2/file2.go": {"pkg2"},
+			},
+			expected:    []string{"pkg1", "pkg2"},
+			expectError: false,
+		},
+		{
+			name:        "Empty Input",
+			out:         "",
+			excludes:    []string{},
+			fileMap:     map[string][]string{},
+			expected:    []string{},
+			expectError: false,
+		},
+		{
+			name:     "Non-Go Files Ignored",
+			out:      "pkg1/file1.txt\npkg2/file2.go\n",
+			excludes: []string{},
+			fileMap: map[string][]string{
+				"pkg2/file2.go": {"pkg2"},
+			},
+			expected:    []string{"pkg2"},
+			expectError: false,
+		},
+		{
+			name:     "Exclusions Applied",
+			out:      "pkg1/file1.go\npkg2/file2.go\npkg3/file3.go\n",
+			excludes: []string{"pkg2"},
+			fileMap: map[string][]string{
+				"pkg1/file1.go": {"pkg1"},
+				"pkg2/file2.go": {"pkg2"},
+				"pkg3/file3.go": {"pkg3"},
+			},
+			expected:    []string{"pkg1", "pkg3"},
+			expectError: false,
+		},
+		{
+			name:     "Multiple Imports",
+			out:      "pkg1/file1.go\n",
+			excludes: []string{},
+			fileMap: map[string][]string{
+				"pkg1/file1.go": {"pkg1", "pkg1/subpkg"},
+			},
+			expected:    []string{"pkg1", "pkg1/subpkg"},
+			expectError: false,
+		},
+		{
+			name:     "Duplicate Packages",
+			out:      "pkg1/file1.go\npkg1/file1.go\n",
+			excludes: []string{},
+			fileMap: map[string][]string{
+				"pkg1/file1.go": {"pkg1"},
+			},
+			expected:    []string{"pkg1"},
+			expectError: false,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			outBuffer := bytes.Buffer{}
+			outBuffer.WriteString(tt.out)
+			result, err := GetChangedGoPackagesFromDiff(outBuffer, tt.projectPath, tt.excludes, tt.fileMap)
+			if tt.expectError {
+				assert.Error(t, err)
+			} else {
+				assert.NoError(t, err)
+				assert.ElementsMatch(t, tt.expected, result)
+			}
+		})
+	}
+}