Write a test case using a framework like Truffle or Hardhat for a Solidity function that calculates the total supply of a custom ERC20 token.