Improve test code

pkg/gee/gee.go

@@ -4,6 +4,7 @@ import (
 	"bufio"
 	"os"
 	"strconv"
+
 	// "sync"
 
 	model "github.com/hahwul/gee/pkg/model"

pkg/gee/gee_test.go

@@ -1,6 +1,7 @@
 package gee
 
 import (
+	"os"
 	"testing"
 
 	model "github.com/hahwul/gee/pkg/model"
@@ -15,7 +16,7 @@ func TestGee(t *testing.T) {
 		args args
 	}{
 		{
-			name: "test",
+			name: "test debug",
 			args: args{
 				options: model.Options{
 					Append: true,
@@ -25,7 +26,7 @@ func TestGee(t *testing.T) {
 			},
 		},
 		{
-			name: "test",
+			name: "test uniq",
 			args: args{
 				options: model.Options{
 					Append: true,
@@ -34,9 +35,64 @@ func TestGee(t *testing.T) {
 				},
 			},
 		},
+		{
+			name: "test regex",
+			args: args{
+				options: model.Options{
+					Append: false,
+					Files:  []string{"", ""},
+					Regex:  "HAHWUL",
+				},
+			},
+		},
+		{
+			name: "test regexv",
+			args: args{
+				options: model.Options{
+					Append: false,
+					Files:  []string{"", ""},
+					RegexV: "HAHWUL",
+				},
+			},
+		},
+		{
+			name: "test chunkedLine",
+			args: args{
+				options: model.Options{
+					Append:      false,
+					Files:       []string{"", ""},
+					ChunkedLine: 1,
+				},
+			},
+		},
+		{
+			name: "test distribute",
+			args: args{
+				options: model.Options{
+					Append:     false,
+					Files:      []string{"", ""},
+					Distribute: true,
+				},
+			},
+		},
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
+			input := []byte("Alice")
+			r, w, err := os.Pipe()
+			if err != nil {
+				t.Fatal(err)
+			}
+
+			_, err = w.Write(input)
+			if err != nil {
+				t.Error(err)
+			}
+			w.Close()
+			stdin := os.Stdin
+			// Restore stdin right after the test.
+			defer func() { os.Stdin = stdin }()
+			os.Stdin = r
 			Gee(tt.args.options)
 		})
 	}