Add TEST_STAGE support for deploy scripts

Author: guo-yu

deploy/membership.js

@@ -1,14 +1,26 @@
+const { ethers } = require('hardhat');
 const { testArgs } = require('../utils/configs');
+const { setupProof, membershipMintAndDelegate, isLocalhost } = require('../utils/helpers');
 
-module.exports = async ({ getNamedAccounts, deployments }) => {
+module.exports = async ({ getNamedAccounts, deployments, getChainId }) => {
   const { deploy } = deployments;
   const { deployer } = await getNamedAccounts();
+  const chainId = await getChainId();
 
   await deploy('Membership', {
     from: deployer,
     args: testArgs(),
     log: true,
   });
+
+  if (isLocalhost(chainId)) {
+    const deps = {};
+    await setupProof(deps);
+
+    if (process.env.TEST_STAGE === 'MINT_READY') {
+      await membershipMintAndDelegate(await ethers.getContract('Membership'), deps);
+    }
+  }
 };
 
 module.exports.tags = ['Membership'];

package.json

@@ -5,6 +5,8 @@
   "main": "index.js",
   "scripts": {
     "start": "npx hardhat node --network hardhat --no-deploy",
+    "dev": "npx hardhat node --export-all ../website/contracts/localhost.json",
+    "dev:mint_ready": "TEST_STAGE=MINT_READY npx hardhat node --export-all ../website/contracts/localhost.json",
     "test": "npx hardhat test --network hardhat",
     "test:membership": "npx hardhat test --network hardhat tests/membership.js",
     "test:governor": "npx hardhat test --network hardhat tests/governor.js",