From 6ebd21270ec3b92ac021f6d09a4e45f3310a3bbb Mon Sep 17 00:00:00 2001
From: gonzaloriestra <14979109+gonzaloriestra@users.noreply.github.com>
Date: Mon, 31 Aug 2026 00:30:50 +0000
Subject: [PATCH] [Tests] Add unit tests for archiver.ts

---
 .../cli-kit/src/public/node/archiver.test.ts  | 95 +++++++++++++++++++
 1 file changed, 95 insertions(+)
 create mode 100644 packages/cli-kit/src/public/node/archiver.test.ts

diff --git a/packages/cli-kit/src/public/node/archiver.test.ts b/packages/cli-kit/src/public/node/archiver.test.ts
new file mode 100644
index 00000000000..6940beda2f8
--- /dev/null
+++ b/packages/cli-kit/src/public/node/archiver.test.ts
@@ -0,0 +1,95 @@
+import {zip, brotliCompress} from './archiver.js'
+import {inTemporaryDirectory, writeFileSync, fileExists, mkdir, readFile} from './fs.js'
+import {joinPath} from './path.js'
+import {exec} from './system.js'
+import {describe, expect, test} from 'vitest'
+import StreamZip from 'node-stream-zip'
+import {brotliDecompressSync} from 'zlib'
+import {writeFileSync as nodeWriteFileSync} from 'node:fs'
+import {readFile as fsPromisesReadFile} from 'node:fs/promises'
+
+describe('zip', () => {
+  test('zips directory files including subdirectories and respects matchFilePattern', async () => {
+    await inTemporaryDirectory(async (tmpDir) => {
+      const inputDir = joinPath(tmpDir, 'input')
+      const subDir = joinPath(inputDir, 'nested')
+      await mkdir(subDir)
+
+      const file1Path = joinPath(inputDir, 'file1.txt')
+      const file2Path = joinPath(subDir, 'file2.js')
+      const file3Path = joinPath(inputDir, 'file3.log')
+
+      writeFileSync(file1Path, 'hello world')
+      writeFileSync(file2Path, 'console.log("test")')
+      writeFileSync(file3Path, 'log entry')
+
+      const zipPath = joinPath(tmpDir, 'output.zip')
+
+      await zip({
+        inputDirectory: inputDir,
+        outputZipPath: zipPath,
+        matchFilePattern: ['**/*.txt', '**/*.js'],
+      })
+
+      await expect(fileExists(zipPath)).resolves.toBe(true)
+
+      // eslint-disable-next-line new-cap
+      const zipFile = new StreamZip.async({file: zipPath})
+      const entries = Object.keys(await zipFile.entries())
+
+      expect(entries).toContain('file1.txt')
+      expect(entries).toContain('nested/')
+      expect(entries).toContain('nested/file2.js')
+      expect(entries).not.toContain('file3.log')
+
+      const file1Buffer = await zipFile.entryData('file1.txt')
+      expect(file1Buffer.toString('utf-8')).toBe('hello world')
+
+      await zipFile.close()
+    })
+  })
+})
+
+describe('brotliCompress', () => {
+  test('compresses directory contents into a brotli tarball', async () => {
+    await inTemporaryDirectory(async (tmpDir) => {
+      const inputDir = joinPath(tmpDir, 'input')
+      const subDir = joinPath(inputDir, 'sub')
+      await mkdir(subDir)
+
+      const file1Path = joinPath(inputDir, 'a.txt')
+      const file2Path = joinPath(subDir, 'b.txt')
+
+      writeFileSync(file1Path, 'alpha')
+      writeFileSync(file2Path, 'beta')
+
+      const outputPath = joinPath(tmpDir, 'archive.tar.br')
+
+      await brotliCompress({
+        inputDirectory: inputDir,
+        outputPath,
+      })
+
+      await expect(fileExists(outputPath)).resolves.toBe(true)
+
+      const compressedData = await fsPromisesReadFile(outputPath)
+      const decompressedTar = brotliDecompressSync(compressedData)
+
+      const extractedDir = joinPath(tmpDir, 'extracted')
+      await mkdir(extractedDir)
+      const tarPath = joinPath(tmpDir, 'extracted.tar')
+      nodeWriteFileSync(tarPath, decompressedTar)
+
+      await exec('tar', ['-xf', tarPath, '-C', extractedDir])
+
+      await expect(fileExists(joinPath(extractedDir, 'a.txt'))).resolves.toBe(true)
+      await expect(fileExists(joinPath(extractedDir, 'sub', 'b.txt'))).resolves.toBe(true)
+
+      const fileAContent = await readFile(joinPath(extractedDir, 'a.txt'))
+      const fileBContent = await readFile(joinPath(extractedDir, 'sub', 'b.txt'))
+
+      expect(fileAContent).toBe('alpha')
+      expect(fileBContent).toBe('beta')
+    })
+  })
+})