Browse Source

Try to fix infinite conn read in tests

yuhan6665 1 năm trước cách đây
mục cha
commit
60b2c349d2

+ 2 - 0
transport/internet/httpupgrade/httpupgrade_test.go

@@ -28,6 +28,7 @@ func Test_listenHTTPUpgradeAndDial(t *testing.T) {
 			defer c.Close()
 
 			var b [1024]byte
+			c.SetReadDeadline(time.Now().Add(2 * time.Second))
 			_, err := c.Read(b[:])
 			if err != nil {
 				return
@@ -87,6 +88,7 @@ func Test_listenHTTPUpgradeAndDialWithHeaders(t *testing.T) {
 			defer c.Close()
 
 			var b [1024]byte
+			c.SetReadDeadline(time.Now().Add(2 * time.Second))
 			_, err := c.Read(b[:])
 			if err != nil {
 				return

+ 1 - 0
transport/internet/splithttp/splithttp_test.go

@@ -33,6 +33,7 @@ func Test_listenSHAndDial(t *testing.T) {
 			defer c.Close()
 
 			var b [1024]byte
+			c.SetReadDeadline(time.Now().Add(2 * time.Second))
 			_, err := c.Read(b[:])
 			if err != nil {
 				return

+ 1 - 0
transport/internet/websocket/ws_test.go

@@ -28,6 +28,7 @@ func Test_listenWSAndDial(t *testing.T) {
 			defer c.Close()
 
 			var b [1024]byte
+			c.SetReadDeadline(time.Now().Add(2 * time.Second))
 			_, err := c.Read(b[:])
 			if err != nil {
 				return