54 lines
1.1 KiB
Go
54 lines
1.1 KiB
Go
package ws
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/gorilla/websocket"
|
|
)
|
|
|
|
func TestWsConn(t *testing.T) {
|
|
initFn := func(ws *WSConn) error {
|
|
var p = map[string]interface{}{
|
|
"op": "subscribe",
|
|
"args": []interface{}{
|
|
map[string]interface{}{"channel": "trades", "instType": "SWAP", "instId": "BTC-USDT-SWAP"},
|
|
},
|
|
}
|
|
return ws.WriteMsg(p)
|
|
}
|
|
messageFn := func(msg []byte) error {
|
|
fmt.Println("msg:", string(msg))
|
|
return nil
|
|
}
|
|
conn, err := NewWSConn("wss://wsaws.okx.com:8443/ws/v5/public", initFn, messageFn)
|
|
if err != nil {
|
|
t.Fatal(err.Error())
|
|
}
|
|
time.Sleep(time.Second * 5)
|
|
conn.Close()
|
|
}
|
|
|
|
func TestWsPing(t *testing.T) {
|
|
initFn := func(ws *WSConn) error {
|
|
return nil
|
|
}
|
|
messageFn := func(msg []byte) error {
|
|
fmt.Println("msg:", string(msg))
|
|
return nil
|
|
}
|
|
pongFn := func(msg []byte) bool {
|
|
fmt.Println("recv pong msg:", string(msg))
|
|
return true
|
|
}
|
|
conn, err := NewWSConn("wss://wsaws.okx.com:8443/ws/v5/public", initFn, messageFn)
|
|
if err != nil {
|
|
t.Fatal(err.Error())
|
|
}
|
|
conn.SetPingPongFn(defaultPingFn, pongFn)
|
|
conn.ws.WriteMessage(websocket.TextMessage, []byte("ping"))
|
|
time.Sleep(time.Second * 15)
|
|
conn.Close()
|
|
}
|