refine test case

cmd/goodbye/main_test.go

@@ -10,7 +10,7 @@ import (
 	"github.com/amyangfei/fpcov/pkg/hello"
 )
 
-func TestRunMain(t *testing.T) {
+func test1(t *testing.T) {
 	var (
 		args   []string
 		exit   = make(chan int)
@@ -42,8 +42,46 @@ func TestRunMain(t *testing.T) {
 
 	select {
 	case <-waitCh:
-		return
 	case <-exit:
-		return
 	}
 }
+
+func test2(t *testing.T) {
+	var (
+		args []string
+		exit = make(chan int)
+	)
+	for _, arg := range os.Args {
+		switch {
+		case arg == "DEVEL":
+		case strings.HasPrefix(arg, "-test."):
+		default:
+			args = append(args, arg)
+		}
+	}
+
+	oldOsExit := hello.OsExit
+	defer func() { hello.OsExit = oldOsExit }()
+	hello.OsExit = func(code int) {
+		log.Printf("[test] os.Exit with code %d", code)
+		exit <- code
+		// sleep here to prevent following code execution in the caller routine
+		time.Sleep(time.Second * 60)
+	}
+
+	go func() {
+		select {
+		case <-exit:
+			time.Sleep(time.Millisecond * 100)
+			os.Exit(0)
+		}
+	}()
+
+	os.Args = args
+	main()
+}
+
+func TestRunMain(t *testing.T) {
+	// test1(t)
+	test2(t)
+}

tests/run.sh

@@ -5,4 +5,4 @@ CUR=$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )
 
 mkdir -p $TEST_DIR
 
-$CUR/../bin/goodbye.test -test.coverprofile="$TEST_DIR/goodbye.$(date +"%s").out" DEVEL >> $TEST_DIR/goodbye.log 2>&1
+$CUR/../bin/goodbye.test -test.coverprofile="$TEST_DIR/cov.goodbye.$(date +"%s").out" DEVEL >> $TEST_DIR/goodbye.log 2>&1