snapshot of https://github.com/cortezaproject/corteza.git (2024.9.x, 2026-09-18; blobs > 131072B omitted)

server/pkg/websocket/server_test.go

@@ -0,0 +1,64 @@
+package websocket
+
+import (
+	"bytes"
+	"testing"
+
+	"github.com/cortezaproject/corteza/server/pkg/options"
+	"github.com/stretchr/testify/require"
+	"go.uber.org/zap"
+)
+
+func TestWebsocketSend_NoSessions(t *testing.T) {
+	var (
+		req = require.New(t)
+		ws  = Server(zap.NewNop(), options.WebsocketOpt{}, nil)
+	)
+
+	req.NoError(ws.Send("msg", "msg"))
+	req.NoError(ws.Send("msg", "msg", 1))
+	req.NoError(ws.Send("msg", "msg", 1, 2))
+	req.NoError(ws.Send("msg", "msg", 1, 2, 3))
+}
+
+func TestWebsocketSend_ExistingSessions(t *testing.T) {
+	var (
+		req = require.New(t)
+		ws  = Server(zap.NewNop(), options.WebsocketOpt{}, nil)
+
+		s1User uint64 = 100
+		s1ID   uint64 = 101
+
+		s2User uint64 = 200
+		s2ID   uint64 = 201
+
+		s1 = &bytes.Buffer{}
+		s2 = &bytes.Buffer{}
+	)
+
+	ws.storeSession(s1, s1User, s1ID)
+	ws.storeSession(s2, s2User, s2ID)
+
+	req.Empty(s1)
+	req.Empty(s2)
+
+	req.NoError(ws.Send("msg", "msg", 0))
+	req.Empty(s1)
+	req.Empty(s2)
+
+	req.NoError(ws.Send("msg", "msg1", s1User))
+	req.Equal(s1.String(), `{"@type":"msg","@value":"msg1"}`)
+	req.Equal(s2.String(), "")
+
+	req.NoError(ws.Send("msg", "msg2", s2User))
+	req.Equal(s1.String(), `{"@type":"msg","@value":"msg1"}`)
+	req.Equal(s2.String(), `{"@type":"msg","@value":"msg2"}`)
+
+	req.NoError(ws.Send("both", "msg3", s1User, s2User))
+	req.Equal(s1.String(), `{"@type":"msg","@value":"msg1"}{"@type":"both","@value":"msg3"}`)
+	req.Equal(s2.String(), `{"@type":"msg","@value":"msg2"}{"@type":"both","@value":"msg3"}`)
+
+	req.NoError(ws.Send("all", "msg4"))
+	req.Equal(s1.String(), `{"@type":"msg","@value":"msg1"}{"@type":"both","@value":"msg3"}{"@type":"all","@value":"msg4"}`)
+	req.Equal(s2.String(), `{"@type":"msg","@value":"msg2"}{"@type":"both","@value":"msg3"}{"@type":"all","@value":"msg4"}`)
+}