websocket_test.go

package examples

import (
	"net/http"
	"testing"

	"github.com/gavv/httpexpect"
	"github.com/gorilla/websocket"
)

func wsHttpHandlerTester(t *testing.T) *httpexpect.Expect {
	return httpexpect.WithConfig(httpexpect.Config{
		BaseURL:         "ws://example.com",
		WebsocketDialer: httpexpect.NewWebsocketDialer(http.HandlerFunc(WsHttpHandler)),
		Reporter:        httpexpect.NewAssertReporter(t),
		Printers: []httpexpect.Printer{
			httpexpect.NewDebugPrinter(t, true),
		},
	})
}

func TestWsHttpHandlerText(t *testing.T) {
	e := wsHttpHandlerTester(t)

	ws := e.GET("/path").WithWebsocketUpgrade().
		Expect().
		Status(http.StatusSwitchingProtocols).
		Websocket()
	defer ws.Disconnect()

	ws.WriteText("hi").
		Expect().
		TextMessage().Body().Equal("hi")
}

func TestWsHttpHandlerJSON(t *testing.T) {
	e := wsHttpHandlerTester(t)

	ws := e.GET("/path").WithWebsocketUpgrade().
		Expect().
		Status(http.StatusSwitchingProtocols).
		Websocket()
	defer ws.Disconnect()

	ws.WriteJSON(struct {
		Message string `json:"message"`
	}{"hi"}).
		Expect().
		TextMessage().JSON().Object().ValueEqual("message", "hi")
}

func TestWsHttpHandlerClose(t *testing.T) {
	e := wsHttpHandlerTester(t)

	ws := e.GET("/path").WithWebsocketUpgrade().
		Expect().
		Status(http.StatusSwitchingProtocols).
		Websocket()
	defer ws.Disconnect()

	ws.CloseWithText("Namárië...", websocket.CloseGoingAway).
		Expect().
		CloseMessage().NoContent()
}

func wsFastHandlerTester(t *testing.T) *httpexpect.Expect {
	return httpexpect.WithConfig(httpexpect.Config{
		BaseURL:         "http://example.com",
		WebsocketDialer: httpexpect.NewFastWebsocketDialer(WsFastHandler),
		Reporter:        httpexpect.NewAssertReporter(t),
		Printers: []httpexpect.Printer{
			httpexpect.NewDebugPrinter(t, true),
		},
	})
}

func TestWsFastHandlerText(t *testing.T) {
	e := wsFastHandlerTester(t)

	ws := e.GET("/path").WithWebsocketUpgrade().
		Expect().
		Status(http.StatusSwitchingProtocols).
		Websocket()
	defer ws.Disconnect()

	ws.WriteText("hi").
		Expect().
		TextMessage().Body().Equal("hi")
}

func TestWsFastHandlerJSON(t *testing.T) {
	e := wsFastHandlerTester(t)

	ws := e.GET("/path").WithWebsocketUpgrade().
		Expect().
		Status(http.StatusSwitchingProtocols).
		Websocket()
	defer ws.Disconnect()

	ws.WriteJSON(struct {
		Message string `json:"message"`
	}{"hi"}).
		Expect().
		TextMessage().JSON().Object().ValueEqual("message", "hi")
}

func TestWsFastHandlerClose(t *testing.T) {
	e := wsFastHandlerTester(t)

	ws := e.GET("/path").WithWebsocketUpgrade().
		Expect().
		Status(http.StatusSwitchingProtocols).
		Websocket()
	defer ws.Disconnect()

	ws.CloseWithText("Namárië...", websocket.CloseGoingAway).
		Expect().
		CloseMessage().NoContent()
}