Add `Message.initializeAttachmentMetadata`
This commit is contained in:
parent
c5352cf26c
commit
867bece952
|
@ -0,0 +1,50 @@
|
|||
import 'mocha';
|
||||
import { assert } from 'chai';
|
||||
|
||||
import * as Message from '../../../../ts/types/message/initializeAttachmentMetadata';
|
||||
import { IncomingMessage } from '../../../../ts/types/Message';
|
||||
import { MIMEType } from '../../../../ts/types/MIME';
|
||||
// @ts-ignore
|
||||
import { stringToArrayBuffer } from '../../../../js/modules/string_to_array_buffer';
|
||||
|
||||
|
||||
describe('Message', () => {
|
||||
describe('initializeAttachmentMetadata', () => {
|
||||
it('should handle visual media attachments', async () => {
|
||||
const input: IncomingMessage = {
|
||||
type: 'incoming',
|
||||
conversationId: 'foo',
|
||||
id: '11111111-1111-1111-1111-111111111111',
|
||||
timestamp: 1523317140899,
|
||||
received_at: 1523317140899,
|
||||
sent_at: 1523317140800,
|
||||
attachments: [{
|
||||
contentType: 'image/jpeg' as MIMEType,
|
||||
data: stringToArrayBuffer('foo'),
|
||||
fileName: 'foo.jpg',
|
||||
size: 1111,
|
||||
}],
|
||||
};
|
||||
const expected: IncomingMessage = {
|
||||
type: 'incoming',
|
||||
conversationId: 'foo',
|
||||
id: '11111111-1111-1111-1111-111111111111',
|
||||
timestamp: 1523317140899,
|
||||
received_at: 1523317140899,
|
||||
sent_at: 1523317140800,
|
||||
attachments: [{
|
||||
contentType: 'image/jpeg' as MIMEType,
|
||||
data: stringToArrayBuffer('foo'),
|
||||
fileName: 'foo.jpg',
|
||||
size: 1111,
|
||||
}],
|
||||
numAttachments: 1,
|
||||
numVisualMediaAttachments: 1,
|
||||
numFileAttachments: 0,
|
||||
};
|
||||
|
||||
const actual = await Message.initializeAttachmentMetadata(input);
|
||||
assert.deepEqual(actual, expected);
|
||||
});
|
||||
});
|
||||
});
|
|
@ -181,6 +181,9 @@ describe('Message', () => {
|
|||
fileName: 'test\uFFFDfig.exe',
|
||||
size: 1111,
|
||||
}],
|
||||
numAttachments: 1,
|
||||
numVisualMediaAttachments: 0,
|
||||
numFileAttachments: 1,
|
||||
schemaVersion: Message.CURRENT_SCHEMA_VERSION,
|
||||
};
|
||||
|
||||
|
|
|
@ -0,0 +1,20 @@
|
|||
import { partition } from 'lodash';
|
||||
|
||||
import * as Attachment from '../Attachment';
|
||||
import { Message } from '../message';
|
||||
|
||||
|
||||
export const initializeAttachmentMetadata =
|
||||
async (message: Message): Promise<Message> => {
|
||||
const numAttachments = message.attachments.length;
|
||||
const [numVisualMediaAttachments, numFileAttachments] =
|
||||
partition(message.attachments, Attachment.isVisualMedia)
|
||||
.map((attachments) => attachments.length);
|
||||
|
||||
return {
|
||||
...message,
|
||||
numAttachments,
|
||||
numVisualMediaAttachments,
|
||||
numFileAttachments,
|
||||
};
|
||||
};
|
Loading…
Reference in New Issue