bbgo/pkg/service/margin_test.go

53 lines
1.1 KiB
Go
Raw Normal View History

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)
}