@nearform/doctor/test/format-process-stat.test.js

'use strict'

const test = require('tap').test
const endpoint = require('endpoint')
const ProcessStat = require('../collect/process-stat.js')
const ProcessStatDecoder = require('../format/process-stat-decoder.js')
const ProcessStatEncoder = require('../format/process-stat-encoder.js')

const normalizeSample = Object.prototype.hasOwnProperty.call(process.memoryUsage(), 'arrayBuffers')
  ? function (sample) { return sample }
  : function (sample) {
    return {
      ...sample,
      memory: {
        arrayBuffers: 0,
        ...sample.memory
      }
    }
  }

test('Format - process stat - encoder-decoder', function (t) {
  const stat = new ProcessStat(1)

  const encoder = new ProcessStatEncoder()
  const decoder = new ProcessStatDecoder()
  encoder.pipe(decoder)

  const inputSamples = []
  for (let i = 0; i < 1; i++) {
    const sample = stat.sample()
    encoder.write(sample)
    inputSamples.push(normalizeSample(sample))
  }
  encoder.end()

  decoder.pipe(endpoint({ objectMode: true }, function (err, outputSamples) {
    if (err) t.ifError(err)
    t.strictDeepEqual(inputSamples, outputSamples)
    t.end()
  }))
})

test('Format - process stat - partial decoding', function (t) {
  const stat = new ProcessStat(1)

  const encoder = new ProcessStatEncoder()
  const decoder = new ProcessStatDecoder()

  // encode a sample
  const sample = stat.sample()
  encoder.write(sample)
  const sampleEncoded = encoder.read()

  // No data, chunk is too small
  decoder.write(sampleEncoded.slice(0, 20))
  t.strictEqual(decoder.read(), null)

  // Ended previuse sample, but a partial remains
  decoder.write(Buffer.concat([
    sampleEncoded.slice(20),
    sampleEncoded.slice(0, 30)
  ]))
  t.strictDeepEqual(decoder.read(), normalizeSample(sample))
  t.strictEqual(decoder.read(), null)

  // Ended previuse, no partial remains
  decoder.write(Buffer.concat([
    sampleEncoded.slice(30),
    sampleEncoded
  ]))
  t.strictDeepEqual(decoder.read(), normalizeSample(sample))
  t.strictDeepEqual(decoder.read(), normalizeSample(sample))
  t.strictEqual(decoder.read(), null)

  // No previuse ended
  decoder.write(sampleEncoded)
  t.strictDeepEqual(decoder.read(), normalizeSample(sample))

  // No more data
  t.strictEqual(decoder.read(), null)
  t.end()
})