mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-13 23:16:45 +00:00
19b8957f3f
* tests: Replace !errors.Is(err, target) with testify equivalents * codebase: Manual !errors.Is(err, target) replacements * typo: Replace errMisMatchedEvent with errMismatchedEvent * tests: Enhance error messages for better output * tests: Refactor error assertions in various test cases to use require and improve clarity * misc linter: Fix assert should wording * tests: Simplify assertions in TestCreateSignals for clarity and conciseness * tests: Enhance assertion message in TestCreateSignals
277 lines
7.3 KiB
Go
277 lines
7.3 KiB
Go
package engine
|
|
|
|
import (
|
|
"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)
|
|
assert.ErrorIs(t, err, errNilComManager)
|
|
|
|
_, err = setupEventManager(&CommunicationManager{}, nil, 0, false)
|
|
assert.ErrorIs(t, err, errNilExchangeManager)
|
|
|
|
m, err := setupEventManager(&CommunicationManager{}, &ExchangeManager{}, 0, false)
|
|
require.NoError(t, err)
|
|
|
|
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)
|
|
assert.NoError(t, err)
|
|
|
|
err = m.Start()
|
|
assert.NoError(t, err)
|
|
|
|
err = m.Start()
|
|
assert.ErrorIs(t, err, ErrSubSystemAlreadyStarted)
|
|
|
|
m = nil
|
|
err = m.Start()
|
|
assert.ErrorIs(t, err, ErrNilSubsystem)
|
|
}
|
|
|
|
func TestEventManagerIsRunning(t *testing.T) {
|
|
t.Parallel()
|
|
m, err := setupEventManager(&CommunicationManager{}, &ExchangeManager{}, 0, false)
|
|
assert.NoError(t, err)
|
|
|
|
err = m.Start()
|
|
assert.NoError(t, err)
|
|
|
|
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)
|
|
assert.NoError(t, err)
|
|
|
|
err = m.Start()
|
|
assert.NoError(t, err)
|
|
|
|
err = m.Stop()
|
|
assert.NoError(t, err)
|
|
|
|
err = m.Stop()
|
|
assert.ErrorIs(t, err, ErrSubSystemNotStarted)
|
|
|
|
m = nil
|
|
err = m.Stop()
|
|
assert.ErrorIs(t, err, ErrNilSubsystem)
|
|
}
|
|
|
|
func TestEventManagerAdd(t *testing.T) {
|
|
t.Parallel()
|
|
em := NewExchangeManager()
|
|
m, err := setupEventManager(&CommunicationManager{}, em, 0, false)
|
|
assert.NoError(t, err)
|
|
|
|
_, err = m.Add("", "", EventConditionParams{}, currency.NewPair(currency.BTC, currency.USDC), asset.Spot, "")
|
|
assert.ErrorIs(t, err, ErrSubSystemNotStarted)
|
|
|
|
err = m.Start()
|
|
assert.NoError(t, err)
|
|
|
|
_, err = m.Add("", "", EventConditionParams{}, currency.NewPair(currency.BTC, currency.USDC), asset.Spot, "")
|
|
assert.ErrorIs(t, err, errExchangeDisabled)
|
|
|
|
exch, err := em.NewExchangeByName(testExchange)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
exch.SetDefaults()
|
|
err = em.Add(exch)
|
|
require.NoError(t, err)
|
|
|
|
_, err = m.Add(testExchange, "", EventConditionParams{}, currency.NewPair(currency.BTC, currency.USDC), asset.Spot, "")
|
|
assert.ErrorIs(t, err, errInvalidItem)
|
|
|
|
cond := EventConditionParams{
|
|
Condition: ConditionGreaterThan,
|
|
Price: 1337,
|
|
OrderbookAmount: 1337,
|
|
}
|
|
_, err = m.Add(testExchange, ItemPrice, cond, currency.NewPair(currency.BTC, currency.USDC), asset.Spot, "")
|
|
assert.ErrorIs(t, err, errInvalidAction)
|
|
|
|
_, err = m.Add(testExchange, ItemPrice, cond, currency.NewPair(currency.BTC, currency.USDC), asset.Spot, ActionTest)
|
|
assert.NoError(t, err)
|
|
|
|
action := ActionSMSNotify + "," + ActionTest
|
|
_, err = m.Add(testExchange, ItemPrice, cond, currency.NewPair(currency.BTC, currency.USDC), asset.Spot, action)
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func TestEventManagerRemove(t *testing.T) {
|
|
t.Parallel()
|
|
em := NewExchangeManager()
|
|
m, err := setupEventManager(&CommunicationManager{}, em, 0, false)
|
|
assert.NoError(t, err)
|
|
|
|
if m.Remove(0) {
|
|
t.Error("expected false")
|
|
}
|
|
err = m.Start()
|
|
assert.NoError(t, err)
|
|
|
|
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)
|
|
require.NoError(t, err)
|
|
|
|
id, err := m.Add(testExchange, ItemPrice, cond, currency.NewPair(currency.BTC, currency.USDC), asset.Spot, action)
|
|
assert.NoError(t, err)
|
|
|
|
if !m.Remove(id) {
|
|
t.Error("expected true")
|
|
}
|
|
}
|
|
|
|
func TestGetEventCounter(t *testing.T) {
|
|
t.Parallel()
|
|
em := NewExchangeManager()
|
|
m, err := setupEventManager(&CommunicationManager{}, em, 0, false)
|
|
assert.NoError(t, err)
|
|
|
|
total, executed := m.getEventCounter()
|
|
if total != 0 && executed != 0 {
|
|
t.Error("expected 0")
|
|
}
|
|
err = m.Start()
|
|
assert.NoError(t, err)
|
|
|
|
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)
|
|
require.NoError(t, err)
|
|
|
|
_, err = m.Add(testExchange, ItemPrice, cond, currency.NewPair(currency.BTC, currency.USDC), asset.Spot, action)
|
|
assert.NoError(t, err)
|
|
|
|
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.NewBTCUSD(), 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.NewBTCUSD(), 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.NewBTCUSD(), 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()
|
|
}
|