github.com/philippseith/signalr@v0.6.3/httpserver_test.go

package signalr

import (
	"bytes"
	"context"
	"encoding/json"
	"fmt"
	"io"
	"net"
	"net/http"
	"net/http/httptest"
	"net/url"
	"strconv"

	"strings"
	"time"

	"github.com/go-kit/log/level"
	. "github.com/onsi/ginkgo"
	. "github.com/onsi/gomega"
	"nhooyr.io/websocket"
)

type addHub struct {
	Hub
}

func (w *addHub) Add2(i int) int {
	return i + 2
}

func (w *addHub) Echo(s string) string {
	return s
}

var _ = Describe("HTTP server", func() {
	for i := 0; i < 3; i++ {
		var transport TransportType
		var transferFormat TransferFormatType
		switch i {
		case 0:
			transport = TransportWebSockets
			transferFormat = TransferFormatText
		case 1:
			transport = TransportWebSockets
			transferFormat = TransferFormatBinary
		case 2:
			transport = TransportServerSentEvents
			transferFormat = TransferFormatText
		}
		Context(fmt.Sprintf("%v %v", transport, transferFormat), func() {
			Context("A correct negotiation request is sent", func() {
				It(fmt.Sprintf("should send a correct negotiation response with support for %v with text protocol", transport), func(done Done) {
					// Start server
					server, err := NewServer(context.TODO(), SimpleHubFactory(&addHub{}), HTTPTransports(transport), testLoggerOption())
					Expect(err).NotTo(HaveOccurred())
					router := http.NewServeMux()
					server.MapHTTP(WithHTTPServeMux(router), "/hub")
					testServer := httptest.NewServer(router)
					url, _ := url.Parse(testServer.URL)
					port, _ := strconv.Atoi(url.Port())
					// Negotiate
					negResp := negotiateWebSocketTestServer(port)
					Expect(negResp["connectionId"]).NotTo(BeNil())
					Expect(negResp["availableTransports"]).To(BeAssignableToTypeOf([]interface{}{}))
					avt := negResp["availableTransports"].([]interface{})
					Expect(len(avt)).To(BeNumerically(">", 0))
					Expect(avt[0]).To(BeAssignableToTypeOf(map[string]interface{}{}))
					avtVal := avt[0].(map[string]interface{})
					Expect(avtVal["transport"]).To(Equal(string(transport)))
					Expect(avtVal["transferFormats"]).To(BeAssignableToTypeOf([]interface{}{}))
					tf := avtVal["transferFormats"].([]interface{})
					Expect(tf).To(ContainElement("Text"))
					if transport == TransportWebSockets {
						Expect(tf).To(ContainElement("Binary"))
					}
					testServer.Close()
					close(done)
				}, 2.0)
			})

			Context("A invalid negotiation request is sent", func() {
				It(fmt.Sprintf("should send a correct negotiation response with support for %v with text protocol", transport), func(done Done) {
					// Start server
					server, err := NewServer(context.TODO(), SimpleHubFactory(&addHub{}), HTTPTransports(transport), testLoggerOption())
					Expect(err).NotTo(HaveOccurred())
					router := http.NewServeMux()
					server.MapHTTP(WithHTTPServeMux(router), "/hub")
					testServer := httptest.NewServer(router)
					url, _ := url.Parse(testServer.URL)
					port, _ := strconv.Atoi(url.Port())
					waitForPort(port)
					// Negotiate the wrong way
					resp, err := http.Get(fmt.Sprintf("http://127.0.0.1:%v/hub/negotiate", port))
					Expect(err).To(BeNil())
					Expect(resp).NotTo(BeNil())
					Expect(resp.StatusCode).ToNot(Equal(200))
					testServer.Close()
					close(done)
				}, 2.0)
			})

			Context("Connection with client", func() {
				It("should successfully handle an Invoke call", func(done Done) {
					logger := &nonProtocolLogger{testLogger()}
					// Start server
					ctx, cancel := context.WithCancel(context.Background())
					server, err := NewServer(ctx,
						SimpleHubFactory(&addHub{}), HTTPTransports(transport),
						MaximumReceiveMessageSize(50000),
						Logger(logger, true))
					Expect(err).NotTo(HaveOccurred())
					router := http.NewServeMux()
					server.MapHTTP(WithHTTPServeMux(router), "/hub")
					testServer := httptest.NewServer(router)
					url, _ := url.Parse(testServer.URL)
					port, _ := strconv.Atoi(url.Port())
					waitForPort(port)
					// Try first connection
					conn, err := NewHTTPConnection(context.Background(), fmt.Sprintf("http://127.0.0.1:%v/hub", port))
					Expect(err).NotTo(HaveOccurred())
					client, err := NewClient(ctx,
						WithConnection(conn),
						MaximumReceiveMessageSize(60000),
						Logger(logger, true),
						TransferFormat(transferFormat))
					Expect(err).NotTo(HaveOccurred())
					Expect(client).NotTo(BeNil())
					client.Start()
					Expect(<-client.WaitForState(context.Background(), ClientConnected)).NotTo(HaveOccurred())
					result := <-client.Invoke("Add2", 1)
					Expect(result.Error).NotTo(HaveOccurred())
					Expect(result.Value).To(BeEquivalentTo(3))

					// Try second connection
					conn2, err := NewHTTPConnection(context.Background(), fmt.Sprintf("http://127.0.0.1:%v/hub", port))
					Expect(err).NotTo(HaveOccurred())
					client2, err := NewClient(ctx,
						WithConnection(conn2),
						Logger(logger, true),
						TransferFormat(transferFormat))
					Expect(err).NotTo(HaveOccurred())
					Expect(client2).NotTo(BeNil())
					client2.Start()
					Expect(<-client2.WaitForState(context.Background(), ClientConnected)).NotTo(HaveOccurred())
					result = <-client2.Invoke("Add2", 2)
					Expect(result.Error).NotTo(HaveOccurred())
					Expect(result.Value).To(BeEquivalentTo(4))
					// Huge message
					hugo := strings.Repeat("#", 2500)
					result = <-client.Invoke("Echo", hugo)
					Expect(result.Error).NotTo(HaveOccurred())
					s := result.Value.(string)
					Expect(s).To(Equal(hugo))
					cancel()
					go testServer.Close()
					close(done)
				}, 2.0)
			})
		})
	}
	Context("When no negotiation is send", func() {
		It("should serve websocket requests", func(done Done) {
			// Start server
			server, err := NewServer(context.TODO(), SimpleHubFactory(&addHub{}), HTTPTransports(TransportWebSockets), testLoggerOption())
			Expect(err).NotTo(HaveOccurred())
			router := http.NewServeMux()
			server.MapHTTP(WithHTTPServeMux(router), "/hub")
			testServer := httptest.NewServer(router)
			url, _ := url.Parse(testServer.URL)
			port, _ := strconv.Atoi(url.Port())
			waitForPort(port)
			handShakeAndCallWebSocketTestServer(port, "")
			testServer.Close()
			close(done)
		}, 5.0)
	})
})

