👍 improve prefork test coverage

prefork.go

@@ -17,6 +17,8 @@ const (
 	envPreforkChildVal = "1"
 )
 
+var testPreforkMaster = false
+
 // IsChild determines if the current process is a result of Prefork
 func (app *App) IsChild() bool {
 	return os.Getenv(envPreforkChildKey) == envPreforkChildVal
@@ -43,11 +45,12 @@ func (app *App) prefork(addr string, tlsconfig ...*tls.Config) (err error) {
 
 		// kill child proc when master exits
 		go func() {
-			ppid, err := os.FindProcess(os.Getppid())
+			p, err := os.FindProcess(os.Getppid())
 			if err == nil {
-				_, _ = ppid.Wait()
+				_, _ = p.Wait()
+			} else {
+				os.Exit(1)
 			}
-			os.Exit(1)
 		}()
 		// listen for incoming connections
 		return app.server.Serve(ln)
@@ -71,12 +74,15 @@ func (app *App) prefork(addr string, tlsconfig ...*tls.Config) (err error) {
 	}()
 
 	// collect child pids
-	pids := []string{}
+	var pids []string
 
 	// launch child procs
 	for i := 0; i < max; i++ {
 		/* #nosec G204 */
 		cmd := exec.Command(os.Args[0], os.Args[1:]...)
+		if testPreforkMaster {
+			cmd = exec.Command("cd")
+		}
 		cmd.Stdout = os.Stdout
 		cmd.Stderr = os.Stderr
 
@@ -89,12 +95,13 @@ func (app *App) prefork(addr string, tlsconfig ...*tls.Config) (err error) {
 		}
 
 		// store child process
-		childs[cmd.Process.Pid] = cmd
-		pids = append(pids, strconv.Itoa(cmd.Process.Pid))
+		pid := cmd.Process.Pid
+		childs[pid] = cmd
+		pids = append(pids, strconv.Itoa(pid))
 
 		// notify master if child crashes
 		go func() {
-			channel <- child{cmd.Process.Pid, cmd.Wait()}
+			channel <- child{pid, cmd.Wait()}
 		}()
 	}
 

prefork_test.go

@@ -0,0 +1,42 @@
+package fiber
+
+import (
+	"os"
+	"testing"
+	"time"
+
+	utils "github.com/gofiber/utils"
+)
+
+func Test_App_Prefork_Child_Process(t *testing.T) {
+	utils.AssertEqual(t, nil, os.Setenv(envPreforkChildKey, envPreforkChildVal))
+	defer os.Setenv(envPreforkChildKey, "")
+
+	app := New(&Settings{
+		DisableStartupMessage: true,
+	})
+	app.init()
+
+	go func() {
+		time.Sleep(1000 * time.Millisecond)
+		utils.AssertEqual(t, nil, app.Shutdown())
+	}()
+
+	utils.AssertEqual(t, nil, app.prefork("127.0.0.1:"))
+}
+
+func Test_App_Prefork_Main_Process(t *testing.T) {
+	testPreforkMaster = true
+
+	app := New(&Settings{
+		DisableStartupMessage: true,
+	})
+	app.init()
+
+	go func() {
+		time.Sleep(1000 * time.Millisecond)
+		utils.AssertEqual(t, nil, app.Shutdown())
+	}()
+
+	utils.AssertEqual(t, nil, app.prefork("127.0.0.1:"))
+}