From d6ee49dce5b2402e2f5fddb30e29e31833651da3 Mon Sep 17 00:00:00 2001
From: Fenny <25108519+Fenny@users.noreply.github.com>
Date: Thu, 22 Oct 2020 12:16:04 +0200
Subject: [PATCH] =?UTF-8?q?=F0=9F=A9=B9=20fix=20workflow?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 middleware/proxy/proxy_test.go | 29 ++++++++++++++---------------
 1 file changed, 14 insertions(+), 15 deletions(-)

diff --git a/middleware/proxy/proxy_test.go b/middleware/proxy/proxy_test.go
index 9a8cb79b..04350e3b 100644
--- a/middleware/proxy/proxy_test.go
+++ b/middleware/proxy/proxy_test.go
@@ -40,16 +40,14 @@ func Test_Proxy_Next(t *testing.T) {
 // go test -run Test_Proxy
 func Test_Proxy(t *testing.T) {
 	target := fiber.New()
-
 	target.Get("/", func(c *fiber.Ctx) error {
 		return c.SendStatus(fiber.StatusTeapot)
 	})
-
 	go func() {
 		utils.AssertEqual(t, nil, target.Listen(":3001"))
 	}()
 
-	time.Sleep(time.Second)
+	time.Sleep(1 * time.Second)
 
 	resp, err := target.Test(httptest.NewRequest("GET", "/", nil), 2000)
 	utils.AssertEqual(t, nil, err)
@@ -85,12 +83,13 @@ func Test_Proxy_Forward(t *testing.T) {
 	app := fiber.New()
 
 	target := fiber.New(fiber.Config{DisableStartupMessage: true})
-	go func() {
-		utils.AssertEqual(t, nil, target.Listen(":50001"))
-	}()
 	target.Get("/", func(c *fiber.Ctx) error {
 		return c.SendString("forwarded")
 	})
+	go func() {
+		utils.AssertEqual(t, nil, target.Listen(":50001"))
+	}()
+	time.Sleep(1 * time.Second)
 
 	app.Use(Forward("http://127.0.0.1:50001"))
 
@@ -105,9 +104,13 @@ func Test_Proxy_Forward(t *testing.T) {
 
 func Test_Proxy_Modify_Response(t *testing.T) {
 	target := fiber.New(fiber.Config{DisableStartupMessage: true})
+	target.Get("/", func(c *fiber.Ctx) error {
+		return c.Status(500).SendString("not modified")
+	})
 	go func() {
 		utils.AssertEqual(t, nil, target.Listen(":50002"))
 	}()
+	time.Sleep(1 * time.Second)
 
 	app := fiber.New()
 	app.Use(Balancer(Config{
@@ -118,10 +121,6 @@ func Test_Proxy_Modify_Response(t *testing.T) {
 		},
 	}))
 
-	target.Get("/", func(c *fiber.Ctx) error {
-		return c.SendString("not modified")
-	})
-
 	resp, err := app.Test(httptest.NewRequest("GET", "/", nil))
 	utils.AssertEqual(t, nil, err)
 	utils.AssertEqual(t, fiber.StatusOK, resp.StatusCode)
@@ -133,9 +132,14 @@ func Test_Proxy_Modify_Response(t *testing.T) {
 
 func Test_Proxy_Modify_Request(t *testing.T) {
 	target := fiber.New(fiber.Config{DisableStartupMessage: true})
+	target.Get("/", func(c *fiber.Ctx) error {
+		b := c.Request().Body()
+		return c.SendString(string(b))
+	})
 	go func() {
 		utils.AssertEqual(t, nil, target.Listen(":50003"))
 	}()
+	time.Sleep(1 * time.Second)
 
 	app := fiber.New()
 	app.Use(Balancer(Config{
@@ -146,11 +150,6 @@ func Test_Proxy_Modify_Request(t *testing.T) {
 		},
 	}))
 
-	target.Get("/", func(c *fiber.Ctx) error {
-		b := c.Request().Body()
-		return c.SendString(string(b))
-	})
-
 	resp, err := app.Test(httptest.NewRequest("GET", "/", nil))
 	utils.AssertEqual(t, nil, err)
 	utils.AssertEqual(t, fiber.StatusOK, resp.StatusCode)