Improve wasm test coverage

Author: nhooyr

assert_test.go

@@ -2,7 +2,9 @@ package websocket_test
 
 import (
 	"context"
+	"encoding/hex"
 	"fmt"
+	"math/rand"
 	"reflect"
 
 	"github.com/google/go-cmp/cmp"
@@ -91,9 +93,32 @@ func assertJSONRead(ctx context.Context, c *websocket.Conn, exp interface{}) err
 }
 
 func randBytes(n int) []byte {
-	return make([]byte, n)
+	b := make([]byte, n)
+	rand.Read(b)
+	return b
 }
 
 func randString(n int) string {
-	return string(randBytes(n))
+	return hex.EncodeToString(randBytes(n))[:n]
+}
+
+func assertEcho(ctx context.Context, c *websocket.Conn, typ websocket.MessageType, n int) error {
+	p := randBytes(n)
+	err := c.Write(ctx, typ, p)
+	if err != nil {
+		return err
+	}
+	typ2, p2, err := c.Read(ctx)
+	if err != nil {
+		return err
+	}
+	err = assertEqualf(typ, typ2, "unexpected data type")
+	if err != nil {
+		return err
+	}
+	return assertEqualf(p, p2, "unexpected payload")
+}
+
+func assertSubprotocol(c *websocket.Conn, exp string) error {
+	return assertEqualf(exp, c.Subprotocol(), "unexpected subprotocol")
 }

ci/wasm.sh

@@ -9,19 +9,22 @@ GOOS=js GOARCH=wasm go vet ./...
 go install golang.org/x/lint/golint
 GOOS=js GOARCH=wasm golint -set_exit_status ./...
 
-wsEchoOut="$(mktemp -d)/stdout"
-mkfifo "$wsEchoOut"
-go install ./internal/wsecho/cmd/wsecho
-wsecho > "$wsEchoOut" &
+wsjstestOut="$(mktemp -d)/stdout"
+mkfifo "$wsjstestOut"
+go install ./internal/wsjstest
+timeout 30s wsjstest > "$wsjstestOut" &
+wsjstestPID=$!
 
-WS_ECHO_SERVER_URL="$(timeout 10s head -n 1 "$wsEchoOut")" || true
+WS_ECHO_SERVER_URL="$(timeout 10s head -n 1 "$wsjstestOut")" || true
 if [[ -z $WS_ECHO_SERVER_URL ]]; then
-  echo "./internal/wsecho/cmd/wsecho failed to start in 10s"
+  echo "./internal/wsjstest failed to start in 10s"
   exit 1
 fi
 
 go install github.com/agnivade/wasmbrowsertest
 GOOS=js GOARCH=wasm go test -exec=wasmbrowsertest ./... -args "$WS_ECHO_SERVER_URL"
 
-kill %1
-wait -n || true
+if ! wait "$wsjstestPID"; then
+  echo "wsjstest exited unsuccessfully"
+  exit 1
+fi

go.mod

@@ -23,7 +23,6 @@ require (
 	golang.org/x/sys v0.0.0-20190919044723-0c1ff786ef13 // indirect
 	golang.org/x/time v0.0.0-20190308202827-9d24e82272b4
 	golang.org/x/tools v0.0.0-20190920225731-5eefd052ad72
-	golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7
 	gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127 // indirect
 	gotest.tools/gotestsum v0.3.5
 	mvdan.cc/sh v2.6.4+incompatible

internal/wsecho/cmd/wsecho/main.go

@@ -5,33 +5,15 @@ package wsecho
 import (
 	"context"
 	"io"
-	"log"
-	"net/http"
 	"time"
 
 	"nhooyr.io/websocket"
 )
 
-// Serve provides a streaming WebSocket echo server
-// for use in tests.
-func Serve(w http.ResponseWriter, r *http.Request) {
-	c, err := websocket.Accept(w, r, &websocket.AcceptOptions{
-		Subprotocols:       []string{"echo"},
-		InsecureSkipVerify: true,
-	})
-	if err != nil {
-		log.Printf("echo server: failed to accept: %+v", err)
-		return
-	}
-	defer c.Close(websocket.StatusInternalError, "")
-
-	Loop(r.Context(), c)
-}
-
 // Loop echos every msg received from c until an error
 // occurs or the context expires.
 // The read limit is set to 1 << 30.
-func Loop(ctx context.Context, c *websocket.Conn) {
+func Loop(ctx context.Context, c *websocket.Conn) error {
 	defer c.Close(websocket.StatusInternalError, "")
 
 	c.SetReadLimit(1 << 30)
@@ -67,7 +49,7 @@ func Loop(ctx context.Context, c *websocket.Conn) {
 	for {
 		err := echo()
 		if err != nil {
-			return
+			return err
 		}
 	}
 }

internal/wsecho/wsecho.go

@@ -41,7 +41,7 @@ func New(url string, protocols []string) (c WebSocket, err error) {
 		v: js.Global().Get("WebSocket").New(url, jsProtocols),
 	}
 
-	c.setBinaryType("arrayBuffer")
+	c.setBinaryType("arraybuffer")
 
 	c.Extensions = c.v.Get("extensions").String()
 	c.Protocol = c.v.Get("protocol").String()

internal/wsjs/wsjs.go

@@ -0,0 +1,43 @@
+// +build !js
+
+package main
+
+import (
+	"errors"
+	"fmt"
+	"log"
+	"net/http"
+	"net/http/httptest"
+	"os"
+	"runtime"
+	"strings"
+
+	"nhooyr.io/websocket"
+	"nhooyr.io/websocket/internal/wsecho"
+)
+
+func main() {
+	s := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		c, err := websocket.Accept(w, r, &websocket.AcceptOptions{
+			Subprotocols:       []string{"echo"},
+			InsecureSkipVerify: true,
+		})
+		if err != nil {
+			log.Fatalf("echo server: failed to accept: %+v", err)
+		}
+		defer c.Close(websocket.StatusInternalError, "")
+
+		err = wsecho.Loop(r.Context(), c)
+
+		var ce websocket.CloseError
+		if !errors.As(err, &ce) || ce.Code != websocket.StatusNormalClosure {
+			log.Fatalf("unexpected loop error: %+v", err)
+		}
+
+		os.Exit(0)
+	}))
+	wsURL := strings.Replace(s.URL, "http", "ws", 1)
+	fmt.Printf("%v\n", wsURL)
+
+	runtime.Goexit()
+}

internal/wsjstest/main.go

@@ -946,3 +946,7 @@ func (c *Conn) extractBufioWriterBuf(w io.Writer) {
 
 	c.bw.Reset(w)
 }
+
+func (c *netConn) netConnReader(ctx context.Context) (MessageType, io.Reader, error) {
+	return c.c.Reader(c.readContext)
+}