Add exchange GetAvailableCurrencies function

marclop · Aug 23, 2017 · a4c996b · a4c996b
1 parent 7be8c02
commit a4c996b
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/exchanges/exchange.go b/exchanges/exchange.go
@@ -85,6 +85,12 @@ func (e *Base) GetEnabledCurrencies() []string {
 	return e.EnabledPairs
 }
 
+// GetAvailableCurrencies is a method that returns the available currency pairs
+// of the exchange base
+func (e *Base) GetAvailableCurrencies() []string {
+	return e.AvailablePairs
+}
+
 // FormatCurrency is a method that formats and returns a currency pair
 // based on the user currency display preferences
 func FormatCurrency(p pair.CurrencyPair) pair.CurrencyItem {

diff --git a/exchanges/exchange_test.go b/exchanges/exchange_test.go
@@ -32,6 +32,19 @@ func TestGetEnabledCurrencies(t *testing.T) {
 	}
 }
 
+func TestGetAvailableCurrencies(t *testing.T) {
+	availablePairs := []string{"BTCUSD", "BTCAUD", "LTCUSD", "LTCAUD"}
+	GetEnabledCurrencies := Base{
+		Name:           "TESTNAME",
+		AvailablePairs: availablePairs,
+	}
+
+	enCurr := GetEnabledCurrencies.GetAvailableCurrencies()
+	if enCurr[0] != "BTCUSD" {
+		t.Error("Test Failed - Exchange GetAvailableCurrencies() incorrect string")
+	}
+}
+
 func TestFormatCurrency(t *testing.T) {
 	cfg := config.GetConfig()
 	err := cfg.LoadConfig(config.ConfigTestFile)