Add GrantableToken tests

2017-02-09 13:23:12 +01:00
parent fb0a96332c
commit 8d9e12eda3
6 changed files with 119 additions and 94 deletions
--- a/contracts/test-helpers/GrantableTokenMock.sol
+++ b/contracts/test-helpers/GrantableTokenMock.sol
@ -1,7 +1,11 @@
 pragma solidity ^0.4.4;
-import "./StandardTokenMock.sol";

-contract GrantableTokenMock is StandardTokenMock {
-  function GrantableTokenMock(address initialAccount, uint initialBalance)
-           StandardTokenMock(initialAccount, initialBalance) {}
+import '../token/GrantableToken.sol';
+
+// mock class using StandardToken
+contract GrantableTokenMock is GrantableToken {
+  function GrantableTokenMock(address initialAccount, uint initialBalance) {
+    balances[initialAccount] = initialBalance;
+    totalSupply = initialBalance;
+  }
 }