diff --git a/pkg/service/account.go b/pkg/service/account.go index 030f62d92..5244cc45a 100644 --- a/pkg/service/account.go +++ b/pkg/service/account.go @@ -38,7 +38,7 @@ func (s *AccountService) InsertAsset(time time.Time, session string, name types. borrowed, net_asset, price_in_usd) - values (?,?,?,?,?,?,?,?,?,?,?); + values (?,?,?,?,?,?,?,?,?,?,?,?,?); `, session, name, account, time, v.Currency, v.InUSD, v.InBTC, v.Total, v.Available, v.Locked, v.Borrowed, v.NetAsset, v.PriceInUSD) err = multierr.Append(err, _err) // successful request diff --git a/pkg/service/account_test.go b/pkg/service/account_test.go new file mode 100644 index 000000000..d6433ca7a --- /dev/null +++ b/pkg/service/account_test.go @@ -0,0 +1,41 @@ +package service + +import ( + "testing" + "time" + + "github.com/jmoiron/sqlx" + "github.com/stretchr/testify/assert" + + "github.com/c9s/bbgo/pkg/fixedpoint" + "github.com/c9s/bbgo/pkg/types" +) + +func TestAccountService(t *testing.T) { + db, err := prepareDB(t) + if err != nil { + t.Fatal(err) + } + + defer db.Close() + + xdb := sqlx.NewDb(db.DB, "sqlite3") + service := &AccountService{DB: xdb} + + t1 := time.Now() + err = service.InsertAsset(t1, "binance", types.ExchangeBinance, "main", types.AssetMap{ + "BTC": types.Asset{ + Currency: "BTC", + Total: fixedpoint.MustNewFromString("1.0"), + InUSD: fixedpoint.MustNewFromString("10.0"), + InBTC: fixedpoint.MustNewFromString("0.0001"), + Time: t1, + Locked: fixedpoint.MustNewFromString("0"), + Available: fixedpoint.MustNewFromString("1.0"), + Borrowed: fixedpoint.MustNewFromString("0"), + NetAsset: fixedpoint.MustNewFromString("1"), + PriceInUSD: fixedpoint.MustNewFromString("44870"), + }, + }) + assert.NoError(t, err) +}