gocryptotrader/engine/currency_state_manager_test.go

package engine

import (
	"context"
	"errors"
	"sync"
	"testing"
	"time"

	"github.com/stretchr/testify/require"
	"github.com/thrasher-corp/gocryptotrader/common"
	"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/currencystate"
)

func TestSetupCurrencyStateManager(t *testing.T) {
	t.Parallel()
	_, err := SetupCurrencyStateManager(0, nil)
	require.ErrorIs(t, err, errNilExchangeManager)

	cm, err := SetupCurrencyStateManager(0, &ExchangeManager{})
	require.NoError(t, err)

	if cm.sleep != DefaultStateManagerDelay {
		t.Fatal("unexpected value")
	}
}

var (
	errManager  = errors.New("manager level error")
	errExchange = errors.New("exchange level error")
)

type fakeExchangeManagerino struct {
	ErrorMeOne bool
	ErrorMeTwo bool
}

func (f *fakeExchangeManagerino) GetExchanges() ([]exchange.IBotExchange, error) {
	if f.ErrorMeOne {
		return nil, errManager
	}
	return []exchange.IBotExchange{&fakerino{errorMe: f.ErrorMeTwo}}, nil
}

func (f *fakeExchangeManagerino) GetExchangeByName(_ string) (exchange.IBotExchange, error) {
	if f.ErrorMeOne {
		return nil, errManager
	}
	return &fakerino{errorMe: f.ErrorMeTwo}, nil
}

type fakerino struct {
	exchange.IBotExchange
	errorMe                bool
	GetAvailablePairsError bool
	GetBaseError           bool
}

func (f *fakerino) UpdateCurrencyStates(_ context.Context, _ asset.Item) error {
	if f.errorMe {
		return common.ErrNotYetImplemented
	}
	return nil
}

func (f *fakerino) GetAssetTypes(_ bool) asset.Items {
	return asset.Items{asset.Spot}
}

func (f *fakerino) GetName() string {
	return "testssssssssssssss"
}

func (f *fakerino) GetCurrencyStateSnapshot() ([]currencystate.Snapshot, error) {
	if f.errorMe {
		return nil, errExchange
	}
	return []currencystate.Snapshot{
		{Code: currency.SHORTY, Asset: asset.Spot},
	}, nil
}

func (f *fakerino) CanWithdraw(_ currency.Code, _ asset.Item) error {
	if f.errorMe {
		return errExchange
	}
	return nil
}

func (f *fakerino) CanDeposit(_ currency.Code, _ asset.Item) error {
	if f.errorMe {
		return errExchange
	}
	return nil
}

func (f *fakerino) CanTrade(_ currency.Code, _ asset.Item) error {
	if f.errorMe {
		return errExchange
	}
	return nil
}

func (f *fakerino) CanTradePair(_ currency.Pair, _ asset.Item) error {
	if f.errorMe {
		return errExchange
	}
	return nil
}

func (f *fakerino) GetAvailablePairs(_ asset.Item) (currency.Pairs, error) {
	if f.GetAvailablePairsError {
		return nil, errExchange
	}
	return currency.Pairs{currency.NewBTCUSD()}, nil
}

func (f *fakerino) GetBase() *exchange.Base {
	if f.GetBaseError {
		return nil
	}
	return &exchange.Base{States: currencystate.NewCurrencyStates()}
}

func TestCurrencyStateManagerIsRunning(t *testing.T) {
	t.Parallel()
	err := (*CurrencyStateManager)(nil).Stop()
	require.ErrorIs(t, err, ErrNilSubsystem)

	err = (&CurrencyStateManager{}).Stop()
	require.ErrorIs(t, err, ErrSubSystemNotStarted)

	err = (&CurrencyStateManager{started: 1, shutdown: make(chan struct{})}).Stop()
	require.NoError(t, err)

	err = (*CurrencyStateManager)(nil).Start()
	require.ErrorIs(t, err, ErrNilSubsystem)

	err = (&CurrencyStateManager{started: 1}).Start()
	require.ErrorIs(t, err, ErrSubSystemAlreadyStarted)

	man := &CurrencyStateManager{
		shutdown:         make(chan struct{}),
		iExchangeManager: &fakeExchangeManagerino{ErrorMeOne: true},
		sleep:            time.Minute,
	}
	err = man.Start()
	require.NoError(t, err)

	time.Sleep(time.Millisecond)

	err = man.Stop()
	require.NoError(t, err)

	man.iExchangeManager = &fakeExchangeManagerino{ErrorMeOne: true}
	err = man.Start()
	require.NoError(t, err)

	time.Sleep(time.Millisecond)

	err = man.Stop()
	require.NoError(t, err)

	man.iExchangeManager = &fakeExchangeManagerino{ErrorMeOne: true}
	err = man.Start()
	require.NoError(t, err)

	time.Sleep(time.Millisecond)

	if !man.IsRunning() {
		t.Fatal("this should be running")
	}

	err = man.Stop()
	require.NoError(t, err)

	if man.IsRunning() {
		t.Fatal("this should be stopped")
	}
}

