import { expect } from 'chai'
// @ts-ignore-next-line
import { ethers } from 'hardhat'

let mockERC20Upgradeable: any
let deployer: any
let user1: any
let user2: any

describe('MockERC20UpgradeableUpgradeable', function () {
    beforeEach(async function () {
        ;[deployer, user1, user2] = await ethers.getSigners()

        const MockERC20Upgradeable = await ethers.getContractFactory('MockERC20Upgradeable')
        mockERC20Upgradeable = await MockERC20Upgradeable.deploy()
        await mockERC20Upgradeable.deployed()
        await mockERC20Upgradeable.initialize('MockERC20Upgradeable', 'MOCK')
    })

    it('Should return the name of the token', async function () {
        expect(await mockERC20Upgradeable.name()).to.equal('MockERC20Upgradeable')
    })

    it('Should return the symbol of the token', async function () {
        expect(await mockERC20Upgradeable.symbol()).to.equal('MOCK')
    })

    it('Should mint token and have the right balanceOf', async function () {
        const amount = 1000
        await mockERC20Upgradeable.mint(deployer.address, amount)

        expect(await mockERC20Upgradeable.balanceOf(deployer.address)).to.equal(amount)
    })

    it('Should mint token and have the right totalSupply', async function () {
        const amount = 1000
        await mockERC20Upgradeable.mint(deployer.address, amount)

        expect(await mockERC20Upgradeable.totalSupply()).to.equal(amount)
    })

    it('Should mint token and burn them', async function () {
        const amount = 1000
        await mockERC20Upgradeable.mint(deployer.address, amount)
        expect(await mockERC20Upgradeable.balanceOf(deployer.address)).to.equal(amount)

        await mockERC20Upgradeable.burn(amount)
        expect(await mockERC20Upgradeable.balanceOf(deployer.address)).to.equal(0)
    })

    it('Should mint token and transfer them', async function () {
        const amount = 1000
        await mockERC20Upgradeable.mint(deployer.address, amount)
        expect(await mockERC20Upgradeable.balanceOf(deployer.address)).to.equal(amount)

        await mockERC20Upgradeable.transfer(user1.address, amount)
        expect(await mockERC20Upgradeable.balanceOf(user1.address)).to.equal(1000)
    })

    it('Should mint token and transferFrom them', async function () {
        const amount = 1000
        await mockERC20Upgradeable.mint(deployer.address, amount)
        expect(await mockERC20Upgradeable.balanceOf(deployer.address)).to.equal(amount)

        await mockERC20Upgradeable.approve(user1.address, amount)
        await mockERC20Upgradeable.connect(user1).transferFrom(deployer.address, user2.address, amount)
        expect(await mockERC20Upgradeable.balanceOf(user2.address)).to.equal(1000)
    })
})