chore: modify unit test

hertz-contrib · Jan 4, 2024 · 5f1febd · 5f1febd
1 parent 039247d
commit 5f1febd
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/ws_reverse_proxy_test.go b/ws_reverse_proxy_test.go
@@ -55,6 +55,7 @@ func TestProxy(t *testing.T) {
 
 	// proxy server
 	ps := server.Default(server.WithHostPorts(":7777"))
+	ps.NoHijackConnPool = true
 	ps.GET("/proxy", proxy.ServeHTTP)
 	go ps.Spin()
 
@@ -63,6 +64,7 @@ func TestProxy(t *testing.T) {
 	go func() {
 		// backend server
 		bs := server.Default()
+		bs.NoHijackConnPool = true
 		bs.GET("/", func(ctx context.Context, c *app.RequestContext) {
 			// Don't upgrade if original host header isn't preserved
 			host := string(c.Host())