mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-25 00:05:15 +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()
|
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 {
|
func SelectLastDeposits(ex types.ExchangeName, limit uint64) sq.SelectBuilder {
|
||||||
return sq.Select("*").
|
return sq.Select("*").
|
||||||
From("deposits").
|
From("deposits").
|
||||||
|
|
|
@ -1,39 +1 @@
|
||||||
package service
|
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