TokenDestructible.test.js 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839
  1. const { ethGetBalance } = require('../helpers/web3');
  2. const TokenDestructible = artifacts.require('TokenDestructible');
  3. const ERC20Mock = artifacts.require('ERC20Mock');
  4. const BigNumber = web3.BigNumber;
  5. require('chai')
  6. .use(require('chai-bignumber')(BigNumber))
  7. .should();
  8. contract('TokenDestructible', function ([_, owner]) {
  9. let tokenDestructible;
  10. beforeEach(async function () {
  11. tokenDestructible = await TokenDestructible.new({
  12. from: owner,
  13. value: web3.toWei('10', 'ether'),
  14. });
  15. });
  16. it('should send balance to owner after destruction', async function () {
  17. const initBalance = await ethGetBalance(owner);
  18. await tokenDestructible.destroy([], { from: owner });
  19. const newBalance = await ethGetBalance(owner);
  20. newBalance.should.be.bignumber.gt(initBalance);
  21. });
  22. it('should send tokens to owner after destruction', async function () {
  23. const token = await ERC20Mock.new(tokenDestructible.address, 100);
  24. (await token.balanceOf(tokenDestructible.address)).should.be.bignumber.equal(100);
  25. (await token.balanceOf(owner)).should.be.bignumber.equal(0);
  26. await tokenDestructible.destroy([token.address], { from: owner });
  27. (await token.balanceOf(tokenDestructible.address)).should.be.bignumber.equal(0);
  28. (await token.balanceOf(owner)).should.be.bignumber.equal(100);
  29. });
  30. });