Function newTestWSPair

internal/ws/hub_test.go:16–48

newTestWSPair creates a connected client/server websocket pair.

(t *testing.T)

// newTestWSPair creates a connected client/server websocket pair.
func newTestWSPair(t *testing.T) (*websocket.Conn, *websocket.Conn) {
	t.Helper()
	var serverConn *websocket.Conn
	ready := make(chan struct{})

	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		var err error
		serverConn, err = websocket.Accept(w, r, nil)
		if err != nil {
			t.Fatalf("accept: %v", err)
		}
		close(ready)
		// Keep open until test closes it
		select {}
	}))
	t.Cleanup(srv.Close)

	ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
	defer cancel()

	clientConn, _, err := websocket.Dial(ctx, "ws"+strings.TrimPrefix(srv.URL, "http"), nil)
	if err != nil {
		t.Fatalf("dial: %v", err)
	}

	select {
	case <-ready:
	case <-ctx.Done():
		t.Fatal("timeout waiting for server accept")
	}

	return clientConn, serverConn
}

func TestHubRegisterAndIsConnected(t *testing.T) {
	hub := NewHub()