@sanity/migrate/src/tar-webstream/__test__/small.test.ts

Tooling for running data migrations on Sanity.io projects

import path from 'node:path'
import {fileURLToPath} from 'node:url'

import {expect, test} from 'vitest'

import {readFileAsWebStream} from '../../fs-webstream/readFileAsWebStream.js'
import {decodeText} from '../../it-utils/decodeText.js'
import {toArray} from '../../it-utils/toArray.js'
import {streamToAsyncIterator} from '../../utils/streamToAsyncIterator.js'
import {untar} from '../untar.js'

const __dirname = path.dirname(fileURLToPath(import.meta.url))

async function* extract(file: string) {
  const fileStream = readFileAsWebStream(file)
  for await (const [header, body] of streamToAsyncIterator(untar(fileStream))) {
    const content = await toArray(decodeText(streamToAsyncIterator(body)))
    yield [header.name, {content, type: header.type}]
  }
}

test('untar a small file', async () => {
  const values = Object.fromEntries(await toArray(extract(`${__dirname}/fixtures/small.tar`)))

  expect(values).toEqual({
    'a.txt': {
      content: ['a'],
      type: 'file',
    },
    'b.txt': {
      content: ['b'],
      type: 'file',
    },
    'c/': {
      content: [],
      type: 'directory',
    },
    'c/d/': {
      content: [],
      type: 'directory',
    },
    'c/d/e/': {
      content: [],
      type: 'directory',
    },
    'c/d/e/f.txt': {
      content: ['f'],
      type: 'file',
    },
    'c/g.txt': {
      content: ['g'],
      type: 'file',
    },
  })
})