mirror of
https://github.com/go-vikunja/vikunja.git
synced 2026-05-07 12:37:14 -05:00
9255fe07a9
Add the core WebSocket infrastructure: - Message type definitions for the wire protocol (subscribe, unsubscribe, auth, error, push events) - In-memory connection hub that tracks per-user connections and routes messages to subscribed clients - Connection wrapper with auth-after-connect flow: connections start unauthenticated, client sends JWT as first message, only then can subscribe to event topics Includes auth timeout (30s), shared cancellation context for read/write loops, hub map cleanup on last connection removal, and proper error delivery before closing on auth failure.
78 lines
2.6 KiB
Go
78 lines
2.6 KiB
Go
// Vikunja is a to-do list application to facilitate your life.
|
|
// Copyright 2018-present Vikunja and contributors. All rights reserved.
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Affero General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
package websocket
|
|
|
|
import (
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestIncomingAuthMessageDeserialization(t *testing.T) {
|
|
raw := `{"action":"auth","token":"eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9..."}`
|
|
var msg IncomingMessage
|
|
err := json.Unmarshal([]byte(raw), &msg)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, ActionAuth, msg.Action)
|
|
assert.Equal(t, "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9...", msg.Token)
|
|
}
|
|
|
|
func TestIncomingSubscribeMessageDeserialization(t *testing.T) {
|
|
raw := `{"action":"subscribe","event":"notification.created"}`
|
|
var msg IncomingMessage
|
|
err := json.Unmarshal([]byte(raw), &msg)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, ActionSubscribe, msg.Action)
|
|
assert.Equal(t, "notification.created", msg.Event)
|
|
}
|
|
|
|
func TestOutgoingEventSerialization(t *testing.T) {
|
|
msg := OutgoingMessage{
|
|
Event: "notification.created",
|
|
Data: map[string]string{"hello": "world"},
|
|
}
|
|
data, err := json.Marshal(msg)
|
|
require.NoError(t, err)
|
|
assert.Contains(t, string(data), `"event":"notification.created"`)
|
|
assert.NotContains(t, string(data), `"topic"`)
|
|
assert.Contains(t, string(data), `"hello":"world"`)
|
|
}
|
|
|
|
func TestOutgoingErrorSerialization(t *testing.T) {
|
|
msg := OutgoingMessage{
|
|
Error: "forbidden",
|
|
Event: "project.tasks",
|
|
}
|
|
data, err := json.Marshal(msg)
|
|
require.NoError(t, err)
|
|
assert.Contains(t, string(data), `"error":"forbidden"`)
|
|
assert.Contains(t, string(data), `"event":"project.tasks"`)
|
|
}
|
|
|
|
func TestOutgoingAuthSuccessSerialization(t *testing.T) {
|
|
msg := OutgoingMessage{
|
|
Action: ActionAuthSuccess,
|
|
Success: true,
|
|
}
|
|
data, err := json.Marshal(msg)
|
|
require.NoError(t, err)
|
|
assert.Contains(t, string(data), `"action":"auth.success"`)
|
|
assert.Contains(t, string(data), `"success":true`)
|
|
}
|