Merge pull request #778 from Fenny/master

🔨 fix proxy test
gofiber · Sep 14, 2020 · f3434e5 · f3434e5
2 parents 67a4cb1 + 1a3bd53
commit f3434e5
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/middleware/proxy/proxy_test.go b/middleware/proxy/proxy_test.go
@@ -6,6 +6,7 @@ import (
 	"net/http/httptest"
 	"strings"
 	"testing"
+	"time"
 
 	"github.com/gofiber/fiber/v2"
 	"github.com/gofiber/fiber/v2/utils"
@@ -56,6 +57,8 @@ func Test_Proxy(t *testing.T) {
 		utils.AssertEqual(t, nil, target.Listen(":3001"))
 	}()
 
+	time.Sleep(time.Second)
+
 	resp, err := target.Test(httptest.NewRequest("GET", "/", nil), 2000)
 	utils.AssertEqual(t, nil, err)
 	utils.AssertEqual(t, fiber.StatusTeapot, resp.StatusCode)
@@ -116,6 +119,8 @@ func Test_Proxy_After_With_Error(t *testing.T) {
 		utils.AssertEqual(t, nil, target.Listen(":3002"))
 	}()
 
+	time.Sleep(time.Second)
+
 	resp, err := target.Test(httptest.NewRequest("GET", "/", nil), 2000)
 	utils.AssertEqual(t, nil, err)
 	utils.AssertEqual(t, fiber.StatusTeapot, resp.StatusCode)