Newer
Older
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
"net/http"
"testing"
"time"
"gfx.cafe/open/jrpc"
"gfx.cafe/open/jrpc/contrib/codecs"
"gfx.cafe/open/jrpc/contrib/jmux"
"gfx.cafe/open/jrpc/pkg/jsonrpc"
"gfx.cafe/open/jrpc/pkg/server"
)
func TestWrapClient(t *testing.T) {
engine := NewEngine()
r := jmux.NewRouter()
r.Use(engine.Middleware())
r.HandleFunc("echo", func(w jsonrpc.ResponseWriter, r *jsonrpc.Request) {
_ = w.Send(r.Params, nil)
})
// extremely fast subscription to fill buffers to get a higher chance that we receive another message while trying
// to unsubscribe
r.HandleFunc("test/subscribe", func(w jsonrpc.ResponseWriter, r *jsonrpc.Request) {
notifier, ok := NotifierFromContext(r.Context())
if !ok {
_ = w.Send(nil, ErrNotificationsUnsupported)
return
}
go func() {
idx := 0
for {
select {
case <-r.Context().Done():
return
case <-notifier.Err():
return
default:
}
_ = notifier.Notify(idx)
idx += 1
}
}()
})
srv := server.NewServer(r)
handler := codecs.WebsocketHandler(srv, []string{"*"})
httpSrv := http.Server{
Addr: ":8855",
Handler: handler,
}
listener, err := net.Listen("tcp", ":8855")
if err != nil {
t.Error(err)
return
}
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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
t.Error(err)
return
}
}()
cl, err := UpgradeConn(jrpc.Dial("ws://localhost:8855"))
if err != nil {
t.Error(err)
return
}
for i := 0; i < 10; i++ {
var res string
if err = cl.Do(context.Background(), &res, "echo", "test"); err != nil {
t.Error(err)
return
}
if res != "test" {
t.Errorf(`expected "test" but got %#v`, res)
return
}
ch := make(chan int, 1)
sub, err := cl.Subscribe(context.Background(), "test", ch, nil)
if err != nil {
t.Error(err)
return
}
go func() {
time.Sleep(2 * time.Second)
_ = sub.Unsubscribe()
}()
func() {
for {
select {
case err, ok := <-sub.Err():
if ok {
t.Errorf("sub errored: %v", err)
}
return
case v := <-ch:
log.Printf("%v", v)
}
}
}()
}
}