From 158288162938d342e2ab9314b63052021bb6686e Mon Sep 17 00:00:00 2001
From: Ricardo Guilherme Schmidt <3esmit@gmail.com>
Date: Sat, 19 May 2018 22:37:58 -0300
Subject: [PATCH] improve abstract erc20token test

---
 test/erc20token.js | 13 +++++--------
 test/testtoken.js  |  9 ++++-----
 2 files changed, 9 insertions(+), 13 deletions(-)

diff --git a/test/erc20token.js b/test/erc20token.js
index 7069c75..9c3d0ba 100644
--- a/test/erc20token.js
+++ b/test/erc20token.js
@@ -1,18 +1,15 @@
 
-exports.Test = (tokenContract, afterDeploy) => {
+exports.Test = (contractsConfig, afterDeploy) => {
   describe("ERC20Token", async function() {
     this.timeout(0);
     var ERC20Token;
     var accountsArr;
     before(function(done) {
-      var contractsConfig = {
-        "Token": tokenContract,
-        "ERC20Receiver": { }
-      };
+      contractsConfig["ERC20Receiver"] = {};
       EmbarkSpec.deployAll(contractsConfig, async function(accounts) { 
         ERC20Token = Token;
         accountsArr = accounts; 
-        afterDeploy(accounts, Token);
+        await afterDeploy(accounts, Token);
         done()
       });
     });
@@ -24,8 +21,8 @@ exports.Test = (tokenContract, afterDeploy) => {
       let result0 = await ERC20Token.methods.balanceOf(accountsArr[0]).call();
       let result1 = await ERC20Token.methods.balanceOf(accountsArr[1]).call();
       
-      assert.equal(result0, +initialBalance0-1);
-      assert.equal(result1, +initialBalance1+1);
+      assert.equal(result0, +initialBalance0-1, "account 0 balance unexpected");
+      assert.equal(result1, +initialBalance1+1, "account 1 balance unexpected");
     });
 
     it("should set approved amount", async function() {
diff --git a/test/testtoken.js b/test/testtoken.js
index d125da7..a8fad7b 100644
--- a/test/testtoken.js
+++ b/test/testtoken.js
@@ -11,9 +11,6 @@ describe("TestToken", async function() {
     };
     EmbarkSpec.deployAll(contractsConfig, async function(accounts) { 
       accountsArr = accounts
-      for(i=0;i<accountsArr.length;i++) {
-        await TestToken.methods.mint(7 * 10 ^ 18).send({from: accountsArr[i]})
-      }
       done() 
     });
   });
@@ -31,8 +28,10 @@ describe("TestToken", async function() {
     let result = await TestToken.methods.balanceOf(accountsArr[0]).call();
     assert.equal(result, +initialBalance+100);
   });
-
-  ERC20Token.Test({ "instanceOf" : "TestToken" }, async function (accounts, TestToken) {
+  var erc20tokenConfig = {
+    "Token": { "instanceOf" : "TestToken" }
+  }
+  ERC20Token.Test(erc20tokenConfig, async function (accounts, TestToken) {
     for(i=0;i<accounts.length;i++){
       await TestToken.methods.mint(7 * 10 ^ 18).send({from: accounts[i]})
     }