fei-protocol · Joeysantoro · Sep 13, 2021 · Sep 12, 2021 · Sep 13, 2021 · Sep 13, 2021
diff --git a/hardhat.config.ts b/hardhat.config.ts
@@ -14,10 +14,21 @@ const rinkebyAlchemyApiKey = process.env.RINKEBY_ALCHEMY_API_KEY;
 const testnetPrivateKey = process.env.TESTNET_PRIVATE_KEY;
 const privateKey = process.env.ETH_PRIVATE_KEY;
 const runE2ETests = process.env.RUN_E2E_TESTS;
+const enableMainnetForking = process.env.ENABLE_MAINNET_FORKING;
 const mainnetAlchemyApiKey = process.env.MAINNET_ALCHEMY_API_KEY;
 
 if (!rinkebyAlchemyApiKey || !testnetPrivateKey || !privateKey || !mainnetAlchemyApiKey) {
-  throw new Error('Please set your Ethereum keys in a .env')
+  console.warn("Not all Ethereum keys provided; some functionality will be unavailable.")
+}
+
+if (enableMainnetForking) {
+  if (!mainnetAlchemyApiKey) {
+    throw new Error("Cannot fork mainnet without mainnet alchemy api key.")
+  }
+
+  console.log("Mainnet forking enabled.")
+} else {
+  console.log("Mainnet forking disabled.")
 }
 
 const config: HardhatUserConfig = {
@@ -28,21 +39,21 @@ const config: HardhatUserConfig = {
     hardhat: {
       gas: 12e6,
       chainId: 5777, // Any network (default: none)
-      forking: {
+      forking: enableMainnetForking ? {
         url: `https://eth-mainnet.alchemyapi.io/v2/${mainnetAlchemyApiKey}`,
         blockNumber: 13135475
-      }
+      } : undefined
     },
     localhost: {
       url: 'http://127.0.0.1:8545'
     },
     rinkeby: {
       url: `https://eth-rinkeby.alchemyapi.io/v2/${rinkebyAlchemyApiKey}`,
-      accounts: [testnetPrivateKey]
+      accounts: testnetPrivateKey ? [testnetPrivateKey] : []
     },
     mainnet: {
       url: `https://eth-mainnet.alchemyapi.io/v2/${mainnetAlchemyApiKey}`,
-      accounts: [privateKey]
+      accounts: privateKey ? [privateKey] : []
     },
   },
   solidity: {

diff --git a/test/bondingcurve/EthBondingCurve.test.js b/test/bondingcurve/EthBondingCurve.test.js
@@ -1,3 +1,5 @@
+const hre = require('hardhat');
+const { env, hrtime } = require('process');
 const {
   BN,
   expectEvent,
@@ -22,6 +24,15 @@ describe('EthBondingCurve', function () {
   let beneficiaryAddress1;
   let beneficiaryAddress2;
 
+  this.beforeAll(async function() {
+    // Can only get the current price on a forked network (since we haven't deployed Uniswap stuff in test setup)
+    if (!hre.network.config.forking) {
+      return this.skip();
+    }
+
+    return undefined;
+  });
+
   beforeEach(async function () {
     ({
       userAddress,

diff --git a/test/pcv/PCVSwapperUniswap.test.js b/test/pcv/PCVSwapperUniswap.test.js
@@ -1,3 +1,5 @@
+const hre = require('hardhat');
+
 const {
   BN,
   time,
@@ -27,6 +29,15 @@ describe('PCVSwapperUniswap', function () {
   let governorAddress;
   let pcvControllerAddress;
 
+  this.beforeAll(async function() {
+    // Can only get the current price on a forked network (since we haven't deployed Uniswap stuff in test setup)
+    if (!hre.network.config.forking) {
+      return this.skip();
+    }
+
+    return undefined;
+  });
+
   beforeEach(async function () {
     ({
       userAddress,

diff --git a/test/stablizer/EthReserveStabilizer.test.js b/test/stablizer/EthReserveStabilizer.test.js
@@ -1,3 +1,5 @@
+const hre = require('hardhat');
+
 const {
   web3,
   BN,
@@ -20,14 +22,23 @@ describe('EthReserveStabilizer', function () {
   let minterAddress;
   let pcvControllerAddress;
 
+  this.beforeAll(async function() {
+    // Can only get the current price on a forked network (since we haven't deployed Uniswap stuff in test setup)
+    if (!hre.network.config.forking) {
+      return this.skip();
+    } 
+
+    return undefined;
+  });
+
   beforeEach(async function () {
     ({
       userAddress,
       governorAddress,
       minterAddress,
       pcvControllerAddress,
     } = await getAddresses());
-    
+
     this.core = await getCore(true);
 
     this.fei = await Fei.at(await this.core.fei());