fix: datastream client test timeouts (#1552)

0xPolygonHermez · Dec 12, 2024 · 6239b09 · 6239b09
1 parent cb5f06a
commit 6239b09
Showing 1 changed file with 18 additions and 18 deletions.
diff --git a/zk/datastream/client/stream_client_test.go b/zk/datastream/client/stream_client_test.go
@@ -50,13 +50,13 @@ func TestStreamClientReadHeaderEntry(t *testing.T) {
 	}
 
 	for _, testCase := range testCases {
-		c := NewClient(context.Background(), "", false, 0, 500*time.Millisecond, 0)
-		server, conn := net.Pipe()
-		defer server.Close()
-		defer c.Stop()
-
-		c.conn = conn
 		t.Run(testCase.name, func(t *testing.T) {
+			c := NewClient(context.Background(), "", false, 0, 500*time.Millisecond, 0)
+			server, conn := net.Pipe()
+			defer server.Close()
+			defer c.Stop()
+
+			c.conn = conn
 			go func() {
 				server.Write(testCase.input)
 				server.Close()
@@ -118,13 +118,13 @@ func TestStreamClientReadResultEntry(t *testing.T) {
 	}
 
 	for _, testCase := range testCases {
-		c := NewClient(context.Background(), "", false, 0, 500*time.Millisecond, 0)
-		server, conn := net.Pipe()
-		defer server.Close()
-		defer c.Stop()
-
-		c.conn = conn
 		t.Run(testCase.name, func(t *testing.T) {
+			c := NewClient(context.Background(), "", false, 0, 500*time.Millisecond, 0)
+			server, conn := net.Pipe()
+			defer server.Close()
+			defer c.Stop()
+
+			c.conn = conn
 			go func() {
 				server.Write(testCase.input)
 				server.Close()
@@ -191,13 +191,13 @@ func TestStreamClientReadFileEntry(t *testing.T) {
 		},
 	}
 	for _, testCase := range testCases {
-		c := NewClient(context.Background(), "", false, 0, 500*time.Millisecond, 0)
-		server, conn := net.Pipe()
-		defer c.Stop()
-		defer server.Close()
-
-		c.conn = conn
 		t.Run(testCase.name, func(t *testing.T) {
+			c := NewClient(context.Background(), "", false, 0, 500*time.Millisecond, 0)
+			server, conn := net.Pipe()
+			defer c.Stop()
+			defer server.Close()
+
+			c.conn = conn
 			go func() {
 				server.Write(testCase.input)
 				server.Close()