1
0
mirror of https://github.com/gofiber/fiber.git synced 2025-02-22 20:04:20 +00:00
fiber/prefork_test.go
2020-07-15 15:20:10 +02:00

49 lines
969 B
Go

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()
app.Settings.DisableStartupMessage = true
app.init()
err := app.prefork("invalid")
utils.AssertEqual(t, false, err == nil)
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()
app.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:"))
dummyChildCmd = "invalid"
err := app.prefork("127.0.0.1:")
utils.AssertEqual(t, false, err == nil)
}