github.com/frankrap/okex-api@v1.0.4/futures_ws_test.go (about)

     1  package okex
     2  
     3  import (
     4  	"github.com/spf13/viper"
     5  	"log"
     6  	"testing"
     7  )
     8  
     9  func newFuturesWSForTest() *FuturesWS {
    10  	viper.SetConfigName("test_config")
    11  	viper.AddConfigPath("./conf")
    12  	err := viper.ReadInConfig()
    13  	if err != nil {
    14  		log.Panic(err)
    15  	}
    16  
    17  	accessKey := viper.GetString("access_key")
    18  	secretKey := viper.GetString("secret_key")
    19  	passphrase := viper.GetString("passphrase")
    20  	wsURL := "wss://real.okex.com:8443/ws/v3"
    21  
    22  	ws := NewFuturesWS(wsURL,
    23  		accessKey, secretKey, passphrase, true)
    24  	err = ws.SetProxy("socks5://127.0.0.1:1080")
    25  	//ws.SetProxy("http://127.0.0.1:1080")
    26  	if err != nil {
    27  		log.Fatal(err)
    28  	}
    29  	return ws
    30  }
    31  
    32  func TestFuturesWS_AllInOne(t *testing.T) {
    33  	ws := newFuturesWSForTest()
    34  	ws.SetTickerCallback(func(tickers []WSTicker) {
    35  		log.Printf("%#v", tickers)
    36  	})
    37  	ws.SetAccountCallback(func(accounts []WSAccount) {
    38  		log.Printf("%#v", accounts)
    39  	})
    40  	//ws.SubscribeTicker("ticker_1", "BTC-USD-200626")
    41  	//ws.SubscribeTrade("trade_1", "BTC-USD-200626")
    42  	//ws.SubscribeDepthL2Tbt("depthL2_1", "BTC-USD-200626")
    43  	//ws.SubscribeOrder("order_1", "BTC-USD-200626")
    44  	ws.SubscribePosition("position_1", "BTC-USD-200626")
    45  	ws.SubscribeAccount("account_1", "BTC") // BTC/BTC-USDT
    46  	ws.Start()
    47  
    48  	select {}
    49  }
    50  
    51  func TestFuturesWS_Depth20(t *testing.T) {
    52  	ws := newFuturesWSForTest()
    53  	ws.SetDepth20SnapshotCallback(func(ob *OrderBook) {
    54  		log.Printf("%#v", ob)
    55  	})
    56  	ws.SubscribeDepthL2Tbt("depthL2_1", "BTC-USD-200626")
    57  	ws.Start()
    58  
    59  	select {}
    60  }
    61  
    62  func TestFuturesWS_SubscribeOrder(t *testing.T) {
    63  	ws := newFuturesWSForTest()
    64  	ws.SetOrderCallback(func(orders []WSOrder) {
    65  		log.Printf("%#v", orders)
    66  	})
    67  	ws.SubscribeOrder("order_1", "BTC-USD-200626")
    68  	ws.Start()
    69  
    70  	select {}
    71  }