Add web3@1.0 to web3Factory

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;
+    },
 };