diff options
Diffstat (limited to 'test/utils/token_utils.ts')
-rw-r--r-- | test/utils/token_utils.ts | 10 |
1 files changed, 4 insertions, 6 deletions
diff --git a/test/utils/token_utils.ts b/test/utils/token_utils.ts index 00fa55260..f4fa7ac31 100644 --- a/test/utils/token_utils.ts +++ b/test/utils/token_utils.ts @@ -1,6 +1,4 @@ -import find = require('lodash/find'); -import filter = require('lodash/filter'); -import isUndefined = require('lodash/isUndefined'); +import * as _ from 'lodash'; import {Token, ZeroExError} from '../../src'; const PROTOCOL_TOKEN_SYMBOL = 'ZRX'; @@ -11,14 +9,14 @@ export class TokenUtils { this.tokens = tokens; } public getProtocolTokenOrThrow(): Token { - const zrxToken = find(this.tokens, {symbol: PROTOCOL_TOKEN_SYMBOL}); - if (isUndefined(zrxToken)) { + const zrxToken = _.find(this.tokens, {symbol: PROTOCOL_TOKEN_SYMBOL}); + if (_.isUndefined(zrxToken)) { throw new Error(ZeroExError.ZRX_NOT_IN_TOKEN_REGISTRY); } return zrxToken; } public getNonProtocolTokens(): Token[] { - const nonProtocolTokens = filter(this.tokens, token => { + const nonProtocolTokens = _.filter(this.tokens, token => { return token.symbol !== PROTOCOL_TOKEN_SYMBOL; }); return nonProtocolTokens; |