78 lines
2.2 KiB
Go
78 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"net"
|
|
"os"
|
|
"syscall"
|
|
"testing"
|
|
|
|
"github.com/gorilla/websocket"
|
|
)
|
|
|
|
func TestClassifyWSError(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
err error
|
|
want string
|
|
}{
|
|
{name: "none", want: "none"},
|
|
{name: "normal close", err: &websocket.CloseError{Code: websocket.CloseNormalClosure}, want: "close_normal"},
|
|
{name: "abnormal close", err: &websocket.CloseError{Code: websocket.CloseAbnormalClosure}, want: "close_abnormal"},
|
|
{name: "timeout", err: &net.OpError{Op: "write", Err: os.ErrDeadlineExceeded}, want: "timeout"},
|
|
{name: "reset", err: fmt.Errorf("write failed: %w", syscall.ECONNRESET), want: "connection_reset"},
|
|
{name: "broken pipe", err: fmt.Errorf("write failed: %w", syscall.EPIPE), want: "broken_pipe"},
|
|
{name: "closed", err: net.ErrClosed, want: "closed"},
|
|
{name: "eof", err: io.EOF, want: "eof"},
|
|
{name: "other", err: fmt.Errorf("unknown"), want: "other"},
|
|
}
|
|
|
|
for _, tc := range tests {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
if got := classifyWSError(tc.err); got != tc.want {
|
|
t.Fatalf("classifyWSError()=%q want=%q", got, tc.want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestShouldLogWSWriteError(t *testing.T) {
|
|
tests := []struct {
|
|
reason string
|
|
want bool
|
|
}{
|
|
{reason: "close_normal", want: false},
|
|
{reason: "close_going_away", want: false},
|
|
{reason: "closed", want: false},
|
|
{reason: "eof", want: false},
|
|
{reason: "broken_pipe", want: false},
|
|
{reason: "connection_reset", want: false},
|
|
{reason: "timeout", want: true},
|
|
{reason: "close_abnormal", want: true},
|
|
{reason: "other", want: true},
|
|
}
|
|
|
|
for _, tc := range tests {
|
|
t.Run(tc.reason, func(t *testing.T) {
|
|
if got := shouldLogWSWriteError(tc.reason); got != tc.want {
|
|
t.Fatalf("shouldLogWSWriteError(%q)=%v want %v", tc.reason, got, tc.want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestWebSocketOutboundQueueTiming(t *testing.T) {
|
|
sess := &Session{proto: WEBSOCK, send: make(chan any, 2)}
|
|
payload := []byte("message")
|
|
if !sess.queueOutBytes(payload) {
|
|
t.Fatal("queueOutBytes rejected payload")
|
|
}
|
|
|
|
queued := <-sess.send
|
|
sess.observeOutboundDequeue(false)
|
|
if string(queued.([]byte)) != string(payload) {
|
|
t.Fatalf("payload=%q want=%q", queued, payload)
|
|
}
|
|
}
|