wasm: Add dial timeout test

nhooyr · Jan 9, 2021 · 65dfbdd · 65dfbdd
1 parent 6840778
commit 65dfbdd
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/ws_js_test.go b/ws_js_test.go
@@ -36,3 +36,19 @@ func TestWasm(t *testing.T) {
 	err = c.Close(websocket.StatusNormalClosure, "")
 	assert.Success(t, err)
 }
+
+func TestWasmDialTimeout(t *testing.T) {
+	t.Parallel()
+
+	ctx, cancel := context.WithTimeout(context.Background(), time.Millisecond)
+	defer cancel()
+
+	beforeDial := time.Now()
+	_, _, err := websocket.Dial(ctx, "ws://example.com:9893", &websocket.DialOptions{
+		Subprotocols: []string{"echo"},
+	})
+	assert.Error(t, err)
+	if time.Since(beforeDial) >= time.Second {
+		t.Fatal("wasm context dial timeout is not working", time.Since(beforeDial))
+	}
+}