This commit is contained in:
c9s 2022-06-08 17:33:52 +08:00
parent f3a7428b48
commit f1cce3e123
No known key found for this signature in database
GPG Key ID: 7385E7E464CB0A54
2 changed files with 0 additions and 45 deletions

View File

@ -94,13 +94,6 @@ func (s *DepositService) scanRows(rows *sqlx.Rows) (deposits []types.Deposit, er
return deposits, rows.Err()
}
func (s *DepositService) Insert(deposit types.Deposit) error {
sql := `INSERT INTO deposits (exchange, asset, address, amount, txn_id, time)
VALUES (:exchange, :asset, :address, :amount, :txn_id, :time)`
_, err := s.DB.NamedExec(sql, deposit)
return err
}
func SelectLastDeposits(ex types.ExchangeName, limit uint64) sq.SelectBuilder {
return sq.Select("*").
From("deposits").

View File

@ -1,39 +1 @@
package service
import (
"testing"
"time"
"github.com/jmoiron/sqlx"
"github.com/stretchr/testify/assert"
"github.com/c9s/bbgo/pkg/types"
"github.com/c9s/bbgo/pkg/fixedpoint"
)
func TestDepositService(t *testing.T) {
db, err := prepareDB(t)
if err != nil {
t.Fatal(err)
}
defer db.Close()
xdb := sqlx.NewDb(db.DB, "sqlite3")
service := &DepositService{DB: xdb}
err = service.Insert(types.Deposit{
Exchange: types.ExchangeMax,
Time: types.Time(time.Now()),
Amount: fixedpoint.NewFromFloat(0.001),
Asset: "BTC",
Address: "test",
TransactionID: "02",
Status: types.DepositSuccess,
})
assert.NoError(t, err)
deposits, err := service.Query(types.ExchangeMax)
assert.NoError(t, err)
assert.NotEmpty(t, deposits)
}