var _ = Describe("HTTP client", func() {
	Context("WithHttpConnection", func() {
		It("should fallback to SSE (this can only be tested when httpConnection is tampered with)", func(done Done) {
			ctx, cancel := context.WithCancel(context.Background())
			defer cancel()

			server, err := NewServer(ctx, SimpleHubFactory(&addHub{}), HTTPTransports(TransportWebSockets, TransportServerSentEvents), testLoggerOption())
			Expect(err).NotTo(HaveOccurred())
			router := http.NewServeMux()
			server.MapHTTP(WithHTTPServeMux(router), "/hub")
			testServer := httptest.NewServer(router)
			url, _ := url.Parse(testServer.URL)
			port, _ := strconv.Atoi(url.Port())
			waitForPort(port)

			client, err := NewClient(ctx, WithHttpConnection(ctx, fmt.Sprintf("http://127.0.0.1:%v/hub", port)))
			Expect(err).NotTo(HaveOccurred())

			client.Start()
			Expect(<-client.WaitForState(context.Background(), ClientConnected)).NotTo(HaveOccurred())
			result := <-client.Invoke("Add2", 2)
			Expect(result.Error).NotTo(HaveOccurred())

			close(done)
		}, 2.0)
	})
})

type nonProtocolLogger struct {
	logger StructuredLogger
}

func (n *nonProtocolLogger) Log(keyVals ...interface{}) error {
	for _, kv := range keyVals {
		if kv == "protocol" {
			return nil
		}
	}
	return n.logger.Log(keyVals...)
}

func negotiateWebSocketTestServer(port int) map[string]interface{} {
	waitForPort(port)
	buf := bytes.Buffer{}
	resp, err := http.Post(fmt.Sprintf("http://127.0.0.1:%v/hub/negotiate", port), "text/plain;charset=UTF-8", &buf)
	Expect(err).To(BeNil())
	Expect(resp).ToNot(BeNil())
	defer func() {
		_ = resp.Body.Close()
	}()
	var body []byte
	body, err = io.ReadAll(resp.Body)
	Expect(err).To(BeNil())
	response := make(map[string]interface{})
	err = json.Unmarshal(body, &response)
	Expect(err).To(BeNil())
	return response
}

func handShakeAndCallWebSocketTestServer(port int, connectionID string) {
	waitForPort(port)
	logger := testLogger()
	protocol := jsonHubProtocol{}
	protocol.setDebugLogger(level.Debug(logger))
	var urlParam string
	if connectionID != "" {
		urlParam = fmt.Sprintf("?id=%v", connectionID)
	}
	ws, _, err := websocket.Dial(context.Background(), fmt.Sprintf("ws://127.0.0.1:%v/hub%v", port, urlParam), nil)
	Expect(err).To(BeNil())
	defer func() {
		_ = ws.Close(websocket.StatusNormalClosure, "")
	}()
	wsConn := newWebSocketConnection(context.TODO(), connectionID, ws)
	cliConn := newHubConnection(wsConn, &protocol, 1<<15, testLogger())
	_, _ = wsConn.Write(append([]byte(`{"protocol": "json","version": 1}`), 30))
	_, _ = wsConn.Write(append([]byte(`{"type":1,"invocationId":"666","target":"add2","arguments":[1]}`), 30))
	result := make(chan interface{})
	go func() {
		for recvResult := range cliConn.Receive() {
			if completionMessage, ok := recvResult.message.(completionMessage); ok {
				result <- completionMessage.Result
				return
			}
		}
	}()
	select {
	case r := <-result:
		var f float64
		Expect(protocol.UnmarshalArgument(r, &f)).NotTo(HaveOccurred())
		Expect(f).To(Equal(3.0))
	case <-time.After(1000 * time.Millisecond):
		Fail("timed out")
	}
}

func waitForPort(port int) {
	for {
		if _, err := net.Dial("tcp", fmt.Sprintf("127.0.0.1:%v", port)); err == nil {
			return
		}
		time.Sleep(100 * time.Millisecond)
	}
}