wechaty-puppet-service/src/deprecated/conversation-id-file-box.spec.ts

Puppet Service for Wechaty

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

/**
 *   Wechaty Chatbot SDK - https://github.com/wechaty/wechaty
 *
 *   @copyright 2016 Huan LI (李卓桓) <https://github.com/huan>, and
 *                   Wechaty Contributors <https://github.com/wechaty>.
 *
 *   Licensed under the Apache License, Version 2.0 (the "License");
 *   you may not use this file except in compliance with the License.
 *   You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 *   Unless required by applicable law or agreed to in writing, software
 *   distributed under the License is distributed on an "AS IS" BASIS,
 *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *   See the License for the specific language governing permissions and
 *   limitations under the License.
 *
 */
import {
  puppet,
}                                       from 'wechaty-grpc'

import { test }         from 'tstest'
import { PassThrough }  from 'stream'
import { FileBox }      from 'file-box'

import { nextData }     from './next-data.js'
import {
  packConversationIdFileBoxToPb,
  unpackConversationIdFileBoxArgsFromPb,
}                                         from './conversation-id-file-box.js'

test('unpackConversationIdFileBoxArgsFromPb()', async t => {
  const FILE_BOX_DATA = 'test'
  const FILE_BOX_NAME = 'test.dat'
  const CONVERSATION_ID = 'conversation_id'

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

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

  const req1 = new puppet.MessageSendFileStreamRequest()
  req1.setConversationId(CONVERSATION_ID)
  stream.write(req1)

  const req2 = new puppet.MessageSendFileStreamRequest()
  const chunk1 = new puppet.FileBoxChunk()
  chunk1.setName(fileBox.name)
  req2.setFileBoxChunk(chunk1)
  stream.write(req2)

  const fileBoxStream = await fileBox.toStream()
  fileBoxStream.on('data', chunk => {
    const fileBoxChunk = new puppet.FileBoxChunk()
    fileBoxChunk.setData(chunk)
    const req3 = new puppet.MessageSendFileStreamRequest()
    req3.setFileBoxChunk(fileBoxChunk)
    stream.write(req3)
  })
  fileBoxStream.on('end', () => stream.end())

  const args = await unpackConversationIdFileBoxArgsFromPb(stream)
  const data = (await args.fileBox.toBuffer()).toString()

  t.equal(args.conversationId, CONVERSATION_ID, 'should get conversation id')
  t.equal(args.fileBox.name, FILE_BOX_NAME, 'should get file box name')
  t.equal(data, FILE_BOX_DATA, 'should get file box data')
})

test('packConversationIdFileBoxToPb()', async t => {
  const FILE_BOX_DATA = 'test'
  const FILE_BOX_NAME = 'test.dat'
  const CONVERSATION_ID = 'conv_id'

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

  const stream = await packConversationIdFileBoxToPb(puppet.MessageSendFileStreamRequest)(
    CONVERSATION_ID,
    fileBox,
  )

  const data1 = await nextData(stream)
  t.equal(data1.getConversationId(), CONVERSATION_ID, 'match conversation id')

  const data2 = await nextData(stream)
  t.ok(data2.hasFileBoxChunk(), 'has file box chunk')
  t.ok(data2.getFileBoxChunk()!.hasName(), 'has file box name')
  t.equal(data2.getFileBoxChunk()!.getName(), FILE_BOX_NAME, 'match file box name')

  let data = ''
  stream.on('data', (chunk: puppet.MessageSendFileStreamRequest) => {
    if (!chunk.hasFileBoxChunk()) {
      throw new Error('no file box chunk')
    }
    if (!chunk.getFileBoxChunk()!.hasData()) {
      throw new Error('no file box chunk data')
    }
    data += chunk.getFileBoxChunk()!.getData()
  })

  await new Promise(resolve => stream.on('end', resolve))

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

test('unpackConversationIdFileBoxArgsFromPb() <-> packConversationIdFileBoxToPb()', async t => {
  const FILE_BOX_DATA = 'test'
  const FILE_BOX_NAME = 'test.dat'
  const CONVERSATION_ID = 'conv_id'

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

  const stream = await packConversationIdFileBoxToPb(puppet.MessageSendFileStreamRequest)(CONVERSATION_ID, fileBox)
  const args = await unpackConversationIdFileBoxArgsFromPb(stream)

  t.equal(args.conversationId, CONVERSATION_ID, 'should match conversation id')
  t.equal(args.fileBox.name, FILE_BOX_NAME, 'should be same name')
  t.equal((await args.fileBox.toBuffer()).toString(), FILE_BOX_DATA, 'should be same content')
})