wechaty-puppet-service/src/deprecated/chunk-pb.spec.ts

Puppet Service for Wechaty

#!/usr/bin/env -S node --no-warnings --loader ts-node/esm

import { test } from 'tstest'

import { PassThrough } from 'stream'

import { FileBox } from 'file-box'

import {
  puppet,
}                                 from 'wechaty-grpc'

import {
  unpackFileBoxFromChunk,
  packFileBoxToChunk,
}                             from './file-box-chunk.js'
import {
  unpackFileBoxChunkFromPb,
  packFileBoxChunkToPb,
}                             from './chunk-pb.js'

test('packFileBoxChunk()', async t => {
  const FILE_BOX_DATA = 'test'
  const FILE_BOX_NAME = 'test.dat'

  const fileBox = FileBox.fromBuffer(
    Buffer.from(FILE_BOX_DATA),
    FILE_BOX_NAME,
  )

  const chunkStream = await packFileBoxToChunk(fileBox)
  const pbStream    = await packFileBoxChunkToPb(puppet.MessageFileStreamResponse)(chunkStream)

  let   name        = ''
  let   buffer      = ''
  pbStream.on('data', (data: puppet.MessageFileStreamResponse) => {
    if (data.hasFileBoxChunk()) {
      const fileBoxChunk = data.getFileBoxChunk()
      if (fileBoxChunk!.hasData()) {
        buffer += fileBoxChunk!.getData()
      } else if (fileBoxChunk!.hasName()) {
        name = fileBoxChunk!.getName()
      }
    }
  })

  await new Promise(resolve => chunkStream.on('end', resolve))
  t.equal(name, FILE_BOX_NAME, 'should get file box name')
  t.equal(buffer, FILE_BOX_DATA, 'should get file box data')

})

test('unpackFileBoxChunkFromPb()', async t => {
  const FILE_BOX_DATA = 'test'
  const FILE_BOX_NAME = 'test.dat'

  const fileBox = FileBox.fromBuffer(
    Buffer.from(FILE_BOX_DATA),
    FILE_BOX_NAME,
  )

  const chunkStream = await packFileBoxToChunk(fileBox)
  const request = new puppet.MessageSendFileStreamRequest()

  const packedStream = new PassThrough({ objectMode: true })

  chunkStream.on('data', (data: puppet.FileBoxChunk) => {
    request.setFileBoxChunk(data)
    packedStream.write(request)
  }).on('end', () => {
    packedStream.end()
  })

  const outputChunkStream = unpackFileBoxChunkFromPb(packedStream)
  const outputFileBox = await unpackFileBoxFromChunk(outputChunkStream)

  t.equal((await outputFileBox.toBuffer()).toString(), FILE_BOX_DATA, 'should get file box data')
})

test('packFileBoxChunk() <-> unpackFileBoxChunkFromPb()', async t => {
  const FILE_BOX_DATA = 'test'
  const FILE_BOX_NAME = 'test.dat'

  const fileBox = FileBox.fromBuffer(
    Buffer.from(FILE_BOX_DATA),
    FILE_BOX_NAME,
  )

  const chunkStream = await packFileBoxToChunk(fileBox)
  const packedStream = packFileBoxChunkToPb(puppet.MessageFileStreamResponse)(chunkStream)

  const unpackedStream = unpackFileBoxChunkFromPb(packedStream)
  const restoredBox = await unpackFileBoxFromChunk(unpackedStream)
  t.equal(fileBox.name, restoredBox.name, 'should be same name')

  const EXPECTED_BASE64 = await fileBox.toBase64()
  const actualBase64 = await restoredBox.toBase64()

  t.equal(EXPECTED_BASE64, actualBase64, 'should be same content')
})

test('packFileBoxChunk(): should not throw if no read on the stream', async t => {

  t.plan(1)
  const stream = await getTestChunkStream({})
  let outStream
  try {
    outStream = packFileBoxChunkToPb(puppet.MessageFileStreamResponse)(stream)
  } catch (e) {
    t.ok((e as Error).message)
    return
  }
  outStream.on('error', _ => { /* Do nothing */ })
  t.pass()
})

test('packFileBoxChunk(): should emit error in the output stream', async t => {
  const EXPECTED_MESSAGE = 'test emit error'

  const stream = await getTestChunkStream({ errorMessage: EXPECTED_MESSAGE })
  const outStream = packFileBoxChunkToPb(puppet.MessageFileStreamResponse)(stream)

  const error = await new Promise<Error>(resolve => outStream.on('error', resolve))
  // await new Promise(resolve => outStream.on('end', resolve))
  t.equal(error.message, EXPECTED_MESSAGE, 'should emit error message')
})

test('unpackFileBoxChunkFromPb(): should not throw if no read on the stream', async t => {

  t.plan(1)
  const stream = await getTestPackedStream({})
  let outStream
  try {
    outStream = unpackFileBoxChunkFromPb(stream)
    t.pass('should no rejection')
  } catch (e) {
    t.fail((e as Error).message)
    return
  }
  outStream.on('error', _ => { /* Do nothing */ })
})

test('unpackFileBoxChunkFromPb(): should emit error in the output stream', async t => {

  const errorMessage = 'test emit error'
  const stream = await getTestPackedStream({ errorMessage })

  const outStream = packFileBoxChunkToPb(puppet.MessageFileStreamResponse)(stream)

  try {
    await new Promise((resolve, reject) => {
      outStream.on('error', reject)
      outStream.on('end', resolve)
    })
    t.fail('should reject the promise')
  } catch (e) {
    t.equal((e as Error).message, errorMessage, 'should get the expected rejection error message')
  }
})

async function getTestChunkStream (options: {
  errorMessage?: string,
}) {
  const { errorMessage } = options

  const FILE_BOX_DATA = 'test'
  const FILE_BOX_NAME = 'test.dat'

  const fileBox = FileBox.fromBuffer(
    Buffer.from(FILE_BOX_DATA),
    FILE_BOX_NAME,
  )

  const chunkStream = await packFileBoxToChunk(fileBox)
  setImmediate(() => {
    chunkStream.emit('error', new Error(errorMessage))
  })

  return chunkStream
}

async function getTestPackedStream (options: {
  errorMessage?: string,
}) {
  const { errorMessage } = options

  const FILE_BOX_DATA = 'test'
  const FILE_BOX_NAME = 'test.dat'

  const fileBox = FileBox.fromBuffer(
    Buffer.from(FILE_BOX_DATA),
    FILE_BOX_NAME,
  )

  const chunkStream = await packFileBoxToChunk(fileBox)
  const packedStream = new PassThrough({ objectMode: true })
  chunkStream.on('data', d => {
    const packedChunk = new puppet.MessageFileStreamResponse()
    packedChunk.setFileBoxChunk(d)
    packedStream.write(packedChunk)
  }).on('error', e => {
    packedStream.emit('error', e)
  })

  setImmediate(() => {
    chunkStream.emit('error', new Error(errorMessage))
  })

  return packedStream
}