diff options
Diffstat (limited to 'test/utils/token_utils.ts')
-rw-r--r-- | test/utils/token_utils.ts | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/test/utils/token_utils.ts b/test/utils/token_utils.ts index f4fa7ac31..00fa55260 100644 --- a/test/utils/token_utils.ts +++ b/test/utils/token_utils.ts @@ -1,4 +1,6 @@ -import * as _ from 'lodash'; +import find = require('lodash/find'); +import filter = require('lodash/filter'); +import isUndefined = require('lodash/isUndefined'); import {Token, ZeroExError} from '../../src'; const PROTOCOL_TOKEN_SYMBOL = 'ZRX'; @@ -9,14 +11,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; |