Julien Fontanet
3a74c71f1a
BREAKING: - removes `dist/` in the path of sub-modules - requires Node >=12
95 lines
3.5 KiB
JavaScript
95 lines
3.5 KiB
JavaScript
/* eslint-env jest */
|
|
|
|
const execa = require('execa')
|
|
const fs = require('fs-extra')
|
|
const rimraf = require('rimraf')
|
|
const getStream = require('get-stream')
|
|
const tmp = require('tmp')
|
|
const { createReadStream, createWriteStream } = require('fs')
|
|
const { pFromCallback } = require('promise-toolbox')
|
|
const { pipeline } = require('readable-stream')
|
|
|
|
const { createVhdStreamWithLength } = require('./createVhdStreamWithLength.js')
|
|
const { FOOTER_SIZE } = require('./_constants')
|
|
const { createRandomFile, convertFromRawToVhd, convertFromVhdToRaw } = require('./tests/utils')
|
|
|
|
let tempDir = null
|
|
|
|
beforeEach(async () => {
|
|
tempDir = await pFromCallback(cb => tmp.dir(cb))
|
|
})
|
|
|
|
afterEach(async () => {
|
|
await pFromCallback(cb => rimraf(tempDir, cb))
|
|
})
|
|
|
|
const forOwn = (object, cb) => Object.keys(object).forEach(key => cb(object[key], key, object))
|
|
|
|
describe('createVhdStreamWithLength', () => {
|
|
forOwn(
|
|
{
|
|
// qemu-img requires this length or it fill with null bytes which breaks
|
|
// the test
|
|
'can extract length': 34816,
|
|
|
|
'can handle empty file': 0,
|
|
},
|
|
(size, title) =>
|
|
it(title, async () => {
|
|
const inputRaw = `${tempDir}/input.raw`
|
|
await createRandomFile(inputRaw, size / 1024 / 1024)
|
|
|
|
const inputVhd = `${tempDir}/input.vhd`
|
|
await convertFromRawToVhd(inputRaw, inputVhd)
|
|
|
|
const result = await createVhdStreamWithLength(await createReadStream(inputVhd))
|
|
const { length } = result
|
|
|
|
const outputVhd = `${tempDir}/output.vhd`
|
|
await pFromCallback(pipeline.bind(undefined, result, await createWriteStream(outputVhd)))
|
|
|
|
// ensure the guessed length correspond to the stream length
|
|
const { size: outputSize } = await fs.stat(outputVhd)
|
|
expect(length).toEqual(outputSize)
|
|
|
|
// ensure the generated VHD is correct and contains the same data
|
|
const outputRaw = `${tempDir}/output.raw`
|
|
await convertFromVhdToRaw(outputVhd, outputRaw)
|
|
await execa('cmp', [inputRaw, outputRaw])
|
|
})
|
|
)
|
|
|
|
it('can skip blank after the last block and before the footer', async () => {
|
|
const initialSize = 4
|
|
const rawFileName = `${tempDir}/randomfile`
|
|
const vhdName = `${tempDir}/randomfile.vhd`
|
|
const outputVhdName = `${tempDir}/output.vhd`
|
|
await createRandomFile(rawFileName, initialSize)
|
|
await convertFromRawToVhd(rawFileName, vhdName)
|
|
const { size: vhdSize } = await fs.stat(vhdName)
|
|
// read file footer
|
|
const footer = await getStream.buffer(createReadStream(vhdName, { start: vhdSize - FOOTER_SIZE }))
|
|
|
|
// we'll override the footer
|
|
const endOfFile = await createWriteStream(vhdName, {
|
|
flags: 'r+',
|
|
start: vhdSize - FOOTER_SIZE,
|
|
})
|
|
// write a blank over the previous footer
|
|
await pFromCallback(cb => endOfFile.write(Buffer.alloc(FOOTER_SIZE), cb))
|
|
// write the footer after the new blank
|
|
await pFromCallback(cb => endOfFile.end(footer, cb))
|
|
const { size: longerSize } = await fs.stat(vhdName)
|
|
// check input file has been lengthened
|
|
expect(longerSize).toEqual(vhdSize + FOOTER_SIZE)
|
|
const result = await createVhdStreamWithLength(await createReadStream(vhdName))
|
|
expect(result.length).toEqual(vhdSize)
|
|
const outputFileStream = await createWriteStream(outputVhdName)
|
|
await pFromCallback(cb => pipeline(result, outputFileStream, cb))
|
|
const { size: outputSize } = await fs.stat(outputVhdName)
|
|
// check out file has been shortened again
|
|
expect(outputSize).toEqual(vhdSize)
|
|
await execa('qemu-img', ['compare', outputVhdName, vhdName])
|
|
})
|
|
})
|