mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-10 09:11:55 +00:00
clean up
This commit is contained in:
parent
f3a7428b48
commit
f1cce3e123
|
@ -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").
|
||||
|
|
|
@ -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)
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue
Block a user