node/test/parallel/test-zlib-zstd-pledged-src-size.js

'use strict';
const common = require('../common');
const assert = require('assert');
const zlib = require('zlib');

function compressWithPledgedSrcSize({ pledgedSrcSize, actualSrcSize }) {
  return new Promise((resolve, reject) => {
    const compressor = zlib.createZstdCompress({ pledgedSrcSize });
    compressor.on('error', (e) => {
      reject(e);
    });
    compressor.on('end', resolve);
    compressor.write('x'.repeat(actualSrcSize), () => {
      compressor.end();
      compressor.resume();
    });
  }).then(() => {
    // Compression should only succeed if sizes match
    assert.strictEqual(pledgedSrcSize, actualSrcSize);
  }, (error) => {
    assert.strictEqual(error.code, 'ZSTD_error_srcSize_wrong');
    // Size error should only happen when sizes do not match
    assert.notStrictEqual(pledgedSrcSize, actualSrcSize);
  }).then(common.mustCall());
}

compressWithPledgedSrcSize({ pledgedSrcSize: 0, actualSrcSize: 0 });

compressWithPledgedSrcSize({ pledgedSrcSize: 0, actualSrcSize: 42 });

compressWithPledgedSrcSize({ pledgedSrcSize: 13, actualSrcSize: 42 });

compressWithPledgedSrcSize({ pledgedSrcSize: 42, actualSrcSize: 0 });

compressWithPledgedSrcSize({ pledgedSrcSize: 42, actualSrcSize: 13 });

compressWithPledgedSrcSize({ pledgedSrcSize: 42, actualSrcSize: 42 });