Bug Fixes

TaskServer/node_modules/ccxt/js/test/Exchange/test.fetchBalance.js

@@ -2,15 +2,12 @@
 
 // ----------------------------------------------------------------------------
 
-const log       = require ('ololog')
-    , ansi      = require ('ansicolor').nice
-    , chai      = require ('chai')
-    , expect    = chai.expect
-    , assert    = chai.assert
+const log = require ('ololog')
+    , testBalance = require ('./test.balance.js')
 
-/*  ------------------------------------------------------------------------ */
+// ----------------------------------------------------------------------------
 
-module.exports = async (exchange, symbol) => {
+module.exports = async (exchange) => {
 
     if (!(exchange.has.fetchBalance)) {
         log (exchange.id.green, ' does not have fetchBalance')
@@ -19,61 +16,9 @@ module.exports = async (exchange, symbol) => {
 
     log ('fetching balance...')
 
-    let balance = await exchange.fetchBalance ()
+    const response = await exchange.fetchBalance ()
 
-    let currencies = [
-        'USD',
-        'CNY',
-        'EUR',
-        'BTC',
-        'ETH',
-        'JPY',
-        'LTC',
-        'DASH',
-        'DOGE',
-        'UAH',
-        'RUB',
-        'XRP',
-    ]
+    testBalance (exchange, response)
 
-    // log.yellow (balance)
-
-    assert (typeof balance['total'] === 'object')
-    assert (typeof balance['free'] === 'object')
-    assert (typeof balance['used'] === 'object')
-
-    if ('info' in balance) {
-        for (let currency of Object.keys (balance['total'])) {
-            let total = balance['total'][currency]
-            let free = balance['free'][currency]
-            let used = balance['used'][currency]
-            if (total !== undefined && free !== undefined && used !== undefined) {
-                assert (total === free + used, 'free and used do not sum to total ' + exchange.id)
-            }
-        }
-
-        let result = currencies
-            .filter (currency => (currency in balance) &&
-                (balance[currency]['total'] !== undefined))
-
-        if (result.length > 0) {
-            result = result.map (currency => currency + ': ' + balance[currency]['total'])
-            if (exchange.currencies.length > result.length)
-                result = result.join (', ') + ' + more...'
-            else
-                result = result.join (', ')
-
-        } else {
-
-            result = 'zero balance'
-        }
-
-        log (result)
-
-    } else {
-
-        log (exchange.omit (balance, 'info'))
-    }
-
-    return balance
+    return response
 }