Add unit test for DioneDispute contract

eth-contracts/package.json

@@ -11,11 +11,16 @@
   "license": "ISC",
   "devDependencies": {
     "@nomiclabs/hardhat-ethers": "^2.0.2",
+    "@nomiclabs/hardhat-waffle": "^2.0.1",
     "@types/chai": "^4.2.16",
     "@types/mocha": "^8.2.2",
     "@types/node": "^14.14.39",
     "chai": "^4.3.4",
+    "ethereum-waffle": "^3.3.0",
+    "ethers": "^5.1.3",
     "hardhat": "^2.1.2",
+    "hardhat-ethernal": "^0.2.3",
+    "hardhat-tracer": "*",
     "ts-node": "^9.1.1",
     "typescript": "^4.2.4"
   },

eth-contracts/test/DioneDispute.ts

@@ -0,0 +1,104 @@
+import { ethers } from "hardhat";
+import { Signer, Contract } from "ethers";
+import { expect } from "chai";
+import { soliditySha3 } from "web3-utils";
+
+describe("DioneDispute", function () {
+  let dioneDispute: Contract;
+  let dioneStaking: Contract;
+
+  beforeEach(async function () {
+    const DioneToken = await ethers.getContractFactory("DioneToken");
+    const DioneOracle = await ethers.getContractFactory("DioneOracle");
+    const DioneDispute = await ethers.getContractFactory("DioneDispute");
+    const DioneStaking = await ethers.getContractFactory("DioneStaking");
+    const Mediator = await ethers.getContractFactory("Mediator");
+
+    const dioneToken = await DioneToken.deploy();
+    await dioneToken.deployed();
+    console.log("DioneToken deployed to:", dioneToken.address);
+
+    const _dioneStaking = await DioneStaking.deploy(dioneToken.address, ethers.constants.WeiPerEther.mul(100), 0, ethers.constants.WeiPerEther.mul(5000));
+    await _dioneStaking.deployed();
+    console.log("staking_contract_address = \"" + _dioneStaking.address+ "\"");
+
+    const _dioneDispute = await DioneDispute.deploy(_dioneStaking.address, 2);
+    await _dioneDispute.deployed();
+    console.log("dispute_contract_address = \"" + _dioneDispute.address+ "\"");
+
+    const dioneOracle = await DioneOracle.deploy(_dioneStaking.address);
+    await dioneOracle.deployed();
+    console.log("oracle_contract_address = \"" + dioneOracle.address+ "\"");
+
+    const mediator = await Mediator.deploy(dioneOracle.address);
+    await mediator.deployed();
+    console.log("mediator_contract_address = \"" + mediator.address +"\"")
+
+    await _dioneStaking.setOracleContractAddress(dioneOracle.address);
+    await _dioneStaking.setDisputeContractAddress(_dioneDispute.address);
+
+    const addresses = ["0xf39fd6e51aad88f6f4ce6ab8827279cfffb92266", "0x70997970c51812dc3a010c7d01b50e0d17dc79c8", "0x3C44CdDdB6a900fa2b585dd299e03d12FA4293BC", "0x90F79bf6EB2c4f870365E785982E1f101E93b906"]
+    await dioneToken.mint(addresses[0], ethers.constants.WeiPerEther.mul(36000));
+    for (const address of addresses) {
+      if(address == "0xf39fd6e51aad88f6f4ce6ab8827279cfffb92266") {
+        continue;
+      }
+      await dioneToken.transfer(address, ethers.constants.WeiPerEther.mul(9000));
+    }
+
+    await dioneToken.transferOwnership(_dioneStaking.address);
+
+    const signers = await ethers.getSigners();
+    for (var i = 0; i < addresses.length; i++) {
+      const staking = _dioneStaking.connect(signers[i]);
+      const token = dioneToken.connect(signers[i]);
+      await token.approve(_dioneStaking.address, ethers.constants.WeiPerEther.mul(9000));
+      await staking.stake(ethers.constants.WeiPerEther.mul(9000));
+      const stake = await _dioneStaking.minerStake(addresses[i]);
+      console.log(addresses[i], stake.toString());
+    }
+
+    dioneDispute = _dioneDispute;
+    dioneStaking = _dioneStaking;
+  });
+
+  it("should create dispute, vote it by various eth addresses and then finish it after 2 secs", async function () {
+    const [owner, addr1, addr2, addr3] = await ethers.getSigners();
+   
+    const dhash = soliditySha3(addr1.address, 1);
+
+    await expect(dioneDispute.beginDispute(addr1.address, 1))
+      .to.emit(dioneDispute, 'NewDispute')
+      .withArgs(dhash, 1, addr1.address, owner.address);
+
+    await expect(dioneDispute.connect(addr2).vote(dhash, true))
+      .to.emit(dioneDispute, 'NewVote')
+      .withArgs(dhash, addr2.address);
+
+    await expect(dioneDispute.connect(addr3).vote(dhash, true))
+      .to.emit(dioneDispute, 'NewVote')
+      .withArgs(dhash, addr3.address);
+
+    await delay(2000);
+
+    await expect(dioneDispute.finishDispute(dhash))
+      .to.emit(dioneDispute, 'DisputeFinished')
+      .withArgs(dhash, true);
+
+    expect(await dioneStaking.minerStake(addr1.address))
+      .to.equal(0);
+
+    expect(await dioneStaking.minerStake(addr2.address))
+      .to.equal(ethers.constants.WeiPerEther.mul(12000));
+      
+    expect(await dioneStaking.minerStake(addr3.address))
+      .to.equal(ethers.constants.WeiPerEther.mul(12000));
+
+    expect(await dioneStaking.minerStake(owner.address))
+      .to.equal(ethers.constants.WeiPerEther.mul(12000));
+  });
+});
+
+function delay(ms: number): Promise<void> {
+    return new Promise( resolve => setTimeout(resolve, ms) );
+}