Refactor Portfolio balance.

1 files changed, 16 insertions, 0 deletions

diff --git a/markets/poloniex.go b/markets/poloniex.go
index 9aaeafb..58645b2 100644
--- a/markets/poloniex.go
+++ b/markets/poloniex.go
@@ -52,6 +52,22 @@ func NewPoloniex() *Poloniex {
         }
 }
 
+func (p *Poloniex) TestCredentials(apiKey, apiSecret string) error {
+        client, _ := poloniex.NewClient(apiKey, apiSecret)
+
+        _, err := client.TradeReturnDepositAdresses()
+
+        if poloniexInvalidCredentialsError(err) {
+                return utils.Error{InvalidCredentials, "invalid poloniex credentials"}
+        }
+
+        if poloniexRestrictedIPError(err) {
+                return utils.Error{IPRestricted, "IP restricted api key"}
+        }
+
+        return nil
+}
+
 func (p *Poloniex) GetBalance(apiKey, apiSecret string) (Summary, error) {
         client, _ := poloniex.NewClient(apiKey, apiSecret)
         var summary Summary