add simple tests for BasicToken
This commit is contained in:
12
contracts/test-helpers/BasicTokenMock.sol
Normal file
12
contracts/test-helpers/BasicTokenMock.sol
Normal file
@ -0,0 +1,12 @@
|
|||||||
|
pragma solidity ^0.4.4;
|
||||||
|
import '../token/BasicToken.sol';
|
||||||
|
|
||||||
|
// mock class using BasicToken
|
||||||
|
contract BasicTokenMock is BasicToken {
|
||||||
|
|
||||||
|
function BasicTokenMock(address initialAccount, uint initialBalance) {
|
||||||
|
balances[initialAccount] = initialBalance;
|
||||||
|
totalSupply = initialBalance;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
@ -7,7 +7,7 @@ import '../SafeMath.sol';
|
|||||||
* Basic token
|
* Basic token
|
||||||
* Basic version of StandardToken, with no allowances
|
* Basic version of StandardToken, with no allowances
|
||||||
*/
|
*/
|
||||||
contract BasicToken is ERC20Lite, SafeMath {
|
contract BasicToken is ERC20Basic, SafeMath {
|
||||||
|
|
||||||
mapping(address => uint) balances;
|
mapping(address => uint) balances;
|
||||||
|
|
||||||
|
|||||||
49
test/BasicToken.js
Normal file
49
test/BasicToken.js
Normal file
@ -0,0 +1,49 @@
|
|||||||
|
contract('BasicToken', function(accounts) {
|
||||||
|
|
||||||
|
it("should return the correct totalSupply after construction", function(done) {
|
||||||
|
return BasicTokenMock.new(accounts[0], 100)
|
||||||
|
.then(function(token) {
|
||||||
|
return token.totalSupply();
|
||||||
|
})
|
||||||
|
.then(function(totalSupply) {
|
||||||
|
assert.equal(totalSupply, 100);
|
||||||
|
})
|
||||||
|
.then(done);
|
||||||
|
})
|
||||||
|
|
||||||
|
it("should return correct balances after transfer", function(done) {
|
||||||
|
var token;
|
||||||
|
return BasicTokenMock.new(accounts[0], 100)
|
||||||
|
.then(function(_token) {
|
||||||
|
token = _token;
|
||||||
|
return token.transfer(accounts[1], 100);
|
||||||
|
})
|
||||||
|
.then(function() {
|
||||||
|
return token.balanceOf(accounts[0]);
|
||||||
|
})
|
||||||
|
.then(function(balance) {
|
||||||
|
assert.equal(balance, 0);
|
||||||
|
})
|
||||||
|
.then(function() {
|
||||||
|
return token.balanceOf(accounts[1]);
|
||||||
|
})
|
||||||
|
.then(function(balance) {
|
||||||
|
assert.equal(balance, 100);
|
||||||
|
})
|
||||||
|
.then(done);
|
||||||
|
});
|
||||||
|
|
||||||
|
it("should throw an error when trying to transfer more than balance", function(done) {
|
||||||
|
var token;
|
||||||
|
return BasicTokenMock.new(accounts[0], 100)
|
||||||
|
.then(function(_token) {
|
||||||
|
token = _token;
|
||||||
|
return token.transfer(accounts[1], 101);
|
||||||
|
})
|
||||||
|
.catch(function(error) {
|
||||||
|
if (error.message.search('invalid JUMP') == -1) throw error
|
||||||
|
})
|
||||||
|
.then(done);
|
||||||
|
});
|
||||||
|
|
||||||
|
});
|
||||||
Reference in New Issue
Block a user