import { expect } from '../../../setup'

/* External Imports */
import { ethers } from 'hardhat'
import { Contract } from 'ethers'

/* Internal Imports */
import { Lib_RLPWriter_TEST_JSON } from '../../../data'

const encode = async (Lib_RLPWriter: Contract, input: any): Promise<void> => {
  if (Array.isArray(input)) {
    const elements = await Promise.all(
      input.map(async (el) => {
        return encode(Lib_RLPWriter, el)
      })
    )

    return Lib_RLPWriter.writeList(elements)
  } else if (Number.isInteger(input)) {
    return Lib_RLPWriter.writeUint(input)
  } else {
    return Lib_RLPWriter.writeString(input)
  }
}

describe('Lib_RLPWriter', () => {
  let Lib_RLPWriter: Contract
  before(async () => {
    Lib_RLPWriter = await (
      await ethers.getContractFactory('TestLib_RLPWriter')
    ).deploy()
  })

  describe('Official Ethereum RLP Tests', () => {
    for (const [key, test] of Object.entries(Lib_RLPWriter_TEST_JSON)) {
      it(`should properly encode: ${key}`, async () => {
        expect(await encode(Lib_RLPWriter, test.in)).to.equal(test.out)
      })
    }
  })

  describe('Use of library with other memory-modifying operations', () => {
    it('should allow creation of a contract beforehand and still work', async () => {
      const randomAddress = '0x1234123412341234123412341234123412341234'
      const rlpEncodedRandomAddress =
        '0x941234123412341234123412341234123412341234'
      const encoded = await Lib_RLPWriter.callStatic.writeAddressWithTaintedMemory(
        randomAddress
      )
      expect(encoded).to.eq(rlpEncodedRandomAddress)
    })
  })
})
