mirror of
https://github.com/0x5eal/luau-unzip.git
synced 2025-04-02 22:00:53 +01:00
81 lines
2.8 KiB
Text
81 lines
2.8 KiB
Text
local fs = require("@lune/fs")
|
|
local process = require("@lune/process")
|
|
|
|
local frktest = require("../lune_packages/frktest")
|
|
local check = frktest.assert.check
|
|
|
|
local ZipReader = require("../lib")
|
|
|
|
-- Reuse the same ZIP files from metadata tests
|
|
local ZIPS = fs.readDir("tests/data")
|
|
local FALLIBLES = {
|
|
"invalid_cde_number_of_files_allocation_greater_offset.zip",
|
|
-- FIXME: Incorrectly handled, file tree is empty and walk silently errors
|
|
-- "invalid_cde_number_of_files_allocation_smaller_offset.zip",
|
|
"invalid_offset.zip",
|
|
"invalid_offset2.zip",
|
|
-- FIXME: Does not error when it should
|
|
-- "comment_garbage.zip",
|
|
"chinese.zip",
|
|
"non_utf8.zip", -- FIXME: Lune breaks for non utf8 data in process stdout
|
|
"pandoc_soft_links.zip", -- FIXME: Soft links are not handled correctly
|
|
-- FIXME: Files with a misaligned comments are not correctly located
|
|
-- "misaligned_comment.zip",
|
|
}
|
|
|
|
return function(test: typeof(frktest.test))
|
|
test.suite("ZIP extraction tests", function()
|
|
for _, file in ZIPS do
|
|
if not string.match(file, "%.zip$") then
|
|
continue
|
|
end
|
|
|
|
local checkErr: ((...any) -> any?) -> nil = if table.find(FALLIBLES, file)
|
|
then check.should_error
|
|
else check.should_not_error
|
|
|
|
test.case(`Extracts files correctly - {file}`, function()
|
|
checkErr(function()
|
|
local zipPath = "tests/data/" .. file
|
|
local data = fs.readFile(zipPath)
|
|
local zip = ZipReader.load(buffer.fromstring(data))
|
|
|
|
-- Test both string and buffer extraction
|
|
local stringOptions = { isString = true, decompress = true }
|
|
local bufferOptions = { isString = false, decompress = true }
|
|
|
|
-- Extract and verify each file
|
|
zip:walk(function(entry)
|
|
if entry.isDirectory then
|
|
return
|
|
end
|
|
|
|
-- Extract with unzip for comparison
|
|
local unzipOutput = process.spawn(`unzip`, { "-p", zipPath, entry:getPath() })
|
|
|
|
-- NOTE: We use assert since we don't know whether to expect true or false
|
|
assert(unzipOutput.ok)
|
|
|
|
-- Test string extraction
|
|
local contentString = zip:extract(entry, stringOptions) :: string
|
|
check.equal(#contentString, entry.size)
|
|
check.equal(contentString, unzipOutput.stdout)
|
|
|
|
-- Test buffer extraction
|
|
local contentBuffer = zip:extract(entry, bufferOptions) :: buffer
|
|
check.equal(buffer.len(contentBuffer), entry.size)
|
|
check.equal(buffer.tostring(contentBuffer), unzipOutput.stdout)
|
|
|
|
-- Test directory extraction
|
|
local parentPath = entry:getPath():match("(.+)/[^/]*$") or "/"
|
|
local dirContents = zip:extractDirectory(parentPath, stringOptions)
|
|
check.not_nil(dirContents[entry:getPath()])
|
|
check.equal(dirContents[entry:getPath()], unzipOutput.stdout)
|
|
end)
|
|
|
|
return
|
|
end)
|
|
end)
|
|
end
|
|
end)
|
|
end
|