From e5bb3bc75eb2fe0e2a4adbd99d64362c305bdd3e Mon Sep 17 00:00:00 2001
From: Leonid Logvinov <logvinov.leon@gmail.com>
Date: Tue, 29 Aug 2017 18:41:14 +0200
Subject: Add web3@1.0 to web3Factory

---
 test/utils/web3_factory.ts | 12 ++++++++++++
 1 file changed, 12 insertions(+)

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;
+    },
 };
-- 
cgit v1.2.3