func TestGetAllRPC(t *testing.T) {
	t.Parallel()
	_, err := (*CurrencyStateManager)(nil).GetAllRPC("")
	require.ErrorIs(t, err, ErrSubSystemNotStarted)

	_, err = (&CurrencyStateManager{
		started:          1,
		iExchangeManager: &fakeExchangeManagerino{ErrorMeOne: true},
	}).GetAllRPC("")
	require.ErrorIs(t, err, errManager)

	_, err = (&CurrencyStateManager{
		started:          1,
		iExchangeManager: &fakeExchangeManagerino{ErrorMeTwo: true},
	}).GetAllRPC("")
	require.ErrorIs(t, err, errExchange)

	_, err = (&CurrencyStateManager{
		started:          1,
		iExchangeManager: &fakeExchangeManagerino{},
	}).GetAllRPC("")
	require.NoError(t, err)
}

func TestCanWithdrawRPC(t *testing.T) {
	t.Parallel()
	_, err := (*CurrencyStateManager)(nil).CanWithdrawRPC("", currency.EMPTYCODE, asset.Empty)
	require.ErrorIs(t, err, ErrSubSystemNotStarted)

	_, err = (&CurrencyStateManager{
		started:          1,
		iExchangeManager: &fakeExchangeManagerino{ErrorMeOne: true},
	}).CanWithdrawRPC("", currency.EMPTYCODE, asset.Empty)
	require.ErrorIs(t, err, errManager)

	_, err = (&CurrencyStateManager{
		started:          1,
		iExchangeManager: &fakeExchangeManagerino{ErrorMeTwo: true},
	}).CanWithdrawRPC("", currency.EMPTYCODE, asset.Empty)
	require.ErrorIs(t, err, errExchange)

	_, err = (&CurrencyStateManager{
		started:          1,
		iExchangeManager: &fakeExchangeManagerino{},
	}).CanWithdrawRPC("", currency.EMPTYCODE, asset.Empty)
	require.NoError(t, err)
}

func TestCanDepositRPC(t *testing.T) {
	t.Parallel()
	_, err := (*CurrencyStateManager)(nil).CanDepositRPC("", currency.EMPTYCODE, asset.Empty)
	require.ErrorIs(t, err, ErrSubSystemNotStarted)

	_, err = (&CurrencyStateManager{
		started:          1,
		iExchangeManager: &fakeExchangeManagerino{ErrorMeOne: true},
	}).CanDepositRPC("", currency.EMPTYCODE, asset.Empty)
	require.ErrorIs(t, err, errManager)

	_, err = (&CurrencyStateManager{
		started:          1,
		iExchangeManager: &fakeExchangeManagerino{ErrorMeTwo: true},
	}).CanDepositRPC("", currency.EMPTYCODE, asset.Empty)
	require.ErrorIs(t, err, errExchange)

	_, err = (&CurrencyStateManager{
		started:          1,
		iExchangeManager: &fakeExchangeManagerino{},
	}).CanDepositRPC("", currency.EMPTYCODE, asset.Empty)
	require.NoError(t, err)
}

func TestCanTradeRPC(t *testing.T) {
	t.Parallel()
	_, err := (*CurrencyStateManager)(nil).CanTradeRPC("", currency.EMPTYCODE, asset.Empty)
	require.ErrorIs(t, err, ErrSubSystemNotStarted)

	_, err = (&CurrencyStateManager{
		started:          1,
		iExchangeManager: &fakeExchangeManagerino{ErrorMeOne: true},
	}).CanTradeRPC("", currency.EMPTYCODE, asset.Empty)
	require.ErrorIs(t, err, errManager)

	_, err = (&CurrencyStateManager{
		started:          1,
		iExchangeManager: &fakeExchangeManagerino{ErrorMeTwo: true},
	}).CanTradeRPC("", currency.EMPTYCODE, asset.Empty)
	require.ErrorIs(t, err, errExchange)

	_, err = (&CurrencyStateManager{
		started:          1,
		iExchangeManager: &fakeExchangeManagerino{},
	}).CanTradeRPC("", currency.EMPTYCODE, asset.Empty)
	require.NoError(t, err)
}

func TestCanTradePairRPC(t *testing.T) {
	t.Parallel()
	_, err := (*CurrencyStateManager)(nil).CanTradePairRPC("", currency.EMPTYPAIR, asset.Empty)
	require.ErrorIs(t, err, ErrSubSystemNotStarted)

	_, err = (&CurrencyStateManager{
		started:          1,
		iExchangeManager: &fakeExchangeManagerino{ErrorMeOne: true},
	}).CanTradePairRPC("", currency.EMPTYPAIR, asset.Empty)
	require.ErrorIs(t, err, errManager)

	_, err = (&CurrencyStateManager{
		started:          1,
		iExchangeManager: &fakeExchangeManagerino{ErrorMeTwo: true},
	}).CanTradePairRPC("", currency.EMPTYPAIR, asset.Empty)
	require.ErrorIs(t, err, errExchange)

	_, err = (&CurrencyStateManager{
		started:          1,
		iExchangeManager: &fakeExchangeManagerino{},
	}).CanTradePairRPC("", currency.EMPTYPAIR, asset.Empty)
	require.NoError(t, err)
}

func TestUpdate(_ *testing.T) {
	man := &CurrencyStateManager{}
	var wg sync.WaitGroup
	wg.Add(3)
	man.update(&fakerino{errorMe: true, GetAvailablePairsError: true}, &wg, asset.Items{asset.Spot})
	man.update(&fakerino{errorMe: true, GetBaseError: true}, &wg, asset.Items{asset.Spot})
	man.update(&fakerino{errorMe: true}, &wg, asset.Items{asset.Spot})
}