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

     1  package okex
     2  
     3  import (
     4  	"github.com/spf13/viper"
     5  	"log"
     6  	"testing"
     7  )
     8  
     9  func newSwapWSForTest() *SwapWS {
    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  	ws := NewSwapWS(wsURL, accessKey, secretKey, passphrase, true)
    22  	err = ws.SetProxy("socks5://127.0.0.1:1080")
    23  	////ws.SetProxy("http://127.0.0.1:1080")
    24  	if err != nil {
    25  		log.Fatal(err)
    26  	}
    27  	return ws
    28  }
    29  
    30  func TestSwapWS_AllInOne(t *testing.T) {
    31  	ws := newSwapWSForTest()
    32  	ws.SetTickerCallback(func(tickers []WSTicker) {
    33  		log.Printf("%#v", tickers)
    34  	})
    35  	ws.SetAccountCallback(func(accounts []WSAccount) {
    36  		log.Printf("%#v", accounts)
    37  	})
    38  	//ws.SubscribeTicker("ticker_1", "BTC-USD-SWAP")
    39  	//ws.SubscribeTrade("trade_1", "BTC-USD-SWAP")
    40  	//ws.SubscribeDepthL2Tbt("depthL2_1", "BTC-USD-SWAP")
    41  	//ws.SubscribeOrder("order_1", "BTC-USD-SWAP")
    42  	ws.SubscribePosition("position_1", "BTC-USD-SWAP")
    43  	ws.SubscribeAccount("account_1", "BTC-USD-SWAP")
    44  	ws.Start()
    45  
    46  	select {}
    47  }