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 version of StandardToken, with no allowances
|
||||
*/
|
||||
contract BasicToken is ERC20Lite, SafeMath {
|
||||
contract BasicToken is ERC20Basic, SafeMath {
|
||||
|
||||
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