bbgo_origin/pkg/service/withdraw_test.go

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 TestWithdrawService(t *testing.T) {
	db, err := prepareDB(t)
	if err != nil {
		t.Fatal(err)
	}

	defer db.Close()

	xdb := sqlx.NewDb(db.DB, "sqlite3")
	service := &WithdrawService{DB: xdb}

	err = service.Insert(types.Withdraw{
		Exchange:       types.ExchangeMax,
		Asset:          "BTC",
		Amount:         fixedpoint.NewFromFloat(0.0001),
		Address:        "test",
		TransactionID:  "01",
		TransactionFee: fixedpoint.NewFromFloat(0.0001),
		Network:        "omni",
		ApplyTime:      types.Time(time.Now()),
	})
	assert.NoError(t, err)

	withdraws, err := service.Query(types.ExchangeMax)
	assert.NoError(t, err)
	assert.NotEmpty(t, withdraws)
	assert.Equal(t, types.ExchangeMax, withdraws[0].Exchange)
}