1
2
3
4
5 package websocket
6
7 import (
8 "bytes"
9 "io"
10 "strings"
11 "testing"
12 )
13
14 func TestJoinMessages(t *testing.T) {
15 messages := []string{"a", "bc", "def", "ghij", "klmno", "0", "12", "345", "6789"}
16 for _, readChunk := range []int{1, 2, 3, 4, 5, 6, 7} {
17 for _, term := range []string{"", ","} {
18 var connBuf bytes.Buffer
19 wc := newTestConn(nil, &connBuf, true)
20 rc := newTestConn(&connBuf, nil, false)
21 for _, m := range messages {
22 wc.WriteMessage(BinaryMessage, []byte(m))
23 }
24
25 var result bytes.Buffer
26 _, err := io.CopyBuffer(&result, JoinMessages(rc, term), make([]byte, readChunk))
27 if IsUnexpectedCloseError(err, CloseAbnormalClosure) {
28 t.Errorf("readChunk=%d, term=%q: unexpected error %v", readChunk, term, err)
29 }
30 want := strings.Join(messages, term) + term
31 if result.String() != want {
32 t.Errorf("readChunk=%d, term=%q, got %q, want %q", readChunk, term, result.String(), want)
33 }
34 }
35 }
36 }
37
View as plain text