Improve startup message tests (#1504)

* improve startup message tests

app_test.go

@@ -11,13 +11,16 @@ import (
 	"fmt"
 	"io"
 	"io/ioutil"
+	"log"
 	"mime/multipart"
 	"net"
 	"net/http"
 	"net/http/httptest"
+	"os"
 	"reflect"
 	"regexp"
 	"strings"
+	"sync"
 	"testing"
 	"time"
 
@@ -1257,15 +1260,57 @@ func Test_App_SmallReadBuffer(t *testing.T) {
 	utils.AssertEqual(t, nil, app.Listen(":4006"))
 }
 
+func captureOutput(f func()) string {
+	reader, writer, err := os.Pipe()
+	if err != nil {
+		panic(err)
+	}
+	stdout := os.Stdout
+	stderr := os.Stderr
+	defer func() {
+		os.Stdout = stdout
+		os.Stderr = stderr
+		log.SetOutput(os.Stderr)
+	}()
+	os.Stdout = writer
+	os.Stderr = writer
+	log.SetOutput(writer)
+	out := make(chan string)
+	wg := new(sync.WaitGroup)
+	wg.Add(1)
+	go func() {
+		var buf bytes.Buffer
+		wg.Done()
+		io.Copy(&buf, reader)
+		out <- buf.String()
+	}()
+	wg.Wait()
+	f()
+	writer.Close()
+	return <-out
+}
+
 func Test_App_Master_Process_Show_Startup_Message(t *testing.T) {
-	New(Config{Prefork: true}).
-		startupMessage(":3000", true, strings.Repeat(",11111,22222,33333,44444,55555,60000", 10))
+	startupMessage := captureOutput(func() {
+		New(Config{Prefork: true}).
+			startupMessage(":3000", true, strings.Repeat(",11111,22222,33333,44444,55555,60000", 10))
+	})
+	fmt.Println(startupMessage)
+	utils.AssertEqual(t, true, strings.Contains(startupMessage, "https://127.0.0.1:3000"))
+	utils.AssertEqual(t, true, strings.Contains(startupMessage, "(bound on host 0.0.0.0 and port 3000)"))
+	utils.AssertEqual(t, true, strings.Contains(startupMessage, "Child PIDs"))
+	utils.AssertEqual(t, true, strings.Contains(startupMessage, "11111, 22222, 33333, 44444, 55555, 60000"))
+	utils.AssertEqual(t, true, strings.Contains(startupMessage, "Prefork ........ Enabled"))
 }
 
 func Test_App_Master_Process_Show_Startup_MessageWithAppName(t *testing.T) {
 	app := New(Config{Prefork: true, AppName: "Test App v1.0.1"})
-	app.startupMessage(":3000", true, strings.Repeat(",11111,22222,33333,44444,55555,60000", 10))
+	startupMessage := captureOutput(func() {
+		app.startupMessage(":3000", true, strings.Repeat(",11111,22222,33333,44444,55555,60000", 10))
+	})
+	fmt.Println(startupMessage)
 	utils.AssertEqual(t, "Test App v1.0.1", app.Config().AppName)
+	utils.AssertEqual(t, true, strings.Contains(startupMessage, app.Config().AppName))
 }
 
 func Test_App_Server(t *testing.T) {