SushiToken.test.ts

import { ethers } from "hardhat";
import { expect } from "chai";

describe("SushiToken", function () {
  before(async function () {
    this.SushiToken = await ethers.getContractFactory("SushiToken")
    this.signers = await ethers.getSigners()
    this.alice = this.signers[0]
    this.bob = this.signers[1]
    this.carol = this.signers[2]
  })

  beforeEach(async function () {
    this.sushi = await this.SushiToken.deploy()
    await this.sushi.deployed()
  })

  it("should have correct name and symbol and decimal", async function () {
    const name = await this.sushi.name()
    const symbol = await this.sushi.symbol()
    const decimals = await this.sushi.decimals()
    expect(name, "SushiToken")
    expect(symbol, "SUSHI")
    expect(decimals, "18")
  })

  it("should only allow owner to mint token", async function () {
    await this.sushi.mint(this.alice.address, "100")
    await this.sushi.mint(this.bob.address, "1000")
    await expect(this.sushi.connect(this.bob).mint(this.carol.address, "1000", { from: this.bob.address })).to.be.revertedWith(
      "Ownable: caller is not the owner"
    )
    const totalSupply = await this.sushi.totalSupply()
    const aliceBal = await this.sushi.balanceOf(this.alice.address)
    const bobBal = await this.sushi.balanceOf(this.bob.address)
    const carolBal = await this.sushi.balanceOf(this.carol.address)
    expect(totalSupply).to.equal("1100")
    expect(aliceBal).to.equal("100")
    expect(bobBal).to.equal("1000")
    expect(carolBal).to.equal("0")
  })

  it("should supply token transfers properly", async function () {
    await this.sushi.mint(this.alice.address, "100")
    await this.sushi.mint(this.bob.address, "1000")
    await this.sushi.transfer(this.carol.address, "10")
    await this.sushi.connect(this.bob).transfer(this.carol.address, "100", {
      from: this.bob.address,
    })
    const totalSupply = await this.sushi.totalSupply()
    const aliceBal = await this.sushi.balanceOf(this.alice.address)
    const bobBal = await this.sushi.balanceOf(this.bob.address)
    const carolBal = await this.sushi.balanceOf(this.carol.address)
    expect(totalSupply, "1100")
    expect(aliceBal, "90")
    expect(bobBal, "900")
    expect(carolBal, "110")
  })

  it("should fail if you try to do bad transfers", async function () {
    await this.sushi.mint(this.alice.address, "100")
    await expect(this.sushi.transfer(this.carol.address, "110")).to.be.revertedWith("ERC20: transfer amount exceeds balance")
    await expect(this.sushi.connect(this.bob).transfer(this.carol.address, "1", { from: this.bob.address })).to.be.revertedWith(
      "ERC20: transfer amount exceeds balance"
    )
  })
})