feat: baklava fork tests

Author: baroooo

.env.example

@@ -2,4 +2,5 @@ FRACTAL_CLIENT_ID=
 MERKLE_PROOF=
 AIRGRAB_TESTER=
 AIRGRAB_TESTER_PK=
-RESTRICTED_COUNTRIES=
+RESTRICTED_COUNTRIES=
+DEPLOYER=

.github/workflows/main.yml

@@ -32,10 +32,11 @@ jobs:
         run: pnpm lint
 
       - name: Start node & test
-        run: pnpm test:alfajores
+        run: pnpm test
         env:
           FRACTAL_CLIENT_ID: 'I28eze1-UTsP66nYyBDgiRrGCA7Oe6cRFkl7Gk0Gr1A'
           MERKLE_PROOF: ${{secrets.MERKLE_PROOF}}
           AIRGRAB_TESTER: ${{secrets.AIRGRAB_TESTER}}
           AIRGRAB_TESTER_PK: ${{secrets.AIRGRAB_TESTER_PK}}
           RESTRICTED_COUNTRIES: 'CD,CU,GB,IR,KP,ML,MM,SS,SY,US,YE'
+          DEPLOYER: '0x56fd3f2bee130e9867942d0f463a16fbe49b8d81'

pnpm-lock.yaml

