mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-13 15:09:42 +00:00
bea16af380
* golangci-lint: Enable usetesting and unused linters * tests: Improve assertions in various test cases for clarity and accuracy * tests: Enhance error assertions in TestExecuteStrategyFromFile for improved clarity * tests: Update assertions for improved clarity and accuracy * tests: Replace assert with require for task count checks * config/versions/v7: Replace context.Background() with t.Context() * Bithumb: Centralise and consoliate testPair, relax UpdateTickers check with some glorious Doom Eternal music * Bithumb: Use UpdatePairsOnce and update remaining pair string * Bithumb: Add UpdatePairsOnce to TestUpdateTickers
321 lines
9.2 KiB
Go
321 lines
9.2 KiB
Go
package engine
|
|
|
|
import (
|
|
"errors"
|
|
"sync/atomic"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/thrasher-corp/gocryptotrader/currency"
|
|
exchange "github.com/thrasher-corp/gocryptotrader/exchanges"
|
|
"github.com/thrasher-corp/gocryptotrader/exchanges/asset"
|
|
"github.com/thrasher-corp/gocryptotrader/exchanges/orderbook"
|
|
"github.com/thrasher-corp/gocryptotrader/exchanges/ticker"
|
|
)
|
|
|
|
func TestSetupEventManager(t *testing.T) {
|
|
t.Parallel()
|
|
_, err := setupEventManager(nil, nil, 0, false)
|
|
if !errors.Is(err, errNilComManager) {
|
|
t.Errorf("error '%v', expected '%v'", err, errNilComManager)
|
|
}
|
|
|
|
_, err = setupEventManager(&CommunicationManager{}, nil, 0, false)
|
|
if !errors.Is(err, errNilExchangeManager) {
|
|
t.Errorf("error '%v', expected '%v'", err, errNilExchangeManager)
|
|
}
|
|
|
|
m, err := setupEventManager(&CommunicationManager{}, &ExchangeManager{}, 0, false)
|
|
if !errors.Is(err, nil) {
|
|
t.Fatalf("error '%v', expected '%v'", err, nil)
|
|
}
|
|
if m == nil { //nolint:staticcheck,nolintlint // SA5011 Ignore the nil warnings
|
|
t.Fatal("expected manager")
|
|
}
|
|
if m.sleepDelay == 0 { //nolint:staticcheck,nolintlint // SA5011 Ignore the nil warnings
|
|
t.Error("expected default set")
|
|
}
|
|
}
|
|
|
|
func TestEventManagerStart(t *testing.T) {
|
|
m, err := setupEventManager(&CommunicationManager{}, &ExchangeManager{}, 0, false)
|
|
if !errors.Is(err, nil) {
|
|
t.Errorf("error '%v', expected '%v'", err, nil)
|
|
}
|
|
err = m.Start()
|
|
if !errors.Is(err, nil) {
|
|
t.Errorf("error '%v', expected '%v'", err, nil)
|
|
}
|
|
|
|
err = m.Start()
|
|
if !errors.Is(err, ErrSubSystemAlreadyStarted) {
|
|
t.Errorf("error '%v', expected '%v'", err, ErrSubSystemAlreadyStarted)
|
|
}
|
|
|
|
m = nil
|
|
err = m.Start()
|
|
if !errors.Is(err, ErrNilSubsystem) {
|
|
t.Errorf("error '%v', expected '%v'", err, ErrNilSubsystem)
|
|
}
|
|
}
|
|
|
|
func TestEventManagerIsRunning(t *testing.T) {
|
|
t.Parallel()
|
|
m, err := setupEventManager(&CommunicationManager{}, &ExchangeManager{}, 0, false)
|
|
if !errors.Is(err, nil) {
|
|
t.Errorf("error '%v', expected '%v'", err, nil)
|
|
}
|
|
err = m.Start()
|
|
if !errors.Is(err, nil) {
|
|
t.Errorf("error '%v', expected '%v'", err, nil)
|
|
}
|
|
if !m.IsRunning() {
|
|
t.Error("expected true")
|
|
}
|
|
atomic.StoreInt32(&m.started, 0)
|
|
if m.IsRunning() {
|
|
t.Error("expected false")
|
|
}
|
|
m = nil
|
|
if m.IsRunning() {
|
|
t.Error("expected false")
|
|
}
|
|
}
|
|
|
|
func TestEventManagerStop(t *testing.T) {
|
|
t.Parallel()
|
|
m, err := setupEventManager(&CommunicationManager{}, &ExchangeManager{}, 0, false)
|
|
if !errors.Is(err, nil) {
|
|
t.Errorf("error '%v', expected '%v'", err, nil)
|
|
}
|
|
err = m.Start()
|
|
if !errors.Is(err, nil) {
|
|
t.Errorf("error '%v', expected '%v'", err, nil)
|
|
}
|
|
err = m.Stop()
|
|
if !errors.Is(err, nil) {
|
|
t.Errorf("error '%v', expected '%v'", err, nil)
|
|
}
|
|
err = m.Stop()
|
|
if !errors.Is(err, ErrSubSystemNotStarted) {
|
|
t.Errorf("error '%v', expected '%v'", err, ErrSubSystemNotStarted)
|
|
}
|
|
m = nil
|
|
err = m.Stop()
|
|
if !errors.Is(err, ErrNilSubsystem) {
|
|
t.Errorf("error '%v', expected '%v'", err, ErrNilSubsystem)
|
|
}
|
|
}
|
|
|
|
func TestEventManagerAdd(t *testing.T) {
|
|
t.Parallel()
|
|
em := NewExchangeManager()
|
|
m, err := setupEventManager(&CommunicationManager{}, em, 0, false)
|
|
if !errors.Is(err, nil) {
|
|
t.Errorf("error '%v', expected '%v'", err, nil)
|
|
}
|
|
_, err = m.Add("", "", EventConditionParams{}, currency.NewPair(currency.BTC, currency.USDC), asset.Spot, "")
|
|
if !errors.Is(err, ErrSubSystemNotStarted) {
|
|
t.Errorf("error '%v', expected '%v'", err, ErrSubSystemNotStarted)
|
|
}
|
|
err = m.Start()
|
|
if !errors.Is(err, nil) {
|
|
t.Errorf("error '%v', expected '%v'", err, nil)
|
|
}
|
|
_, err = m.Add("", "", EventConditionParams{}, currency.NewPair(currency.BTC, currency.USDC), asset.Spot, "")
|
|
if !errors.Is(err, errExchangeDisabled) {
|
|
t.Errorf("error '%v', expected '%v'", err, errExchangeDisabled)
|
|
}
|
|
exch, err := em.NewExchangeByName(testExchange)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
exch.SetDefaults()
|
|
err = em.Add(exch)
|
|
if !errors.Is(err, nil) {
|
|
t.Fatalf("received: '%v' but expected: '%v'", err, nil)
|
|
}
|
|
_, err = m.Add(testExchange, "", EventConditionParams{}, currency.NewPair(currency.BTC, currency.USDC), asset.Spot, "")
|
|
if !errors.Is(err, errInvalidItem) {
|
|
t.Errorf("error '%v', expected '%v'", err, errInvalidItem)
|
|
}
|
|
|
|
cond := EventConditionParams{
|
|
Condition: ConditionGreaterThan,
|
|
Price: 1337,
|
|
OrderbookAmount: 1337,
|
|
}
|
|
_, err = m.Add(testExchange, ItemPrice, cond, currency.NewPair(currency.BTC, currency.USDC), asset.Spot, "")
|
|
if !errors.Is(err, errInvalidAction) {
|
|
t.Errorf("error '%v', expected '%v'", err, errInvalidAction)
|
|
}
|
|
|
|
_, err = m.Add(testExchange, ItemPrice, cond, currency.NewPair(currency.BTC, currency.USDC), asset.Spot, ActionTest)
|
|
if !errors.Is(err, nil) {
|
|
t.Errorf("error '%v', expected '%v'", err, nil)
|
|
}
|
|
|
|
action := ActionSMSNotify + "," + ActionTest
|
|
_, err = m.Add(testExchange, ItemPrice, cond, currency.NewPair(currency.BTC, currency.USDC), asset.Spot, action)
|
|
if !errors.Is(err, nil) {
|
|
t.Errorf("error '%v', expected '%v'", err, nil)
|
|
}
|
|
}
|
|
|
|
func TestEventManagerRemove(t *testing.T) {
|
|
t.Parallel()
|
|
em := NewExchangeManager()
|
|
m, err := setupEventManager(&CommunicationManager{}, em, 0, false)
|
|
if !errors.Is(err, nil) {
|
|
t.Errorf("error '%v', expected '%v'", err, nil)
|
|
}
|
|
if m.Remove(0) {
|
|
t.Error("expected false")
|
|
}
|
|
err = m.Start()
|
|
if !errors.Is(err, nil) {
|
|
t.Errorf("error '%v', expected '%v'", err, nil)
|
|
}
|
|
if m.Remove(0) {
|
|
t.Error("expected false")
|
|
}
|
|
action := ActionSMSNotify + "," + ActionTest
|
|
cond := EventConditionParams{
|
|
Condition: ConditionGreaterThan,
|
|
Price: 1337,
|
|
OrderbookAmount: 1337,
|
|
}
|
|
exch, err := em.NewExchangeByName(testExchange)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
exch.SetDefaults()
|
|
err = em.Add(exch)
|
|
if !errors.Is(err, nil) {
|
|
t.Fatalf("received: '%v' but expected: '%v'", err, nil)
|
|
}
|
|
id, err := m.Add(testExchange, ItemPrice, cond, currency.NewPair(currency.BTC, currency.USDC), asset.Spot, action)
|
|
if !errors.Is(err, nil) {
|
|
t.Errorf("error '%v', expected '%v'", err, nil)
|
|
}
|
|
|
|
if !m.Remove(id) {
|
|
t.Error("expected true")
|
|
}
|
|
}
|
|
|
|
func TestGetEventCounter(t *testing.T) {
|
|
t.Parallel()
|
|
em := NewExchangeManager()
|
|
m, err := setupEventManager(&CommunicationManager{}, em, 0, false)
|
|
if !errors.Is(err, nil) {
|
|
t.Errorf("error '%v', expected '%v'", err, nil)
|
|
}
|
|
total, executed := m.getEventCounter()
|
|
if total != 0 && executed != 0 {
|
|
t.Error("expected 0")
|
|
}
|
|
err = m.Start()
|
|
if !errors.Is(err, nil) {
|
|
t.Errorf("error '%v', expected '%v'", err, nil)
|
|
}
|
|
total, executed = m.getEventCounter()
|
|
if total != 0 && executed != 0 {
|
|
t.Error("expected 0")
|
|
}
|
|
action := ActionSMSNotify + "," + ActionTest
|
|
cond := EventConditionParams{
|
|
Condition: ConditionGreaterThan,
|
|
Price: 1337,
|
|
OrderbookAmount: 1337,
|
|
}
|
|
exch, err := em.NewExchangeByName(testExchange)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
exch.SetDefaults()
|
|
err = em.Add(exch)
|
|
if !errors.Is(err, nil) {
|
|
t.Fatalf("received: '%v' but expected: '%v'", err, nil)
|
|
}
|
|
_, err = m.Add(testExchange, ItemPrice, cond, currency.NewPair(currency.BTC, currency.USDC), asset.Spot, action)
|
|
if !errors.Is(err, nil) {
|
|
t.Errorf("error '%v', expected '%v'", err, nil)
|
|
}
|
|
|
|
total, _ = m.getEventCounter()
|
|
if total == 0 {
|
|
t.Error("expected 1")
|
|
}
|
|
}
|
|
|
|
func TestCheckEventCondition(t *testing.T) {
|
|
em := NewExchangeManager()
|
|
m, err := setupEventManager(&CommunicationManager{}, em, 0, false)
|
|
require.NoError(t, err, "setupEventManager must not error")
|
|
|
|
m.m.Lock()
|
|
err = m.checkEventCondition(nil)
|
|
assert.ErrorIs(t, err, ErrSubSystemNotStarted)
|
|
m.m.Unlock()
|
|
|
|
require.NoError(t, m.Start(), "Start must not error")
|
|
|
|
m.m.Lock()
|
|
err = m.checkEventCondition(nil)
|
|
assert.ErrorIs(t, err, errNilEvent)
|
|
m.m.Unlock()
|
|
|
|
action := ActionSMSNotify + "," + ActionTest
|
|
cond := EventConditionParams{
|
|
Condition: ConditionGreaterThan,
|
|
Price: 1337,
|
|
OrderbookAmount: 1337,
|
|
}
|
|
exch, err := em.NewExchangeByName(testExchange)
|
|
require.NoError(t, err, "NewExchangeByName must not error")
|
|
|
|
conf, err := exchange.GetDefaultConfig(t.Context(), exch)
|
|
require.NoError(t, err, "GetDefaultConfig must not error")
|
|
|
|
require.NoError(t, exch.Setup(conf), "Setup must not error")
|
|
|
|
err = em.Add(exch)
|
|
require.NoError(t, err, "ExchangeManager Add must not error")
|
|
|
|
_, err = m.Add(testExchange, ItemPrice, cond, currency.NewPair(currency.BTC, currency.USD), asset.Spot, action)
|
|
require.NoError(t, err, "eventManager Add must not error")
|
|
|
|
m.m.Lock()
|
|
err = m.checkEventCondition(&m.events[0])
|
|
assert.ErrorIs(t, err, ticker.ErrTickerNotFound)
|
|
m.m.Unlock()
|
|
|
|
_, err = exch.UpdateTicker(t.Context(), currency.NewPair(currency.BTC, currency.USD), asset.Spot)
|
|
require.NoError(t, err, "UpdateTicker must not error")
|
|
|
|
m.m.Lock()
|
|
err = m.checkEventCondition(&m.events[0])
|
|
require.NoError(t, err, "checkEventCondition must not error")
|
|
m.m.Unlock()
|
|
|
|
m.events[0].Item = ItemOrderbook
|
|
m.events[0].Executed = false
|
|
m.events[0].Condition.CheckAsks = true
|
|
m.events[0].Condition.CheckBids = true
|
|
|
|
m.m.Lock()
|
|
err = m.checkEventCondition(&m.events[0])
|
|
assert.ErrorIs(t, err, orderbook.ErrOrderbookNotFound)
|
|
m.m.Unlock()
|
|
|
|
_, err = exch.UpdateOrderbook(t.Context(), currency.NewPair(currency.BTC, currency.USD), asset.Spot)
|
|
require.NoError(t, err, "UpdateOrderbook must not error")
|
|
|
|
m.m.Lock()
|
|
err = m.checkEventCondition(&m.events[0])
|
|
assert.NoError(t, err, "checkEventCondition should not error")
|
|
m.m.Unlock()
|
|
}
|