Improve test coverage

Author: nhooyr

websocket.go

@@ -846,8 +846,9 @@ func (c *Conn) ping(ctx context.Context) error {
 	case <-c.closed:
 		return c.closeErr
 	case <-ctx.Done():
-		c.close(xerrors.Errorf("failed to ping: %w", ctx.Err()))
-		return ctx.Err()
+		err := xerrors.Errorf("failed to wait for pong: %w", ctx.Err())
+		c.close(err)
+		return err
 	case <-pong:
 		return nil
 	}

websocket_test.go

@@ -263,7 +263,6 @@ func TestHandshake(t *testing.T) {
 				if err != nil {
 					return err
 				}
-				// Read twice to ensure the un EOFed previous reader works correctly.
 				err = read()
 				if err != nil {
 					return err
@@ -843,11 +842,6 @@ func benchConn(b *testing.B, echo, stream bool, size int) {
 			if err != nil {
 				b.Fatal(err)
 			}
-
-			_, err = r.Read([]byte{0})
-			if !xerrors.Is(err, io.EOF) {
-				b.Fatalf("more data in reader than needed")
-			}
 		}
 	}
 	b.StopTimer()