Fix badPing test duration

coder · Feb 13, 2020 · c5b0a00 · c5b0a00
1 parent 3673c2c
commit c5b0a00
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/conn_test.go b/conn_test.go
@@ -95,7 +95,10 @@ func TestConn(t *testing.T) {
 
 		c2.CloseRead(tt.ctx)
 
-		err := c1.Ping(tt.ctx)
+		ctx, cancel := context.WithTimeout(tt.ctx, time.Millisecond*100)
+		defer cancel()
+
+		err := c1.Ping(ctx)
 		assert.Contains(t, err, "failed to wait for pong")
 	})