import { beforeAll, describe, expect, it } from 'vitest'
import { getPlatform } from '@mtcute/core/platform.js'

import { __getWasm, ige256Decrypt, ige256Encrypt } from '../src/index.js'

import { initWasm } from './init.js'

const p = getPlatform()

beforeAll(async () => {
    await initWasm()
})

describe('aes-ige', () => {
    const key = p.hexDecode('5468697320697320616E20696D706C655468697320697320616E20696D706C65')
    const iv = p.hexDecode('6D656E746174696F6E206F6620494745206D6F646520666F72204F70656E5353')

    const data = p.hexDecode('99706487a1cde613bc6de0b6f24b1c7aa448c8b9c3403e3467a8cad89340f53b')
    const dataEnc = p.hexDecode('792ea8ae577b1a66cb3bd92679b8030ca54ee631976bd3a04547fdcb4639fa69')

    it('should correctly encrypt', () => {
        const aes = ige256Encrypt(data, key, iv)

        expect(p.hexEncode(aes)).toEqual(p.hexEncode(dataEnc))
    })

    it('should correctly decrypt', () => {
        const aes = ige256Decrypt(dataEnc, key, iv)

        expect(p.hexEncode(aes)).toEqual(p.hexEncode(data))
    })

    it('should not leak memory', () => {
        const mem = __getWasm().memory.buffer
        const memSize = mem.byteLength

        for (let i = 0; i < 10000; i++) {
            ige256Decrypt(ige256Encrypt(data, key, iv), key, iv)
        }

        expect(mem.byteLength).toEqual(memSize)
    })
})