mirror of
https://github.com/gofiber/fiber.git
synced 2025-02-19 13:07:52 +00:00
7b7dcf29f7
* run gofmt
* add t.Helper()
* Simplify assigns
* Simplify make operation
* Remove unused field in struct
* Fix typo
* Run gofumpt ./
* Consistent spacing
* len(...) can never be negative
* Use ReplaceAll
* Simplify operation
* Remove deadcode
* Fix typo
* Tidy up `} else { if ...`
* Fix AssertEqual
* Remove t.Helper() to fix go1.14.15
102 lines
2.3 KiB
Go
102 lines
2.3 KiB
Go
// ⚡️ Fiber is an Express inspired web framework written in Go with ☕️
|
|
// 📄 Github Repository: https://github.com/gofiber/fiber
|
|
// 📌 API Documentation: https://docs.gofiber.io
|
|
// 💖 Maintained and modified for Fiber by @renewerner87
|
|
package fiber
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"io/ioutil"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/gofiber/fiber/v2/utils"
|
|
)
|
|
|
|
func Test_App_Prefork_Child_Process(t *testing.T) {
|
|
// Reset test var
|
|
testPreforkMaster = true
|
|
|
|
setupIsChild(t)
|
|
defer teardownIsChild(t)
|
|
|
|
app := New()
|
|
|
|
err := app.prefork(NetworkTCP4, "invalid", nil)
|
|
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(NetworkTCP6, "[::1]:", nil))
|
|
|
|
// Create tls certificate
|
|
cer, err := tls.LoadX509KeyPair("./.github/testdata/ssl.pem", "./.github/testdata/ssl.key")
|
|
if err != nil {
|
|
utils.AssertEqual(t, nil, err)
|
|
}
|
|
config := &tls.Config{Certificates: []tls.Certificate{cer}}
|
|
|
|
go func() {
|
|
time.Sleep(1000 * time.Millisecond)
|
|
utils.AssertEqual(t, nil, app.Shutdown())
|
|
}()
|
|
|
|
utils.AssertEqual(t, nil, app.prefork(NetworkTCP4, "127.0.0.1:", config))
|
|
}
|
|
|
|
func Test_App_Prefork_Master_Process(t *testing.T) {
|
|
// Reset test var
|
|
testPreforkMaster = true
|
|
|
|
app := New()
|
|
|
|
go func() {
|
|
time.Sleep(1000 * time.Millisecond)
|
|
utils.AssertEqual(t, nil, app.Shutdown())
|
|
}()
|
|
|
|
utils.AssertEqual(t, nil, app.prefork(NetworkTCP4, ":3000", nil))
|
|
|
|
dummyChildCmd = "invalid"
|
|
|
|
err := app.prefork(NetworkTCP4, "127.0.0.1:", nil)
|
|
utils.AssertEqual(t, false, err == nil)
|
|
}
|
|
|
|
func Test_App_Prefork_Child_Process_Never_Show_Startup_Message(t *testing.T) {
|
|
setupIsChild(t)
|
|
defer teardownIsChild(t)
|
|
|
|
rescueStdout := os.Stdout
|
|
defer func() { os.Stdout = rescueStdout }()
|
|
|
|
r, w, err := os.Pipe()
|
|
utils.AssertEqual(t, nil, err)
|
|
|
|
os.Stdout = w
|
|
|
|
New().startupProcess().startupMessage(":3000", false, "")
|
|
|
|
utils.AssertEqual(t, nil, w.Close())
|
|
|
|
out, err := ioutil.ReadAll(r)
|
|
utils.AssertEqual(t, nil, err)
|
|
utils.AssertEqual(t, 0, len(out))
|
|
}
|
|
|
|
func setupIsChild(t *testing.T) {
|
|
t.Helper()
|
|
|
|
utils.AssertEqual(t, nil, os.Setenv(envPreforkChildKey, envPreforkChildVal))
|
|
}
|
|
|
|
func teardownIsChild(t *testing.T) {
|
|
t.Helper()
|
|
|
|
utils.AssertEqual(t, nil, os.Setenv(envPreforkChildKey, ""))
|
|
}
|