42 lines
900 B
Go
42 lines
900 B
Go
|
package tradingutil
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
|
||
|
|
"git.qtrade.icu/lychiyu/bbgo/pkg/fixedpoint"
|
||
|
|
"git.qtrade.icu/lychiyu/bbgo/pkg/types"
|
||
|
|
)
|
||
|
|
|
||
|
|
var number = fixedpoint.MustNewFromString
|
||
|
|
|
||
|
|
func Test_CollectTradeFee(t *testing.T) {
|
||
|
|
trades := []types.Trade{
|
||
|
|
{
|
||
|
|
ID: 1,
|
||
|
|
Price: number("21000"),
|
||
|
|
Quantity: number("0.001"),
|
||
|
|
Symbol: "BTCUSDT",
|
||
|
|
Side: types.SideTypeBuy,
|
||
|
|
Fee: number("0.00001"),
|
||
|
|
FeeCurrency: "BTC",
|
||
|
|
FeeDiscounted: false,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
ID: 2,
|
||
|
|
Price: number("21200"),
|
||
|
|
Quantity: number("0.001"),
|
||
|
|
Symbol: "BTCUSDT",
|
||
|
|
Side: types.SideTypeBuy,
|
||
|
|
Fee: number("0.00002"),
|
||
|
|
FeeCurrency: "BTC",
|
||
|
|
FeeDiscounted: false,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
fees := CollectTradeFee(trades)
|
||
|
|
assert.NotNil(t, fees)
|
||
|
|
assert.Equal(t, number("0.00003"), fees["BTC"])
|
||
|
|
}
|