mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-13 15:09:42 +00:00
Added websocket tests
This commit is contained in:
Ryan O'Hara-Reid
232
exchanges/bitfinex/bitfinex_websocket_test.go
Normal file
232
exchanges/bitfinex/bitfinex_websocket_test.go
Normal file
@@ -0,0 +1,232 @@
|
||||
package bitfinex
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"testing"
|
||||
|
||||
"github.com/gorilla/websocket"
|
||||
"github.com/thrasher-/gocryptotrader/common"
|
||||
)
|
||||
|
||||
func TestWebsocketPingHandler(t *testing.T) {
|
||||
wsPingHandler := Bitfinex{}
|
||||
var Dialer websocket.Dialer
|
||||
var err error
|
||||
|
||||
wsPingHandler.WebsocketConn, _, err = Dialer.Dial(BITFINEX_WEBSOCKET, http.Header{})
|
||||
if err != nil {
|
||||
t.Errorf("Test Failed - Bitfinex dialer error: %s", err)
|
||||
}
|
||||
err = wsPingHandler.WebsocketPingHandler()
|
||||
if err != nil {
|
||||
t.Errorf("Test Failed - Bitfinex WebsocketPingHandler() error: %s", err)
|
||||
}
|
||||
err = wsPingHandler.WebsocketConn.Close()
|
||||
if err != nil {
|
||||
t.Errorf("Test Failed - Bitfinex websocketConn.Close() error: %s", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestWebsocketSend(t *testing.T) {
|
||||
wsSend := Bitfinex{}
|
||||
var Dialer websocket.Dialer
|
||||
var err error
|
||||
|
||||
type WebsocketHandshake struct {
|
||||
Event string `json:"event"`
|
||||
Code int64 `json:"code"`
|
||||
Version float64 `json:"version"`
|
||||
}
|
||||
|
||||
request, dodgyrequest := make(map[string]string), make(map[string]string)
|
||||
request["event"] = "ping"
|
||||
dodgyrequest["dodgyEvent"] = "didgereedodge"
|
||||
|
||||
hs := WebsocketHandshake{}
|
||||
|
||||
for {
|
||||
wsSend.WebsocketConn, _, err = Dialer.Dial(BITFINEX_WEBSOCKET, http.Header{})
|
||||
if err != nil {
|
||||
if err.Error() == "websocket: close 1006 (abnormal closure): unexpected EOF" {
|
||||
err = wsSend.WebsocketConn.Close()
|
||||
if err != nil {
|
||||
t.Errorf("Test Failed - Bitfinex websocketConn.Close() error: %s", err)
|
||||
}
|
||||
continue
|
||||
} else {
|
||||
t.Errorf("Test Failed - Bitfinex websocket connection error: %s", err)
|
||||
}
|
||||
}
|
||||
mType, resp, err := wsSend.WebsocketConn.ReadMessage()
|
||||
if err != nil {
|
||||
t.Errorf("Test Failed - Bitfinex websocketconn.ReadMessage() error: %s", err)
|
||||
}
|
||||
if mType != websocket.TextMessage {
|
||||
t.Errorf("Test Failed - Bitfinex websocketconn.ReadMessage() mType error: %d", mType)
|
||||
}
|
||||
err = common.JSONDecode(resp, &hs)
|
||||
if err != nil {
|
||||
t.Errorf("Test Failed - Bitfinex JSONDecode error: %s", err)
|
||||
}
|
||||
if hs.Code != 0 {
|
||||
t.Errorf("Test Failed - Bitfinex hs.Code incorrect: %d", hs.Code)
|
||||
}
|
||||
if hs.Event != "info" {
|
||||
t.Errorf("Test Failed - Bitfinex hs.Event incorrect: %s", hs.Event)
|
||||
}
|
||||
if hs.Version != 1.1 {
|
||||
t.Errorf("Test Failed - Bitfinex hs.Version incorrect: %f", hs.Version)
|
||||
}
|
||||
|
||||
err = wsSend.WebsocketSend(request)
|
||||
if err != nil {
|
||||
t.Errorf("Test Failed - Bitfinex websocket send error: %s", err)
|
||||
}
|
||||
mType, resp, err = wsSend.WebsocketConn.ReadMessage()
|
||||
if err != nil {
|
||||
if err.Error() == "websocket: close 1006 (abnormal closure): unexpected EOF" {
|
||||
err = wsSend.WebsocketConn.Close()
|
||||
if err != nil {
|
||||
t.Errorf("Test Failed - Bitfinex websocketConn.Close() error: %s", err)
|
||||
}
|
||||
continue
|
||||
} else {
|
||||
t.Errorf("Test Failed - Bitfinex websocketConn.ReadMessage() error: %s", err)
|
||||
}
|
||||
}
|
||||
if mType != websocket.TextMessage {
|
||||
t.Errorf("Test Failed - Bitfinex websocketconn.ReadMessage() mType error: %d", mType)
|
||||
}
|
||||
err = common.JSONDecode(resp, &hs)
|
||||
if err != nil {
|
||||
t.Errorf("Test Failed - Bitfinex JSONDecode error: %s", err)
|
||||
}
|
||||
if hs.Code != 0 {
|
||||
t.Errorf("Test Failed - Bitfinex hs.Code incorrect: %d", hs.Code)
|
||||
}
|
||||
if hs.Event != "pong" {
|
||||
t.Errorf("Test Failed - Bitfinex hs.Event incorrect: %s", hs.Event)
|
||||
}
|
||||
if hs.Version != 1.1 {
|
||||
t.Errorf("Test Failed - Bitfinex hs.Version incorrect: %f", hs.Version)
|
||||
}
|
||||
|
||||
err = wsSend.WebsocketSend(dodgyrequest)
|
||||
if err != nil {
|
||||
t.Errorf("Test Failed - Bitfinex websocket send error: %s", err)
|
||||
}
|
||||
mType, resp, err = wsSend.WebsocketConn.ReadMessage()
|
||||
if err != nil {
|
||||
if err.Error() == "websocket: close 1006 (abnormal closure): unexpected EOF" {
|
||||
err = wsSend.WebsocketConn.Close()
|
||||
if err != nil {
|
||||
t.Errorf("Test Failed - Bitfinex websocketConn.Close() error: %s", err)
|
||||
}
|
||||
continue
|
||||
} else {
|
||||
t.Errorf("Test Failed - Bitfinex websocketConn.ReadMessage() error: %s", err)
|
||||
}
|
||||
}
|
||||
if mType != websocket.TextMessage {
|
||||
t.Errorf("Test Failed - Bitfinex websocketconn.ReadMessage() mType error: %d", mType)
|
||||
}
|
||||
err = common.JSONDecode(resp, &hs)
|
||||
if err != nil {
|
||||
t.Errorf("Test Failed - Bitfinex JSONDecode error: %s", err)
|
||||
}
|
||||
if hs.Code != 10000 {
|
||||
t.Errorf("Test Failed - Bitfinex hs.Code incorrect: %d", hs.Code)
|
||||
}
|
||||
if hs.Event != "error" {
|
||||
t.Errorf("Test Failed - Bitfinex hs.Event incorrect: %s", hs.Event)
|
||||
}
|
||||
if hs.Version != 1.1 {
|
||||
t.Errorf("Test Failed - Bitfinex hs.Version incorrect: %f", hs.Version)
|
||||
}
|
||||
|
||||
err = wsSend.WebsocketConn.Close()
|
||||
if err != nil {
|
||||
t.Errorf("Test Failed - Bitfinex websocketConn.Close() error: %s", err)
|
||||
}
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
func TestWebsocketSubscribe(t *testing.T) {
|
||||
websocketSubcribe := Bitfinex{}
|
||||
var Dialer websocket.Dialer
|
||||
var err error
|
||||
params := make(map[string]string)
|
||||
params["pair"] = "BTCUSD"
|
||||
|
||||
websocketSubcribe.WebsocketConn, _, err = Dialer.Dial(BITFINEX_WEBSOCKET, http.Header{})
|
||||
if err != nil {
|
||||
t.Errorf("Test Failed - Bitfinex Dialer error: %s", err)
|
||||
}
|
||||
err = websocketSubcribe.WebsocketSubscribe("ticker", params)
|
||||
if err != nil {
|
||||
t.Errorf("Test Failed - Bitfinex WebsocketSubscribe() error: %s", err)
|
||||
}
|
||||
|
||||
err = websocketSubcribe.WebsocketConn.Close()
|
||||
if err != nil {
|
||||
t.Errorf("Test Failed - Bitfinex websocketConn.Close() error: %s", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestWebsocketSendAuth(t *testing.T) {
|
||||
wsSendAuth := Bitfinex{}
|
||||
var Dialer websocket.Dialer
|
||||
var err error
|
||||
|
||||
wsSendAuth.WebsocketConn, _, err = Dialer.Dial(BITFINEX_WEBSOCKET, http.Header{})
|
||||
if err != nil {
|
||||
t.Errorf("Test Failed - Bitfinex Dialer error: %s", err)
|
||||
}
|
||||
err = wsSendAuth.WebsocketSendAuth()
|
||||
if err != nil {
|
||||
t.Errorf("Test Failed - Bitfinex WebsocketSendAuth() error: %s", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestWebsocketSendUnauth(t *testing.T) {
|
||||
wsSendUnauth := Bitfinex{}
|
||||
var Dialer websocket.Dialer
|
||||
var err error
|
||||
|
||||
wsSendUnauth.WebsocketConn, _, err = Dialer.Dial(BITFINEX_WEBSOCKET, http.Header{})
|
||||
if err != nil {
|
||||
t.Errorf("Test Failed - Bitfinex Dialer error: %s", err)
|
||||
}
|
||||
err = wsSendUnauth.WebsocketSendUnauth()
|
||||
if err != nil {
|
||||
t.Errorf("Test Failed - Bitfinex WebsocketSendAuth() error: %s", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestWebsocketAddSubscriptionChannel(t *testing.T) {
|
||||
wsAddSubscriptionChannel := Bitfinex{}
|
||||
wsAddSubscriptionChannel.SetDefaults()
|
||||
var Dialer websocket.Dialer
|
||||
var err error
|
||||
|
||||
wsAddSubscriptionChannel.WebsocketConn, _, err = Dialer.Dial(BITFINEX_WEBSOCKET, http.Header{})
|
||||
if err != nil {
|
||||
t.Errorf("Test Failed - Bitfinex Dialer error: %s", err)
|
||||
}
|
||||
|
||||
wsAddSubscriptionChannel.WebsocketAddSubscriptionChannel(1337, "ticker", "BTCUSD")
|
||||
if len(wsAddSubscriptionChannel.WebsocketSubdChannels) == 0 {
|
||||
t.Errorf("Test Failed - Bitfinex WebsocketAddSubscriptionChannel() error: %s", err)
|
||||
}
|
||||
if wsAddSubscriptionChannel.WebsocketSubdChannels[1337].Channel != "ticker" {
|
||||
t.Errorf("Test Failed - Bitfinex WebsocketAddSubscriptionChannel() error: %s", err)
|
||||
}
|
||||
if wsAddSubscriptionChannel.WebsocketSubdChannels[1337].Pair != "BTCUSD" {
|
||||
t.Errorf("Test Failed - Bitfinex WebsocketAddSubscriptionChannel() error: %s", err)
|
||||
}
|
||||
}
|
||||
|
||||
// func TestWebsocketClient(t *testing.T) {
|
||||
//
|
||||
// }
|
||||
Reference in New Issue
Block a user