From afad0e49ece26d5316f68b1d93786c87bdd75e43 Mon Sep 17 00:00:00 2001
From: Mark Spanbroek <mark@spanbroek.net>
Date: Thu, 4 Nov 2021 11:32:21 +0100
Subject: [PATCH] Mint test tokens for both client and host

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

diff --git a/contracts/TestToken.sol b/contracts/TestToken.sol
index 9d59ecc..6ec95f6 100644
--- a/contracts/TestToken.sol
+++ b/contracts/TestToken.sol
@@ -4,7 +4,9 @@ pragma solidity ^0.8.0;
 import "@openzeppelin/contracts/token/ERC20/ERC20.sol";
 
 contract TestToken is ERC20 {
-  constructor() ERC20("TestToken", "TST") {
-    _mint(msg.sender, 1000);
+  constructor(address[] memory holders) ERC20("TestToken", "TST") {
+    for (uint i=0; i<holders.length; i++) {
+      _mint(holders[i], 1000);
+    }
   }
 }
diff --git a/test/Stakes.test.js b/test/Stakes.test.js
index 6458132..03f7adc 100644
--- a/test/Stakes.test.js
+++ b/test/Stakes.test.js
@@ -11,7 +11,7 @@ describe("Stakes", function () {
     [host] = await ethers.getSigners()
     const Stakes = await ethers.getContractFactory("TestStakes")
     const TestToken = await ethers.getContractFactory("TestToken")
-    token = await TestToken.deploy()
+    token = await TestToken.deploy([host.address])
     stakes = await Stakes.deploy(token.address)
   })
 
diff --git a/test/Storage.test.js b/test/Storage.test.js
index e9ac522..8bd6b15 100644
--- a/test/Storage.test.js
+++ b/test/Storage.test.js
@@ -18,7 +18,7 @@ describe("Storage", function () {
     [client, host] = await ethers.getSigners()
     let Token = await ethers.getContractFactory("TestToken")
     let StorageContracts = await ethers.getContractFactory("Storage")
-    token = await Token.connect(host).deploy()
+    token = await Token.deploy([client.address, host.address])
     storage = await StorageContracts.deploy(token.address, stakeAmount)
   })