JavaScript Test Cases

TronBox uses the Mocha testing framework and Chai for assertions to provide you with a solid framework from which to write your JavaScript tests. Let's dive in and see how TronBox builds on top of Mocha to make testing your contracts a breeze.

NOTE: If you're unfamiliar with writing unit tests in Mocha, please see Mocha's documentation before continuing.

Use contract() instead of describe()

Structurally, your tests should remain largely unchanged from that of Mocha: Your tests should exist in the ./test directory, they should end with a .js extension, and they should contain code that Mocha will recognize as an automated test. What makes TronBox tests different from that of Mocha is the contract() function.

  • Before each contract() function is run, your contracts are redeployed to the running TRON network so the tests within it run with a clean contract state.
  • The contract() function provides a list of accounts made available by the current TRON network, which you can use to write tests.

Use contract abstractions within your tests

Contract abstractions are the basis for making TRON contract interaction possible from Javascript. Because TronBox has no way of detecting which contracts you'll need to interact with within your tests, you'll need to ask for those contracts explicitly. You do this by using the artifacts.require() method, a method provided by TronBox that allows you to request a usable contract abstraction for a specific Solidity contract. As you'll see in the example below, you can then use this abstraction to make sure your contracts are working properly.
For more information on using contract abstractions see the Interacting With Your Contracts section.

Using artifacts.require()

Using artifacts.require() within your tests works the same way as using it within your migrations; you just need to pass the name of the contract. See the artifacts.require()
documentation in the Migrations section for detailed usage.

Using tronWarp

A tronWarp instance is available in each test file, which can be referenced directly. So calling tronWrap.trx.getBalance(account) just works!

Examples

Using .then

Here's an example test provided in the MetaCoin TronBox Box using async/await notation. Note the use of the contract() function, the accounts array for specifying available TRON accounts, and our use of artifacts.require() for interacting directly with our contracts.

File: ./test/metacoin.js

const MetaCoin = artifacts.require('MetaCoin');
 
contract('MetaCoin', accounts => {
  it('should put 10000 MetaCoin in the first account', () =>
    MetaCoin.deployed()
      .then(instance => instance.getBalance.call(accounts[0]))
      .then(balance => {
        assert.equal(balance.valueOf(), 10000, "10000 wasn't in the first account");
      }));
 
  it('should call a function that depends on a linked library', () => {
    let meta;
    let metaCoinBalance;
    let metaCoinEthBalance;
 
    return MetaCoin.deployed()
      .then(instance => {
        meta = instance;
        return meta.getBalance.call(accounts[0]);
      })
      .then(outCoinBalance => {
        metaCoinBalance = outCoinBalance.toNumber();
        return meta.getBalanceInEth.call(accounts[0]);
      })
      .then(outCoinBalanceEth => {
        metaCoinEthBalance = outCoinBalanceEth.toNumber();
      })
      .then(() => {
        assert.equal(metaCoinEthBalance, 2 * metaCoinBalance, 'Library function returned unexpected function, linkage may be broken');
      });
  });
 
  it('should send coin correctly', () => {
    let meta;
 
    // Get initial balances of first and second account.
    const account_one = accounts[0];
    const account_two = accounts[1];
 
    let account_one_starting_balance;
    let account_two_starting_balance;
    let account_one_ending_balance;
    let account_two_ending_balance;
 
    const amount = 10;
 
    return MetaCoin.deployed()
      .then(instance => {
        meta = instance;
        return meta.getBalance.call(account_one);
      })
      .then(balance => {
        account_one_starting_balance = balance.toNumber();
        return meta.getBalance.call(account_two);
      })
      .then(balance => {
        account_two_starting_balance = balance.toNumber();
        return meta.sendCoin(account_two, amount, { from: account_one });
      })
      .then(() => meta.getBalance.call(account_one))
      .then(balance => {
        account_one_ending_balance = balance.toNumber();
        return meta.getBalance.call(account_two);
      })
      .then(balance => {
        account_two_ending_balance = balance.toNumber();
 
        assert.equal(account_one_ending_balance, account_one_starting_balance - amount, "Amount wasn't correctly taken from the sender");
        assert.equal(account_two_ending_balance, account_two_starting_balance + amount, "Amount wasn't correctly sent to the receiver");
      });
  });
});

This test will produce the following output:

 Contract: MetaCoin
    √ should put 10000 MetaCoin in the first account (83ms)
    √ should call a function that depends on a linked library (43ms)
    √ should send coin correctly (122ms)
 
 
  3 passing (293ms)

Use async/await

Below is a similar example using async/await:

const MetaCoin = artifacts.require('MetaCoin');
 
contract('2nd MetaCoin test', async accounts => {
  it('should put 10000 MetaCoin in the first account', async () => {
    let instance = await MetaCoin.deployed();
    let balance = await instance.getBalance.call(accounts[0]);
    assert.equal(balance.valueOf(), 10000);
  });
 
  it('should call a function that depends on a linked library', async () => {
    let meta = await MetaCoin.deployed();
    let outCoinBalance = await meta.getBalance.call(accounts[0]);
    let metaCoinBalance = outCoinBalance.toNumber();
    let outCoinBalanceEth = await meta.getBalanceInEth.call(accounts[0]);
    let metaCoinEthBalance = outCoinBalanceEth.toNumber();
    assert.equal(metaCoinEthBalance, 2 * metaCoinBalance);
  });
 
  it('should send coin correctly', async () => {
    // Get initial balances of first and second account.
    let account_one = accounts[0];
    let account_two = accounts[1];
 
    let amount = 10;
 
    let instance = await MetaCoin.deployed();
    let meta = instance;
 
    let balance = await meta.getBalance.call(account_one);
    let account_one_starting_balance = balance.toNumber();
 
    balance = await meta.getBalance.call(account_two);
    let account_two_starting_balance = balance.toNumber();
    await meta.sendCoin(account_two, amount, { from: account_one });
 
    balance = await meta.getBalance.call(account_one);
    let account_one_ending_balance = balance.toNumber();
 
    balance = await meta.getBalance.call(account_two);
    let account_two_ending_balance = balance.toNumber();
 
    assert.equal(account_one_ending_balance, account_one_starting_balance - amount, "Amount wasn't correctly taken from the sender");
    assert.equal(account_two_ending_balance, account_two_starting_balance + amount, "Amount wasn't correctly sent to the receiver");
  });
});

This test has the same output as the previous one.

Specifying tests

You can limit the tests to a specific file in the test directory as shown below:

tronbox test ./test/metacoin.js

For more information, please see the full Command Reference.