mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-10 01:01:56 +00:00
pkg/core: add more tests for convert manager
This commit is contained in:
parent
4fefbaf0e7
commit
bc24e8dcaa
|
@ -9,6 +9,200 @@ import (
|
|||
"github.com/c9s/bbgo/pkg/types"
|
||||
)
|
||||
|
||||
func TestInitializeConverter_ValidSymbolConverter(t *testing.T) {
|
||||
setting := ConverterSetting{
|
||||
SymbolConverter: &SymbolConverter{
|
||||
FromSymbol: "MAXEXCHANGEUSDT",
|
||||
ToSymbol: "MAXUSDT",
|
||||
},
|
||||
}
|
||||
converter, err := setting.InitializeConverter()
|
||||
assert.NoError(t, err)
|
||||
assert.NotNil(t, converter)
|
||||
}
|
||||
|
||||
func TestInitializeConverter_ValidCurrencyConverter(t *testing.T) {
|
||||
setting := ConverterSetting{
|
||||
CurrencyConverter: &CurrencyConverter{
|
||||
FromCurrency: "MAXEXCHANGE",
|
||||
ToCurrency: "MAX",
|
||||
},
|
||||
}
|
||||
converter, err := setting.InitializeConverter()
|
||||
assert.NoError(t, err)
|
||||
assert.NotNil(t, converter)
|
||||
}
|
||||
|
||||
func TestInitializeConverter_NoConverter(t *testing.T) {
|
||||
setting := ConverterSetting{}
|
||||
converter, err := setting.InitializeConverter()
|
||||
assert.NoError(t, err)
|
||||
assert.Nil(t, converter)
|
||||
}
|
||||
|
||||
func TestInitialize_ValidConverters(t *testing.T) {
|
||||
manager := ConverterManager{
|
||||
ConverterSettings: []ConverterSetting{
|
||||
{SymbolConverter: &SymbolConverter{
|
||||
FromSymbol: "MAXEXCHANGEUSDT",
|
||||
ToSymbol: "MAXUSDT",
|
||||
}},
|
||||
{CurrencyConverter: &CurrencyConverter{
|
||||
FromCurrency: "MAXEXCHANGE",
|
||||
ToCurrency: "MAX",
|
||||
}},
|
||||
},
|
||||
}
|
||||
err := manager.Initialize()
|
||||
assert.NoError(t, err)
|
||||
assert.Equal(t, 2, len(manager.converters))
|
||||
}
|
||||
|
||||
func TestInitialize_NoConverters(t *testing.T) {
|
||||
manager := ConverterManager{}
|
||||
err := manager.Initialize()
|
||||
assert.NoError(t, err)
|
||||
assert.Equal(t, 0, len(manager.converters))
|
||||
}
|
||||
|
||||
func TestConvertOrder_WithConverters(t *testing.T) {
|
||||
manager := ConverterManager{
|
||||
ConverterSettings: []ConverterSetting{
|
||||
{SymbolConverter: &SymbolConverter{
|
||||
FromSymbol: "MAXEXCHANGEUSDT",
|
||||
ToSymbol: "MAXUSDT",
|
||||
}},
|
||||
{CurrencyConverter: &CurrencyConverter{
|
||||
FromCurrency: "MAXEXCHANGE",
|
||||
ToCurrency: "MAX",
|
||||
}},
|
||||
},
|
||||
}
|
||||
|
||||
order := types.Order{
|
||||
SubmitOrder: types.SubmitOrder{
|
||||
Symbol: "MAXEXCHANGEUSDT",
|
||||
Market: types.Market{
|
||||
Symbol: "MAXEXCHANGEUSDT",
|
||||
QuoteCurrency: "USDT",
|
||||
BaseCurrency: "MAXEXCHANGE",
|
||||
},
|
||||
},
|
||||
}
|
||||
err := manager.Initialize()
|
||||
assert.NoError(t, err)
|
||||
convertedOrder := manager.ConvertOrder(order)
|
||||
assert.Equal(t, "MAXUSDT", convertedOrder.Symbol)
|
||||
assert.Equal(t, "MAX", convertedOrder.Market.BaseCurrency)
|
||||
assert.Equal(t, "USDT", convertedOrder.Market.QuoteCurrency)
|
||||
assert.Equal(t, "MAXUSDT", convertedOrder.Market.Symbol)
|
||||
}
|
||||
|
||||
func TestConvertOrder_NoConverters(t *testing.T) {
|
||||
manager := ConverterManager{}
|
||||
order := types.Order{}
|
||||
err := manager.Initialize()
|
||||
assert.NoError(t, err)
|
||||
convertedOrder := manager.ConvertOrder(order)
|
||||
assert.Equal(t, order, convertedOrder)
|
||||
}
|
||||
|
||||
func TestConvertTrade_WithConverters(t *testing.T) {
|
||||
manager := ConverterManager{}
|
||||
converter := &CurrencyConverter{
|
||||
FromCurrency: "MAXEXCHANGE",
|
||||
ToCurrency: "MAX",
|
||||
}
|
||||
err := manager.Initialize()
|
||||
assert.NoError(t, err)
|
||||
manager.AddConverter(converter)
|
||||
|
||||
trade := types.Trade{}
|
||||
convertedTrade := manager.ConvertTrade(trade)
|
||||
assert.Equal(t, trade, convertedTrade)
|
||||
}
|
||||
|
||||
func TestConvertTrade_NoConverters(t *testing.T) {
|
||||
manager := ConverterManager{}
|
||||
trade := types.Trade{}
|
||||
err := manager.Initialize()
|
||||
assert.NoError(t, err)
|
||||
convertedTrade := manager.ConvertTrade(trade)
|
||||
assert.Equal(t, trade, convertedTrade)
|
||||
}
|
||||
|
||||
func TestConvertKLine_WithConverters(t *testing.T) {
|
||||
manager := ConverterManager{}
|
||||
converter := &CurrencyConverter{
|
||||
FromCurrency: "MAXEXCHANGE",
|
||||
ToCurrency: "MAX",
|
||||
}
|
||||
err := manager.Initialize()
|
||||
assert.NoError(t, err)
|
||||
manager.AddConverter(converter)
|
||||
|
||||
kline := types.KLine{}
|
||||
convertedKline := manager.ConvertKLine(kline)
|
||||
assert.Equal(t, kline, convertedKline)
|
||||
}
|
||||
|
||||
func TestConvertKLine_NoConverters(t *testing.T) {
|
||||
manager := ConverterManager{}
|
||||
kline := types.KLine{}
|
||||
err := manager.Initialize()
|
||||
assert.NoError(t, err)
|
||||
convertedKline := manager.ConvertKLine(kline)
|
||||
assert.Equal(t, kline, convertedKline)
|
||||
}
|
||||
|
||||
func TestConvertMarket_WithConverters(t *testing.T) {
|
||||
manager := ConverterManager{}
|
||||
converter := &CurrencyConverter{
|
||||
FromCurrency: "MAXEXCHANGE",
|
||||
ToCurrency: "MAX",
|
||||
}
|
||||
err := manager.Initialize()
|
||||
assert.NoError(t, err)
|
||||
manager.AddConverter(converter)
|
||||
|
||||
market := types.Market{}
|
||||
convertedMarket := manager.ConvertMarket(market)
|
||||
assert.Equal(t, market, convertedMarket)
|
||||
}
|
||||
|
||||
func TestConvertMarket_NoConverters(t *testing.T) {
|
||||
manager := ConverterManager{}
|
||||
market := types.Market{}
|
||||
err := manager.Initialize()
|
||||
assert.NoError(t, err)
|
||||
convertedMarket := manager.ConvertMarket(market)
|
||||
assert.Equal(t, market, convertedMarket)
|
||||
}
|
||||
|
||||
func TestConvertBalance_WithConverters(t *testing.T) {
|
||||
manager := ConverterManager{}
|
||||
converter := &CurrencyConverter{
|
||||
FromCurrency: "MAXEXCHANGE",
|
||||
ToCurrency: "MAX",
|
||||
}
|
||||
err := manager.Initialize()
|
||||
assert.NoError(t, err)
|
||||
manager.AddConverter(converter)
|
||||
|
||||
balance := types.Balance{}
|
||||
convertedBalance := manager.ConvertBalance(balance)
|
||||
assert.Equal(t, balance, convertedBalance)
|
||||
}
|
||||
|
||||
func TestConvertBalance_NoConverters(t *testing.T) {
|
||||
manager := ConverterManager{}
|
||||
balance := types.Balance{}
|
||||
err := manager.Initialize()
|
||||
assert.NoError(t, err)
|
||||
convertedBalance := manager.ConvertBalance(balance)
|
||||
assert.Equal(t, balance, convertedBalance)
|
||||
}
|
||||
|
||||
func TestTradeCollector_NilConvertManager(t *testing.T) {
|
||||
symbol := "BTCUSDT"
|
||||
position := types.NewPosition(symbol, "BTC", "USDT")
|
||||
|
|
Loading…
Reference in New Issue
Block a user