mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-29 23:16:51 +00:00
f5faca2eb2
* linter: Enable gofumpt and run against codebase * Address shazbert's nits * gofumpt: Fix issues after rebase
67 lines
1.8 KiB
Go
67 lines
1.8 KiB
Go
package strategies
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/thrasher-corp/gocryptotrader/backtester/data"
|
|
"github.com/thrasher-corp/gocryptotrader/backtester/eventhandlers/portfolio"
|
|
"github.com/thrasher-corp/gocryptotrader/backtester/eventhandlers/strategies"
|
|
"github.com/thrasher-corp/gocryptotrader/backtester/eventhandlers/strategies/base"
|
|
"github.com/thrasher-corp/gocryptotrader/backtester/eventhandlers/strategies/dollarcostaverage"
|
|
"github.com/thrasher-corp/gocryptotrader/backtester/eventtypes/signal"
|
|
"github.com/thrasher-corp/gocryptotrader/backtester/funding"
|
|
)
|
|
|
|
func TestAddStrategies(t *testing.T) {
|
|
t.Parallel()
|
|
err := addStrategies(nil)
|
|
if !errors.Is(err, errNoStrategies) {
|
|
t.Error(err)
|
|
}
|
|
|
|
err = addStrategies([]strategies.Handler{&dollarcostaverage.Strategy{}})
|
|
if !errors.Is(err, strategies.ErrStrategyAlreadyExists) {
|
|
t.Error(err)
|
|
}
|
|
|
|
err = addStrategies([]strategies.Handler{&CustomStrategy{}})
|
|
if !errors.Is(err, nil) {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
type CustomStrategy struct {
|
|
base.Strategy
|
|
}
|
|
|
|
func (s *CustomStrategy) Name() string {
|
|
return "custom-strategy"
|
|
}
|
|
|
|
func (s *CustomStrategy) Description() string {
|
|
return "this is a demonstration of loading strategies via custom plugins"
|
|
}
|
|
|
|
func (s *CustomStrategy) SupportsSimultaneousProcessing() bool {
|
|
return true
|
|
}
|
|
|
|
func (s *CustomStrategy) OnSignal(d data.Handler, _ funding.IFundingTransferer, _ portfolio.Handler) (signal.Event, error) {
|
|
return s.createSignal(d)
|
|
}
|
|
|
|
func (s *CustomStrategy) OnSimultaneousSignals(_ []data.Handler, _ funding.IFundingTransferer, _ portfolio.Handler) ([]signal.Event, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
func (s *CustomStrategy) createSignal(_ data.Handler) (*signal.Signal, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
func (s *CustomStrategy) SetCustomSettings(map[string]interface{}) error {
|
|
return nil
|
|
}
|
|
|
|
func (s *CustomStrategy) SetDefaults() {}
|