gocryptotrader/engine/exchange_test.go

package engine

import (
	"testing"

	"github.com/thrasher-corp/gocryptotrader/config"
)

var testSetup = false

func SetupTest(t *testing.T) {
	if !testSetup {
		if Bot == nil {
			Bot = new(Engine)
		}
		Bot.Config = &config.Cfg
		err := Bot.Config.LoadConfig("", true)
		if err != nil {
			t.Fatalf("SetupTest: Failed to load config: %s", err)
		}
		testSetup = true
	}

	if CheckExchangeExists("Bitfinex") {
		return
	}
	err := LoadExchange("Bitfinex", false, nil)
	if err != nil {
		t.Errorf("SetupTest: Failed to load exchange: %s", err)
	}
}

func CleanupTest(t *testing.T) {
	if !CheckExchangeExists("Bitfinex") {
		return
	}

	err := UnloadExchange("Bitfinex")
	if err != nil {
		t.Fatalf("CleanupTest: Failed to unload exchange: %s",
			err)
	}
}

func TestCheckExchangeExists(t *testing.T) {
	SetupTest(t)

	if !CheckExchangeExists("Bitfinex") {
		t.Errorf("TestGetExchangeExists: Unable to find exchange")
	}

	if CheckExchangeExists("Asdsad") {
		t.Errorf("TestGetExchangeExists: Non-existent exchange found")
	}

	CleanupTest(t)
}

func TestGetExchangeByName(t *testing.T) {
	SetupTest(t)

	exch := GetExchangeByName("Bitfinex")
	if exch == nil {
		t.Errorf("TestGetExchangeByName: Failed to get exchange")
	}

	if !exch.IsEnabled() {
		t.Errorf("TestGetExchangeByName: Unexpected result")
	}

	exch.SetEnabled(false)
	bfx := GetExchangeByName("Bitfinex")
	if bfx.IsEnabled() {
		t.Errorf("TestGetExchangeByName: Unexpected result")
	}

	if exch.GetName() != "Bitfinex" {
		t.Errorf("TestGetExchangeByName: Unexpected result")
	}

	exch = GetExchangeByName("Asdasd")
	if exch != nil {
		t.Errorf("TestGetExchangeByName: Non-existent exchange found")
	}

	CleanupTest(t)
}

func TestReloadExchange(t *testing.T) {
	SetupTest(t)

	err := ReloadExchange("asdf")
	if err != ErrExchangeNotFound {
		t.Errorf("TestReloadExchange: Incorrect result: %s",
			err)
	}

	err = ReloadExchange("Bitfinex")
	if err != nil {
		t.Errorf("TestReloadExchange: Incorrect result: %s",
			err)
	}

	CleanupTest(t)

	err = ReloadExchange("asdf")
	if err != ErrNoExchangesLoaded {
		t.Errorf("TestReloadExchange: Incorrect result: %s",
			err)
	}
}

func TestUnloadExchange(t *testing.T) {
	SetupTest(t)

	err := UnloadExchange("asdf")
	if err != ErrExchangeNotFound {
		t.Errorf("TestUnloadExchange: Incorrect result: %s",
			err)
	}

	err = UnloadExchange("Bitfinex")
	if err != nil {
		t.Errorf("TestUnloadExchange: Failed to get exchange. %s",
			err)
	}

	err = UnloadExchange("asdf")
	if err != ErrNoExchangesLoaded {
		t.Errorf("TestUnloadExchange: Incorrect result: %s",
			err)
	}

	CleanupTest(t)
}