@@ -40,11 +40,6 @@ describe('Airgrab', function () {
     // reset the fork state between tests to not pollute the state
     // @ts-expect-error - forking doesn't exist in hre for some reason
     await helpers.reset(hre.network.config.forking.url);
-
-    // if this chain id is mainnet skip the test.
-    if (hre.network.config.chainId === networks.celo.chainId) {
-      this.skip();
-    }
   });
 
   before(async function () {
@@ -53,6 +48,14 @@ describe('Airgrab', function () {
       throw new Error('Chain ID not found');
     }
 
+    // if chain is celo or baklava skip tests
+    if (
+      chainId === networks.celo.chainId ||
+      chainId === networks.baklava.chainId
+    ) {
+      this.skip();
+    }
+
     governanceAddresses = mento.addresses[chainId]!;
     if (!governanceAddresses) {
       throw new Error('Governance addresses not found for this chain');
@@ -82,6 +85,11 @@ describe('Airgrab', function () {
       provider as any,
     );
 
+    // if airdrop is finished skip tests
+    if ((await airgrab.endTimestamp()) < (await helpers.time.latest())) {
+      this.skip();
+    }
+
     console.log('\r\n======================================================');
     console.log('Running Airgrab tests on network with chain id:', chainId);
     console.log(

test/Airgrab.test.ts

@@ -9,6 +9,7 @@ import {
   MentoToken,
   MentoToken__factory,
 } from '@mento-protocol/mento-core-ts';
+import { networks } from '../config';
 
 describe('Emission Contract', function () {
   const { provider } = ethers;
@@ -45,6 +46,11 @@ describe('Emission Contract', function () {
       throw new Error('Chain ID not found');
     }
 
+    // if chain is baklava skip tests
+    if (chainId === networks.baklava.chainId) {
+      this.skip();
+    }
+
     governanceAddresses = mento.addresses[chainId]!;
     if (!governanceAddresses) {
       throw new Error('Governance addresses not found for this chain');

test/Emission.test.ts

@@ -15,8 +15,6 @@ import {
   TimelockController__factory,
   GovernanceFactory,
   GovernanceFactory__factory,
-  Emission,
-  Emission__factory,
 } from '@mento-protocol/mento-core-ts';
 import { ProxyAdmin } from '../typechain-types/@openzeppelin/contracts/proxy/transparent';
 import { ProxyAdmin__factory } from '../typechain-types/factories/@openzeppelin/contracts/proxy/transparent';
@@ -36,7 +34,6 @@ describe('Governance', function () {
   } = ethers;
 
   let governanceAddresses: mento.ContractAddresses;
-  let emission: Emission;
   let mentoToken: MentoToken;
   let locking: Locking;
   let governor: MentoGovernor;
@@ -410,10 +407,9 @@ describe('Governance', function () {
     const newLocking = await deployContract('MockLocking');
     const newTimelock = await deployContract('MockTimelock');
     const newGovernor = await deployContract('MockGovernor');
-    const newEmission = await deployContract('MockEmission');
 
-    const targets = Array(4).fill(proxyAdmin.target);
-    const values = Array(4).fill(0);
+    const targets = Array(3).fill(proxyAdmin.target);
+    const values = Array(3).fill(0);
     const calldatas = [];
     const description = 'Upgrade upgradable contracts';
 
@@ -438,13 +434,6 @@ describe('Governance', function () {
       ]),
     );
 
-    calldatas.push(
-      proxyAdmin.interface.encodeFunctionData('upgrade', [
-        emission.target,
-        newEmission.target,
-      ]),
-    );
-
     // Create a proposal to upgrade contracts
     const proposalId = submitProposal(
       governanceAddresses,
@@ -472,7 +461,6 @@ describe('Governance', function () {
     await locking.getWeek();
     await timelock.getMinDelay();
     await governor.votingDelay();
-    await emission.calculateEmission();
 
     await governor.connect(alice)['execute(uint256)'](proposalId);
 
@@ -486,9 +474,6 @@ describe('Governance', function () {
     await expect(governor.votingDelay()).to.be.revertedWith(
       'MockGovernor: votingDelay not implemented',
     );
-    await expect(emission.calculateEmission()).to.be.revertedWith(
-      'MockEmission: calculateEmission not implemented',
-    );
   });
 
   const setupEnvironment = async (): Promise<number> => {
@@ -507,12 +492,6 @@ describe('Governance', function () {
       throw new Error('Governance addresses not found for this chain');
     }
 
-    emission = Emission__factory.connect(
-      governanceAddresses.Emission,
-      // eslint-disable-next-line @typescript-eslint/no-explicit-any
-      provider as any,
-    );
-
     mentoToken = MentoToken__factory.connect(
       governanceAddresses.MentoToken,
       // eslint-disable-next-line @typescript-eslint/no-explicit-any

test/Governance.test.ts

@@ -124,7 +124,9 @@ describe('Mento Token', function () {
 
   it('should successfully mint when called by the Emission contract', async function () {
     const emissionSigner = await ethers.getImpersonatedSigner(
-      mentoAddresses.Emission,
+      mentoAddresses.Emission === '0xNotDeployed'
+        ? process.env.DEPLOYER!
+        : mentoAddresses.Emission,
     );
     const [receiver] = await ethers.getSigners();
     const supplyBefore = await mentoToken.totalSupply();
@@ -139,7 +141,9 @@ describe('Mento Token', function () {
 
   it('should allow for tokens to be transferred by anyone when unpaused', async function () {
     const emissionSigner = await ethers.getImpersonatedSigner(
-      mentoAddresses.Emission,
+      mentoAddresses.Emission === '0xNotDeployed'
+        ? process.env.DEPLOYER!
+        : mentoAddresses.Emission,
     );
     const [bob, alice] = await ethers.getSigners();
     const supplyBefore = await mentoToken.totalSupply();
@@ -172,7 +176,9 @@ describe('Mento Token', function () {
 
   it('should allow for tokens to be transferred by owner,locking and emission when paused', async function () {
     const addresses = [
-      mentoAddresses.Emission,
+      mentoAddresses.Emission === '0xNotDeployed'
+        ? process.env.DEPLOYER!
+        : mentoAddresses.Emission,
       mentoAddresses.Locking,
       mentoAddresses.TimelockController,
     ];
@@ -201,7 +207,9 @@ describe('Mento Token', function () {
 
   it('should allow for tokens to be burned by anyone when unpaused', async function () {
     const emissionSigner = await ethers.getImpersonatedSigner(
-      mentoAddresses.Emission,
+      mentoAddresses.Emission === '0xNotDeployed'
+        ? process.env.DEPLOYER!
+        : mentoAddresses.Emission,
     );
     const [bob] = await ethers.getSigners();
     const supplyBefore = await mentoToken.totalSupply();

test/MentoToken.test.ts

@@ -42,9 +42,13 @@ export const setUpTestAccounts = async (
   mentoAddresses: mento.ContractAddresses,
   mintAmount: string = '10000000',
 ): Promise<void> => {
+  // if the Emission contract is not deployed on a chain, emitter is deployer
   const emissionSigner = await ethers.getImpersonatedSigner(
-    mentoAddresses.Emission,
+    mentoAddresses.Emission === '0xNotDeployed'
+      ? process.env.DEPLOYER!
+      : mentoAddresses.Emission,
   );
+
   const mentoToken = MentoToken__factory.connect(
     mentoAddresses.MentoToken,
     // eslint-disable-next-line @typescript-eslint/no-explicit-any