From 5d2f9d7a33c4494b06098f13fca487613fe83c73 Mon Sep 17 00:00:00 2001
From: Alex Browne <stephenalexbrowne@gmail.com>
Date: Wed, 6 Jun 2018 10:56:01 -0700
Subject: Use an enum for ProviderType in contracts/src/utils/web3_wrapper

---
 packages/contracts/src/utils/web3_wrapper.ts | 25 +++++++++++++++++++++----
 1 file changed, 21 insertions(+), 4 deletions(-)

(limited to 'packages/contracts')

diff --git a/packages/contracts/src/utils/web3_wrapper.ts b/packages/contracts/src/utils/web3_wrapper.ts
index 6df8ac073..63ce2c8cc 100644
--- a/packages/contracts/src/utils/web3_wrapper.ts
+++ b/packages/contracts/src/utils/web3_wrapper.ts
@@ -5,7 +5,25 @@ import { Provider } from 'ethereum-types';
 
 import { coverage } from './coverage';
 
-const testProvider = process.env.TEST_PROVIDER || 'ganache';
+enum ProviderType {
+    Ganache = 'ganache',
+    Geth = 'geth',
+}
+
+let testProvider: ProviderType;
+switch (process.env.TEST_PROVIDER) {
+    case undefined:
+        testProvider = ProviderType.Ganache;
+        break;
+    case 'ganache':
+        testProvider = ProviderType.Ganache;
+        break;
+    case 'geth':
+        testProvider = ProviderType.Geth;
+        break;
+    default:
+        throw new Error(`Unknown TEST_PROVIDER: ${process.env.TEST_PROVIDER}`);
+}
 
 const ganacheTxDefaults = {
     from: devConstants.TESTRPC_FIRST_ADDRESS,
@@ -14,7 +32,7 @@ const ganacheTxDefaults = {
 const gethTxDefaults = {
     from: devConstants.TESTRPC_FIRST_ADDRESS,
 };
-export const txDefaults = testProvider === 'ganache' ? ganacheTxDefaults : gethTxDefaults;
+export const txDefaults = testProvider === ProviderType.Ganache ? ganacheTxDefaults : gethTxDefaults;
 
 const gethConfigs = {
     shouldUseInProcessGanache: false,
@@ -24,8 +42,7 @@ const gethConfigs = {
 const ganacheConfigs = {
     shouldUseInProcessGanache: true,
 };
-
-const providerConfigs = testProvider === 'ganache' ? ganacheConfigs : gethConfigs;
+const providerConfigs = testProvider === ProviderType.Ganache ? ganacheConfigs : gethConfigs;
 
 export const provider = web3Factory.getRpcProvider(providerConfigs);
 const isCoverageEnabled = env.parseBoolean(EnvVars.SolidityCoverage);
-- 
cgit v1.2.3