Skip to content

Commit 3e9c728

Browse files
committed
Add ping test
1 parent b3e9764 commit 3e9c728

File tree

1 file changed

+33
-0
lines changed

1 file changed

+33
-0
lines changed

websocket_test.go

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -376,6 +376,39 @@ func TestHandshake(t *testing.T) {
376376
return nil
377377
},
378378
},
379+
{
380+
name: "ping",
381+
server: func(w http.ResponseWriter, r *http.Request) error {
382+
c, err := websocket.Accept(w, r, websocket.AcceptOptions{})
383+
if err != nil {
384+
return err
385+
}
386+
defer c.Close(websocket.StatusInternalError, "")
387+
388+
err = c.Ping(r.Context())
389+
if err != nil {
390+
return err
391+
}
392+
393+
c.Close(websocket.StatusNormalClosure, "")
394+
return nil
395+
},
396+
client: func(ctx context.Context, u string) error {
397+
c, _, err := websocket.Dial(ctx, u, websocket.DialOptions{})
398+
if err != nil {
399+
return err
400+
}
401+
defer c.Close(websocket.StatusInternalError, "")
402+
403+
err = c.Ping(ctx)
404+
if err != nil {
405+
return err
406+
}
407+
408+
c.Close(websocket.StatusNormalClosure, "")
409+
return nil
410+
},
411+
},
379412
}
380413

381414
for _, tc := range testCases {

0 commit comments

Comments
 (0)