mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-24 07:45:15 +00:00
52 lines
1.0 KiB
Go
52 lines
1.0 KiB
Go
|
package okex
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"os"
|
||
|
|
"strconv"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
|
||
|
|
"github.com/c9s/bbgo/pkg/testutil"
|
||
|
|
"github.com/c9s/bbgo/pkg/types"
|
||
|
|
)
|
||
|
|
|
||
|
|
func getTestClientOrSkip(t *testing.T) *Stream {
|
||
|
|
if b, _ := strconv.ParseBool(os.Getenv("CI")); b {
|
||
|
|
t.Skip("skip test for CI")
|
||
|
|
}
|
||
|
|
|
||
|
|
key, secret, passphrase, ok := testutil.IntegrationTestWithPassphraseConfigured(t, "OKEX")
|
||
|
|
if !ok {
|
||
|
|
t.Skip("OKEX_* env vars are not configured")
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
exchange := New(key, secret, passphrase)
|
||
|
|
return NewStream(exchange.client)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestStream(t *testing.T) {
|
||
|
|
t.Skip()
|
||
|
|
s := getTestClientOrSkip(t)
|
||
|
|
|
||
|
|
t.Run("book test", func(t *testing.T) {
|
||
|
|
s.Subscribe(types.BookChannel, "BTCUSDT", types.SubscribeOptions{
|
||
|
|
Depth: types.DepthLevel50,
|
||
|
|
})
|
||
|
|
s.SetPublicOnly()
|
||
|
|
err := s.Connect(context.Background())
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
s.OnBookSnapshot(func(book types.SliceOrderBook) {
|
||
|
|
t.Log("got snapshot", book)
|
||
|
|
})
|
||
|
|
s.OnBookUpdate(func(book types.SliceOrderBook) {
|
||
|
|
t.Log("got update", book)
|
||
|
|
})
|
||
|
|
c := make(chan struct{})
|
||
|
|
<-c
|
||
|
|
})
|
||
|
|
}
|