diff options
author | Fabio Berger <me@fabioberger.com> | 2017-06-03 01:03:53 +0800 |
---|---|---|
committer | Fabio Berger <me@fabioberger.com> | 2017-06-03 01:03:53 +0800 |
commit | 35c9330d6246183509a3543cdf1278b9ced191e2 (patch) | |
tree | 5cef8b37a52ef4ce258a971b1ae52c5c2ad63a63 /test/utils/token_utils.ts | |
parent | ab5b5a881bc8f8f8d176ae65440de8fcd25ad906 (diff) | |
parent | c83587a16d016d1efafaf31abb9b39eb54128568 (diff) | |
download | dexon-sol-tools-35c9330d6246183509a3543cdf1278b9ced191e2.tar dexon-sol-tools-35c9330d6246183509a3543cdf1278b9ced191e2.tar.gz dexon-sol-tools-35c9330d6246183509a3543cdf1278b9ced191e2.tar.bz2 dexon-sol-tools-35c9330d6246183509a3543cdf1278b9ced191e2.tar.lz dexon-sol-tools-35c9330d6246183509a3543cdf1278b9ced191e2.tar.xz dexon-sol-tools-35c9330d6246183509a3543cdf1278b9ced191e2.tar.zst dexon-sol-tools-35c9330d6246183509a3543cdf1278b9ced191e2.zip |
Merge branch 'master' into unavailableFilledCancelled
# Conflicts:
# src/contract_wrappers/exchange_wrapper.ts
# src/types.ts
# test/exchange_wrapper_test.ts
Diffstat (limited to 'test/utils/token_utils.ts')
-rw-r--r-- | test/utils/token_utils.ts | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/test/utils/token_utils.ts b/test/utils/token_utils.ts new file mode 100644 index 000000000..14788b299 --- /dev/null +++ b/test/utils/token_utils.ts @@ -0,0 +1,24 @@ +import * as _ from 'lodash'; +import {Token, ZeroExError} from '../../src/types'; + +const PROTOCOL_TOKEN_SYMBOL = 'ZRX'; + +export class TokenUtils { + private tokens: Token[]; + constructor(tokens: Token[]) { + this.tokens = tokens; + } + public getProtocolTokenOrThrow(): Token { + 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 => { + return token.symbol !== PROTOCOL_TOKEN_SYMBOL; + }); + return nonProtocolTokens; + } +} |