diff options
| -rw-r--r-- | test/utils/web3_factory.ts | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/test/utils/web3_factory.ts b/test/utils/web3_factory.ts index b20070c74..863fa1d07 100644 --- a/test/utils/web3_factory.ts +++ b/test/utils/web3_factory.ts @@ -6,6 +6,7 @@ import ProviderEngine = require('web3-provider-engine'); import RpcSubprovider = require('web3-provider-engine/subproviders/rpc'); import * as Web3 from 'web3'; +import * as Web3_beta from 'web3_beta'; import {constants} from './constants'; import {EmptyWalletSubProvider} from '../../src/subproviders/empty_wallet_subprovider'; @@ -16,6 +17,12 @@ export const web3Factory = { web3.setProvider(provider); return web3; }, + createBeta(): any { + const provider_beta = this.getProviderBeta(); + const web3_beta = new Web3_beta(); + web3_beta.setProvider(provider_beta); + return web3_beta; + }, getRpcProvider(hasAddresses: boolean = true): Web3.Provider { const provider = new ProviderEngine(); const rpcUrl = `http://${constants.RPC_HOST}:${constants.RPC_PORT}`; @@ -28,4 +35,9 @@ export const web3Factory = { provider.start(); return provider; }, + getProviderBeta(): any { + const rpcUrl = `http://${constants.RPC_HOST}:${constants.RPC_PORT}`; + const providerBeta = new Web3_beta.providers.HttpProvider(rpcUrl); + return providerBeta; + }, }; |