mtcute/packages/wasm/tests/gunzip.test.ts

59 lines
1.8 KiB
TypeScript

import { gzipSync } from 'node:zlib'
import { beforeAll, describe, expect, it } from 'vitest'
import { utf8 } from '@fuman/utils'
import { __getWasm, gunzip } from '../src/index.js'
import { initWasm } from './init.js'
beforeAll(async () => {
await initWasm()
})
function gzipSyncWrap(data: Uint8Array) {
if (import.meta.env.TEST_ENV === 'browser' || import.meta.env.TEST_ENV === 'deno') {
// @ts-expect-error fucking crutch because @jspm/core uses Buffer.isBuffer for some reason
data._isBuffer = true
return new Uint8Array(gzipSync(data))
}
return gzipSync(data)
}
describe('gunzip', () => {
it('should correctly read zlib headers', () => {
const wasm = __getWasm()
const data = gzipSyncWrap(utf8.encoder.encode('hello world'))
const inputPtr = wasm.__malloc(data.length)
new Uint8Array(wasm.memory.buffer).set(data, inputPtr)
expect(wasm.libdeflate_gzip_get_output_size(inputPtr, data.length)).toEqual(11)
})
it('should correctly inflate', () => {
const data = Array.from({ length: 1000 }, () => 'a').join('')
const res = gzipSyncWrap(utf8.encoder.encode(data))
expect(res).not.toBeNull()
expect(res.length).toBeLessThan(100)
expect(gunzip(res)).toEqual(new Uint8Array(utf8.encoder.encode(data)))
})
it('should not leak memory', () => {
const memSize = __getWasm().memory.buffer.byteLength
for (let i = 0; i < 100; i++) {
const data = Array.from({ length: 1000 }, () => 'a').join('')
const deflated = gzipSyncWrap(utf8.encoder.encode(data))
const res = gunzip(deflated)
expect(utf8.decoder.decode(res)).toEqual(data)
}
expect(__getWasm().memory.buffer.byteLength).toEqual(memSize)
})
})