aboutsummaryrefslogtreecommitdiffstats
path: root/packages/contracts/test/utils/token_registry_wrapper.ts
diff options
context:
space:
mode:
Diffstat (limited to 'packages/contracts/test/utils/token_registry_wrapper.ts')
-rw-r--r--packages/contracts/test/utils/token_registry_wrapper.ts66
1 files changed, 66 insertions, 0 deletions
diff --git a/packages/contracts/test/utils/token_registry_wrapper.ts b/packages/contracts/test/utils/token_registry_wrapper.ts
new file mode 100644
index 000000000..0abf20e03
--- /dev/null
+++ b/packages/contracts/test/utils/token_registry_wrapper.ts
@@ -0,0 +1,66 @@
+import { Web3Wrapper } from '@0xproject/web3-wrapper';
+import { Provider } from 'ethereum-types';
+
+import { TokenRegistryContract } from '../../generated_contract_wrappers/token_registry';
+
+import { Token } from './types';
+
+import { constants } from './constants';
+
+export class TokenRegWrapper {
+ private _tokenReg: TokenRegistryContract;
+ private _web3Wrapper: Web3Wrapper;
+ constructor(tokenRegContract: TokenRegistryContract, provider: Provider) {
+ this._tokenReg = tokenRegContract;
+ this._web3Wrapper = new Web3Wrapper(provider);
+ }
+ public async addTokenAsync(token: Token, from: string): Promise<string> {
+ const txHash = await this._tokenReg.addToken.sendTransactionAsync(
+ token.address as string,
+ token.name,
+ token.symbol,
+ token.decimals,
+ token.ipfsHash,
+ token.swarmHash,
+ { from },
+ );
+ await this._web3Wrapper.awaitTransactionSuccessAsync(txHash, constants.AWAIT_TRANSACTION_MINED_MS);
+ return txHash;
+ }
+ public async getTokenMetaDataAsync(tokenAddress: string): Promise<Token> {
+ const data = await this._tokenReg.getTokenMetaData.callAsync(tokenAddress);
+ const token: Token = {
+ address: data[0],
+ name: data[1],
+ symbol: data[2],
+ decimals: data[3],
+ ipfsHash: data[4],
+ swarmHash: data[5],
+ };
+ return token;
+ }
+ public async getTokenByNameAsync(tokenName: string): Promise<Token> {
+ const data = await this._tokenReg.getTokenByName.callAsync(tokenName);
+ const token: Token = {
+ address: data[0],
+ name: data[1],
+ symbol: data[2],
+ decimals: data[3],
+ ipfsHash: data[4],
+ swarmHash: data[5],
+ };
+ return token;
+ }
+ public async getTokenBySymbolAsync(tokenSymbol: string): Promise<Token> {
+ const data = await this._tokenReg.getTokenBySymbol.callAsync(tokenSymbol);
+ const token: Token = {
+ address: data[0],
+ name: data[1],
+ symbol: data[2],
+ decimals: data[3],
+ ipfsHash: data[4],
+ swarmHash: data[5],
+ };
+ return token;
+ }
+}