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

Tooling for running data migrations on Sanity.io projects

import {expect, test} from 'vitest'

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

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, {type: header.type, content}]
  }
}

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',
    },
  })
})