1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
package sockjs
import (
"net/http"
"net/http/httptest"
"runtime"
"testing"
"time"
)
func TestHandler_EventSource(t *testing.T) {
rw := httptest.NewRecorder()
req, _ := http.NewRequest("POST", "/server/session/eventsource", nil)
h := newTestHandler()
h.options.ResponseLimit = 1024
go func() {
var sess *session
for exists := false; !exists; {
runtime.Gosched()
h.sessionsMux.Lock()
sess, exists = h.sessions["session"]
h.sessionsMux.Unlock()
}
for exists := false; !exists; {
runtime.Gosched()
sess.RLock()
exists = sess.recv != nil
sess.RUnlock()
}
sess.RLock()
sess.recv.close()
sess.RUnlock()
}()
h.eventSource(rw, req)
contentType := rw.Header().Get("content-type")
expected := "text/event-stream; charset=UTF-8"
if contentType != expected {
t.Errorf("Unexpected content type, got '%s', extected '%s'", contentType, expected)
}
if rw.Code != http.StatusOK {
t.Errorf("Unexpected response code, got '%d', expected '%d'", rw.Code, http.StatusOK)
}
if rw.Body.String() != "\r\ndata: o\r\n\r\n" {
t.Errorf("Event stream prelude, got '%s'", rw.Body)
}
}
func TestHandler_EventSourceMultipleConnections(t *testing.T) {
h := newTestHandler()
h.options.ResponseLimit = 1024
rw := httptest.NewRecorder()
req, _ := http.NewRequest("POST", "/server/sess/eventsource", nil)
go func() {
rw := &ClosableRecorder{httptest.NewRecorder(), nil}
h.eventSource(rw, req)
if rw.Body.String() != "\r\ndata: c[2010,\"Another connection still open\"]\r\n\r\n" {
t.Errorf("wrong, got '%v'", rw.Body)
}
h.sessionsMux.Lock()
sess := h.sessions["sess"]
sess.close()
h.sessionsMux.Unlock()
}()
h.eventSource(rw, req)
}
func TestHandler_EventSourceConnectionInterrupted(t *testing.T) {
h := newTestHandler()
sess := newTestSession()
sess.state = SessionActive
h.sessions["session"] = sess
req, _ := http.NewRequest("POST", "/server/session/eventsource", nil)
rw := newClosableRecorder()
close(rw.closeNotifCh)
h.eventSource(rw, req)
select {
case <-sess.closeCh:
case <-time.After(1 * time.Second):
t.Errorf("session close channel should be closed")
}
sess.Lock()
if sess.state != SessionClosed {
t.Errorf("Session should be closed")
}
}
|
