From dc2c8130f33047210cd0c2f076f82e15aa6f42e5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?R=C3=A2u=20Cao?= <raucao@kip.pe>
Date: Wed, 24 Aug 2022 13:33:04 +0100
Subject: [PATCH] Add tests for Contributor#add()

---
 test/contracts/Contributor.js | 55 ++++++++++++++++++++++++++++++-----
 1 file changed, 47 insertions(+), 8 deletions(-)

diff --git a/test/contracts/Contributor.js b/test/contracts/Contributor.js
index d4f2716..e340aa2 100644
--- a/test/contracts/Contributor.js
+++ b/test/contracts/Contributor.js
@@ -1,23 +1,15 @@
 const { expect } = require("chai");
 const { ethers, upgrades } = require("hardhat");
-// const { anyValue } = require("@nomicfoundation/hardhat-chai-matchers/withArgs");
 let owner, addr1, addr2, addr3, addr4, addr5, addr6, addr7;
 let Contribution, Contributor;
 
 describe("Contributor contract", async function () {
   before(async function () {
     [owner, addr1, addr2, addr3, addr4, addr5, addr6, addr7] = await ethers.getSigners();
-    // let accounts = [owner, addr1, addr2, addr3, addr4, addr5, addr6, addr7];
-    // const contributorFactory = await ethers.getContractFactory("Contributor");
-    // Contributor = await upgrades.deployProxy(contributorFactory);
-    // for (const account of accounts) {
-    //   await Contributor.addContributor(account.address, "0x99b8afd7b266e19990924a8be9099e81054b70c36b20937228a77a5cf75723b8", 18, 32);
-    // }
   });
 
   describe("initialize()", function () {
     before(async function () {
-      // const [owner] = await ethers.getSigners();
       const contributorFactory = await ethers.getContractFactory("Contributor");
       Contributor = await upgrades.deployProxy(contributorFactory);
     });
@@ -27,4 +19,51 @@ describe("Contributor contract", async function () {
       expect(await Contributor.deployer()).to.not.equal(addr1.address);
     });
   });
+
+  describe("add()", function () {
+    before(async function () {
+      const contributorFactory = await ethers.getContractFactory("Contributor");
+      Contributor = await upgrades.deployProxy(contributorFactory);
+
+      let accounts = [owner, addr1, addr2, addr3, addr4, addr5, addr6, addr7];
+      for (const account of accounts) {
+        await Contributor.addContributor(account.address, "0x99b8afd7b266e19990924a8be9099e81054b70c36b20937228a77a5cf75723b8", 18, 32);
+      }
+    });
+
+    it("does not allow random accounts to create a contributor profile", async function () {
+      await expect(Contributor.connect(addr7).addContributor(
+        "0x608FD4b95116Ea616990Aaeb1d4f1ce07612f261",
+        "0x1d9de6de5c72eedca6d7a5e8a9159e2f5fe676506aece3000acefcc821723429",
+        18, 32
+      )).to.be.revertedWith("Core only");
+      expect(await Contributor.contributorsCount()).to.equal(8);
+    });
+
+    it("allows core contributors to create a contributor profile", async function () {
+      await Contributor.connect(addr1).addContributor(
+        "0x608FD4b95116Ea616990Aaeb1d4f1ce07612f261",
+        "0x1d9de6de5c72eedca6d7a5e8a9159e2f5fe676506aece3000acefcc821723429",
+        18, 32
+      )
+      expect(await Contributor.contributorsCount()).to.equal(9);
+    });
+
+    it("does not allow to create accounts with an existing address", async function () {
+      await expect(Contributor.connect(addr1).addContributor(
+        "0x608FD4b95116Ea616990Aaeb1d4f1ce07612f261",
+        "0x1d9de6de5c72eedca6d7a5e8a9159e2f5fe676506aece3000acefcc821723429",
+        18, 32
+      )).to.be.revertedWith("Address already in use");
+      expect(await Contributor.contributorsCount()).to.equal(9);
+    });
+
+    it("emits a ContributorAdded event", async function () {
+      await expect(Contributor.connect(addr1).addContributor(
+        "0x765E88b4F9a59C3a3b300C6eFF9E6E9fDDf9FbD9",
+        "0xcfbeeadc244dfdc55bbad50d431871439df067970db84c73023956c96a6f5df2",
+        18, 32
+      )).to.emit(Contributor, "ContributorAdded").withArgs(10, "0x765E88b4F9a59C3a3b300C6eFF9E6E9fDDf9FbD9");
+    });
+  });
 });