HasNoTokens.test.js 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839
  1. const { expectThrow } = require('../helpers/expectThrow');
  2. const HasNoTokens = artifacts.require('HasNoTokens');
  3. const ERC223TokenMock = artifacts.require('ERC223TokenMock');
  4. contract('HasNoTokens', function (accounts) {
  5. let hasNoTokens = null;
  6. let token = null;
  7. beforeEach(async () => {
  8. // Create contract and token
  9. hasNoTokens = await HasNoTokens.new();
  10. token = await ERC223TokenMock.new(accounts[0], 100);
  11. // Force token into contract
  12. await token.transfer(hasNoTokens.address, 10);
  13. const startBalance = await token.balanceOf(hasNoTokens.address);
  14. assert.equal(startBalance, 10);
  15. });
  16. it('should not accept ERC223 tokens', async function () {
  17. await expectThrow(token.transferERC223(hasNoTokens.address, 10, ''));
  18. });
  19. it('should allow owner to reclaim tokens', async function () {
  20. const ownerStartBalance = await token.balanceOf(accounts[0]);
  21. await hasNoTokens.reclaimToken(token.address);
  22. const ownerFinalBalance = await token.balanceOf(accounts[0]);
  23. const finalBalance = await token.balanceOf(hasNoTokens.address);
  24. assert.equal(finalBalance, 0);
  25. assert.equal(ownerFinalBalance - ownerStartBalance, 10);
  26. });
  27. it('should allow only owner to reclaim tokens', async function () {
  28. await expectThrow(
  29. hasNoTokens.reclaimToken(token.address, { from: accounts[1] }),
  30. );
  31. });
  32. });