add more test cases

Author: knbr13

main_test.go

@@ -0,0 +1,31 @@
+package main
+
+import (
+	"os"
+	"testing"
+)
+
+func TestMainFunction(t *testing.T) {
+	originalArgs := os.Args
+	defer func() { os.Args = originalArgs }()
+
+	os.Args = []string{"cmd", "-path=./test_data", "[email protected]"}
+
+	exitCode := 0
+	func() {
+		defer func() {
+			if r := recover(); r != nil {
+				if code, ok := r.(int); ok {
+					exitCode = code
+				} else {
+					t.Fatalf("unexpected panic: %v", r)
+				}
+			}
+		}()
+		main()
+	}()
+
+	if exitCode != 0 {
+		t.Errorf("main() exited with code %d; expected 0", exitCode)
+	}
+}

print_test.go

@@ -94,11 +94,17 @@ func TestPrintCell(t *testing.T) {
 		expectedMessage string
 	}{
 		{
-			description:     "Zero value (/8)",
+			description:     "Zero value",
 			val:             0,
 			maxValue:        10,
 			expectedMessage: color.New(color.FgWhite, color.BgBlack).Sprintf("  - "),
 		},
+		{
+			description:     "Lower bound value (/8)",
+			val:             1,
+			maxValue:        10,
+			expectedMessage: color.New(color.FgBlack, color.BgLightCyan).Sprintf("  1 "),
+		},
 		{
 			description:     "Lower bound value (/4)",
 			val:             2,

scan_test.go

@@ -14,9 +14,10 @@ func TestScanGitFolders(t *testing.T) {
 	}
 
 	test := []struct {
-		Name string
-		Root string
-		Want []string
+		Name      string
+		Root      string
+		Want      []string
+		ExpectErr bool
 	}{
 		{
 			Name: "5 expected repos",
@@ -34,11 +35,23 @@ func TestScanGitFolders(t *testing.T) {
 			Root: path.Join(wd, ".github"),
 			Want: []string{},
 		},
+		{
+			Name:      "path does not exist",
+			Root:      path.Join(wd, "does_not_exist"),
+			Want:      []string{},
+			ExpectErr: true,
+		},
 	}
 
 	for _, tt := range test {
 		t.Run(tt.Name, func(t *testing.T) {
 			got, err := scanGitFolders(tt.Root)
+			if tt.ExpectErr {
+				if err == nil {
+					t.Fatalf("expected error, got nil")
+				}
+				return
+			}
 			if err != nil {
 				t.Fatalf("failed to scan git folders: %v", err)
 			}

stats_test.go

@@ -100,17 +100,20 @@ func TestProcessRepos(t *testing.T) {
 	days := daysAgo(commitsDate)
 
 	tests := []struct {
-		Name     string
-		Repos    []string
-		Email    string
-		Expected map[int]int
+		Name      string
+		Repos     []string
+		Email     string
+		Expected  map[int]int
+		ExpectErr bool
 	}{
 		{
 			Name: "test 1",
 			Repos: []string{
 				path.Join(wd, "test_data", "project_1"),
 				path.Join(wd, "test_data", "project_2"),
 				path.Join(wd, "test_data", "project_3"),
+				path.Join(wd, "test_data", "project_that_has_future_commits"),
+				path.Join(wd, "test_data", "project_by_another_contributor"),
 			},
 			Email:    "[email protected]",
 			Expected: map[int]int{days: 9},
@@ -121,6 +124,13 @@ func TestProcessRepos(t *testing.T) {
 			Email:    "[email protected]",
 			Expected: map[int]int{},
 		},
+		{
+			Name:      "test 3",
+			Repos:     []string{path.Join(wd, "test_data", "repo_that_does_not_exist")},
+			Email:     "[email protected]",
+			Expected:  map[int]int{},
+			ExpectErr: true,
+		},
 	}
 
 	b := Boundary{
@@ -131,6 +141,12 @@ func TestProcessRepos(t *testing.T) {
 	for _, tt := range tests {
 		t.Run(tt.Name, func(t *testing.T) {
 			commits := processRepos(tt.Repos, tt.Email, b)
+			if tt.ExpectErr {
+				if len(commits) != 0 {
+					t.Fatalf("expected error, got commits results: %v", commits)
+				}
+				return
+			}
 			if len(commits) != len(tt.Expected) {
 				t.Errorf("processRepos11() = %v, want %v", commits, tt.Expected)
 			}

utils_test.go

@@ -129,6 +129,13 @@ func TestDaysAgo(t *testing.T) {
 	}
 }
 
+func TestGetGlobalEmailFromGit(t *testing.T) {
+	email := getGlobalEmailFromGit()
+	if email == "" {
+		t.Errorf("Expected email, got empty string")
+	}
+}
+
 func TestGetMaxValue(t *testing.T) {
 	type args struct {
 		m map[int]int
@@ -248,6 +255,13 @@ func TestSetTimeFlags(t *testing.T) {
 			expectedUntil: time.Time{},
 			expectedError: "invalid 'until' date format. please use the format: 2006-01-02",
 		},
+		{
+			name:          "until flag is in the future",
+			untilflag:     "2099-01-01",
+			sinceflag:     "2022-01-01",
+			expectedSince: time.Date(2022, time.January, 1, 0, 0, 0, 0, time.UTC),
+			expectedUntil: now,
+		},
 	}
 
 	for _, tt := range tests {