lychiyu/bbgo_origin
1
0
Fork 0
mirror of https://github.com/c9s/bbgo.git synced 2024-11-10 09:11:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

bbgo: add two test cases for reflect

Browse Source
This commit is contained in:
c9s 2022-09-11 03:13:10 +08:00
parent 278aa026f2
commit 8961c940b2
No known key found for this signature in database
GPG Key ID: 7385E7E464CB0A54
2 changed files with 52 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
pkg/bbgo/interact.go
View File

@ -99,6 +99,24 @@ func filterStrategies(exchangeStrategies map[string]SingleExchangeStrategy, filt
return retStrategies, nil
}
func hasTypeField(obj interface{}, typ interface{}) bool {
targetType := reflect.TypeOf(typ)
found := false
_ = dynamic.IterateFields(obj, func(ft reflect.StructField, fv reflect.Value) error {
if fv.Type() == targetType {
found = true
}
return nil
})
return found
}
func testInterface(obj interface{}, checkType interface{}) bool {
rt := reflect.TypeOf(checkType).Elem()
return reflect.TypeOf(obj).Implements(rt)
}
func filterStrategiesByInterface(exchangeStrategies map[string]SingleExchangeStrategy, checkInterface interface{}) (map[string]SingleExchangeStrategy, error) {
rt := reflect.TypeOf(checkInterface).Elem()
return filterStrategies(exchangeStrategies, func(s SingleExchangeStrategy) bool {
@ -210,9 +228,11 @@ func (it *CoreInteraction) Commands(i *interact.Interact) {
})
i.PrivateCommand("/resetposition", "Reset position", func(reply interact.Reply) error {
// it.trader.exchangeStrategies
// send symbol options
if strategies, err := filterStrategiesByInterface(it.exchangeStrategies, (*PositionResetter)(nil)); err == nil && len(strategies) > 0 {
strategies, err := filterStrategies(it.exchangeStrategies, func(s SingleExchangeStrategy) bool {
return testInterface(s, (*PositionResetter)(nil)) || hasTypeField(s, &types.Position{})
})
if err == nil && len(strategies) > 0 {
reply.AddMultipleButtons(generateStrategyButtonsForm(strategies))
reply.Message("Please choose one strategy")
} else {
@ -481,7 +501,7 @@ func (it *CoreInteraction) Commands(i *interact.Interact) {
i.PrivateCommand("/modifyposition", "Modify Strategy Position", func(reply interact.Reply) error {
// it.trader.exchangeStrategies
// send symbol options
if strategies, err := filterStrategiesByField(it.exchangeStrategies, "Position", reflect.TypeOf(types.NewPosition("", "", ""))); err == nil && len(strategies) > 0 {
if strategies, err := filterStrategiesByField(it.exchangeStrategies, "Position", reflect.TypeOf(&types.Position{})); err == nil && len(strategies) > 0 {
reply.AddMultipleButtons(generateStrategyButtonsForm(strategies))
reply.Message("Please choose one strategy")
} else {

29
pkg/bbgo/interact_test.go
View File

@ -6,20 +6,28 @@ import (
"testing"
"github.com/stretchr/testify/assert"
"github.com/c9s/bbgo/pkg/fixedpoint"
"github.com/c9s/bbgo/pkg/types"
)
type myStrategy struct {
Symbol string `json:"symbol"`
Position *types.Position
}
func (m myStrategy) ID() string {
func (m *myStrategy) ID() string {
return "mystrategy"
}
func (m myStrategy) InstanceID() string {
func (m *myStrategy) InstanceID() string {
return fmt.Sprintf("%s:%s", m.ID(), m.Symbol)
}
func (m *myStrategy) ClosePosition(ctx context.Context, percentage fixedpoint.Value) error {
return nil
}
func (m *myStrategy) Run(ctx context.Context, orderExecutor OrderExecutor, session *ExchangeSession) error {
return nil
}
@ -31,3 +39,20 @@ func Test_getStrategySignature(t *testing.T) {
assert.NoError(t, err)
assert.Equal(t, "mystrategy:BTCUSDT", signature)
}
func Test_hasTypeField(t *testing.T) {
s := &myStrategy{
Symbol: "BTCUSDT",
}
ok := hasTypeField(s, &types.Position{})
assert.True(t, ok)
}
func Test_testInterface(t *testing.T) {
s := &myStrategy{
Symbol: "BTCUSDT",
}
ok := testInterface(s, (*PositionCloser)(nil))
assert.True(t, ok)
}