fix: forking URL interference with unit tests

Author: tamtamchik

.env.example

@@ -28,7 +28,7 @@ MAINNET_RPC_URL=http://localhost:8545
 
 # RPC URL for Hardhat Network forking, required for running tests on mainnet fork with tracing (Infura, Alchemy, etc.)
 # https://hardhat.org/hardhat-network/docs/guides/forking-other-networks#forking-other-networks
-HARDHAT_FORKING_URL=https://eth.drpc.org
+FORK_RPC_URL=https://eth.drpc.org
 
 # https://docs.lido.fi/deployed-contracts
 MAINNET_LOCATOR_ADDRESS=0xC1d0b3DE6792Bf6b4b37EccdcC24e45978Cfd2Eb

CONTRIBUTING.md

@@ -327,7 +327,7 @@ This is the most common method for running integration tests. It uses an instanc
 mainnet environment, allowing you to run integration tests with trace logging.
 
 > [!NOTE]
-> Ensure that `HARDHAT_FORKING_URL` is set to Ethereum Mainnet RPC and `MAINNET_*` environment variables are set in the
+> Ensure that `FORK_RPC_URL` is set to Ethereum Mainnet RPC and `MAINNET_*` environment variables are set in the
 > `.env` file (refer to `.env.example` for guidance). Otherwise, the tests will run against the Scratch deployment.
 
 ```bash

globals.d.ts

@@ -1,14 +1,17 @@
 declare namespace NodeJS {
   export interface ProcessEnv {
-    /* iternal logging verbosity (used in scratch deploy / integration tests) */
+    /* internal logging verbosity (used in scratch deploy / integration tests) */
     LOG_LEVEL?: "all" | "debug" | "info" | "warn" | "error" | "none"; // default: "info"
 
     /**
      * Flags for changing the behavior of the Hardhat Network
      */
 
-    /* RPC URL for Hardhat Network forking, required for running tests on mainnet fork with tracing */
-    HARDHAT_FORKING_URL?: string;
+    /* Test execution mode: 'scratch' for fresh network, 'fork' for forked network */
+    MODE?: "scratch" | "forking"; // default: "scratch"
+
+    /* URL of the network to fork from */
+    FORK_RPC_URL?: string; // default: "https://eth.drpc.org"
 
     /**
      * Flags for changing the behavior of the integration tests

hardhat.helpers.ts

@@ -2,16 +2,23 @@ import { existsSync, readFileSync } from "node:fs";
 
 /* Determines the forking configuration for Hardhat */
 export function getHardhatForkingConfig() {
-  const forkingUrl = process.env.HARDHAT_FORKING_URL || "";
+  const mode = process.env.MODE || "scratch";
 
-  if (!forkingUrl) {
-    // Scratch deploy, need to disable CSM
-    process.env.INTEGRATION_ON_SCRATCH = "on";
-    process.env.INTEGRATION_WITH_CSM = "off";
-    return undefined;
-  }
+  switch (mode) {
+    case "scratch":
+      process.env.INTEGRATION_ON_SCRATCH = "on";
+      process.env.INTEGRATION_WITH_CSM = "off";
+      return undefined;
+
+    case "forking":
+      if (!process.env.FORK_RPC_URL) {
+        throw new Error("FORK_RPC_URL must be set when MODE=forking");
+      }
+      return { url: process.env.FORK_RPC_URL };
 
-  return { url: forkingUrl };
+    default:
+      throw new Error("MODE must be either 'scratch' or 'forking'");
+  }
 }
 
 // TODO: this plaintext accounts.json private keys management is a subject

lib/protocol/context.ts

@@ -4,7 +4,6 @@ import hre from "hardhat";
 import { deployScratchProtocol, deployUpgrade, ether, findEventsWithInterfaces, impersonate, log } from "lib";
 
 import { discover } from "./discover";
-import { isNonForkingHardhatNetwork } from "./networks";
 import { provision } from "./provision";
 import { ProtocolContext, ProtocolContextFlags, ProtocolSigners, Signer } from "./types";
 
@@ -14,8 +13,11 @@ const getSigner = async (signer: Signer, balance = ether("100"), signers: Protoc
 };
 
 export const getProtocolContext = async (): Promise<ProtocolContext> => {
-  if (isNonForkingHardhatNetwork()) {
-    await deployScratchProtocol(hre.network.name);
+  if (hre.network.name === "hardhat") {
+    const networkConfig = hre.config.networks[hre.network.name];
+    if (!networkConfig.forking?.enabled) {
+      await deployScratchProtocol(hre.network.name);
+    }
   } else {
     await deployUpgrade(hre.network.name);
   }

package.json

@@ -22,12 +22,12 @@
     "test:trace": "hardhat test test/**/*.test.ts --trace --disabletracer",
     "test:fulltrace": "hardhat test test/**/*.test.ts --fulltrace --disabletracer",
     "test:watch": "hardhat watch",
-    "test:integration": "hardhat test test/integration/**/*.ts",
-    "test:integration:trace": "hardhat test test/integration/**/*.ts --trace --disabletracer",
-    "test:integration:fulltrace": "hardhat test test/integration/**/*.ts --fulltrace --disabletracer",
-    "test:integration:scratch": "HARDHAT_FORKING_URL= INTEGRATION_WITH_CSM=off INTEGRATION_WITH_SCRATCH_DEPLOY=on hardhat test test/integration/**/*.ts",
-    "test:integration:scratch:trace": "HARDHAT_FORKING_URL= INTEGRATION_WITH_CSM=off INTEGRATION_WITH_SCRATCH_DEPLOY=on hardhat test test/integration/**/*.ts --trace --disabletracer",
-    "test:integration:scratch:fulltrace": "HARDHAT_FORKING_URL= INTEGRATION_WITH_CSM=off INTEGRATION_WITH_SCRATCH_DEPLOY=on hardhat test test/integration/**/*.ts --fulltrace --disabletracer",
+    "test:integration": "MODE=forking hardhat test test/integration/**/*.ts",
+    "test:integration:trace": "MODE=forking hardhat test test/integration/**/*.ts --trace --disabletracer",
+    "test:integration:fulltrace": "MODE=forking hardhat test test/integration/**/*.ts --fulltrace --disabletracer",
+    "test:integration:scratch": "MODE=scratch hardhat test test/integration/**/*.ts",
+    "test:integration:scratch:trace": "MODE=scratch hardhat test test/integration/**/*.ts --trace --disabletracer",
+    "test:integration:scratch:fulltrace": "MODE=scratch hardhat test test/integration/**/*.ts --fulltrace --disabletracer",
     "test:integration:fork:local": "hardhat test test/integration/**/*.ts --network local",
     "test:integration:fork:mainnet": "hardhat test test/integration/**/*.ts --network mainnet-fork",
     "test:integration:fork:mainnet:custom": "hardhat test --network mainnet-fork",