wechaty-puppet-service
Version:
Puppet Service for Wechaty
92 lines • 4.54 kB
JavaScript
/**
* 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) => {
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');
});
//# sourceMappingURL=conversation-id-file-box.spec.js.map