From 7b16841534f86a1f860b672b088f7a12c04559f5 Mon Sep 17 00:00:00 2001
From: Mark Spanbroek <mark@spanbroek.net>
Date: Tue, 16 Nov 2021 14:25:42 +0100
Subject: [PATCH] Change test token to allow arbitrary minting

---
 contracts/TestToken.sol | 6 ++++--
 deploy/token.js         | 2 +-
 test/Stakes.test.js     | 3 ++-
 test/Storage.test.js    | 3 ++-
 4 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/contracts/TestToken.sol b/contracts/TestToken.sol
index 6ec95f6..a2d1725 100644
--- a/contracts/TestToken.sol
+++ b/contracts/TestToken.sol
@@ -4,9 +4,11 @@ pragma solidity ^0.8.0;
 import "@openzeppelin/contracts/token/ERC20/ERC20.sol";
 
 contract TestToken is ERC20 {
-  constructor(address[] memory holders) ERC20("TestToken", "TST") {
+  constructor() ERC20("TestToken", "TST") {}
+
+  function mint(address[] memory holders, uint amount) public {
     for (uint i=0; i<holders.length; i++) {
-      _mint(holders[i], 1000);
+      _mint(holders[i], amount);
     }
   }
 }
diff --git a/deploy/token.js b/deploy/token.js
index adf4bcf..76b271c 100644
--- a/deploy/token.js
+++ b/deploy/token.js
@@ -1,6 +1,6 @@
 module.exports = async ({deployments, getNamedAccounts}) => {
   const { deployer } = await getNamedAccounts()
-  await deployments.deploy('TestToken', { args: [[deployer]], from: deployer })
+  await deployments.deploy('TestToken', { from: deployer })
 }
 
 module.exports.tags = ['TestToken']
diff --git a/test/Stakes.test.js b/test/Stakes.test.js
index 429277b..70be394 100644
--- a/test/Stakes.test.js
+++ b/test/Stakes.test.js
@@ -11,8 +11,9 @@ describe("Stakes", function () {
     [host] = await ethers.getSigners()
     const Stakes = await ethers.getContractFactory("TestStakes")
     const TestToken = await ethers.getContractFactory("TestToken")
-    token = await TestToken.deploy([host.address])
+    token = await TestToken.deploy()
     stakes = await Stakes.deploy(token.address)
+    await token.mint([host.address], 1000)
   })
 
   it("has zero stakes initially", async function () {
diff --git a/test/Storage.test.js b/test/Storage.test.js
index b34ab82..33c547b 100644
--- a/test/Storage.test.js
+++ b/test/Storage.test.js
@@ -20,10 +20,11 @@ describe("Storage", function () {
     [client, host] = await ethers.getSigners()
     let Token = await ethers.getContractFactory("TestToken")
     let StorageContracts = await ethers.getContractFactory("Storage")
-    token = await Token.deploy([client.address, host.address])
+    token = await Token.deploy()
     storage = await StorageContracts.deploy(
       token.address, stakeAmount, slashMisses, slashPercentage
     )
+    await token.mint([client.address, host.address], 1000)
   })
 
   describe("creating a new storage contract", function () {