53 lines
1.1 KiB
Go
53 lines
1.1 KiB
Go
|
package service
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/jmoiron/sqlx"
|
||
|
|
"github.com/sirupsen/logrus"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
|
||
|
|
"git.qtrade.icu/lychiyu/bbgo/pkg/exchange/binance"
|
||
|
|
"git.qtrade.icu/lychiyu/bbgo/pkg/testutil"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestMarginService(t *testing.T) {
|
||
|
|
key, secret, ok := testutil.IntegrationTestConfigured(t, "BINANCE")
|
||
|
|
if !ok {
|
||
|
|
t.SkipNow()
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
ex := binance.New(key, secret)
|
||
|
|
ex.MarginSettings.IsMargin = true
|
||
|
|
ex.MarginSettings.IsIsolatedMargin = true
|
||
|
|
ex.MarginSettings.IsolatedMarginSymbol = "DOTUSDT"
|
||
|
|
|
||
|
|
logrus.SetLevel(logrus.ErrorLevel)
|
||
|
|
db, err := prepareDB(t)
|
||
|
|
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
t.Fail()
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
defer db.Close()
|
||
|
|
|
||
|
|
ctx := context.Background()
|
||
|
|
|
||
|
|
dbx := sqlx.NewDb(db.DB, "sqlite3")
|
||
|
|
service := &MarginService{DB: dbx}
|
||
|
|
|
||
|
|
logrus.SetLevel(logrus.DebugLevel)
|
||
|
|
err = service.Sync(ctx, ex, "USDT", time.Date(2022, time.February, 1, 0, 0, 0, 0, time.UTC))
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
// sync second time to ensure that we can query records
|
||
|
|
err = service.Sync(ctx, ex, "USDT", time.Date(2022, time.February, 1, 0, 0, 0, 0, time.UTC))
|
||
|
|
assert.NoError(t, err)
|
||
|
|
}
|