97 lines
4.1 KiB
TypeScript
97 lines
4.1 KiB
TypeScript
// Copyright 2024 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
|
|
import { assert } from 'chai';
|
|
import { shuffle } from 'lodash';
|
|
|
|
import { IMAGE_JPEG } from '../../../types/MIME';
|
|
import { groupMediaItemsByDate } from '../../../components/conversation/media-gallery/groupMediaItemsByDate';
|
|
import type { MediaItemType } from '../../../types/MediaItem';
|
|
import { fakeAttachment } from '../../../test-both/helpers/fakeAttachment';
|
|
|
|
const testDate = (
|
|
year: number,
|
|
month: number,
|
|
day: number,
|
|
hour: number,
|
|
minute: number,
|
|
second = 0
|
|
): Date => new Date(year, month - 1, day, hour, minute, second, 0);
|
|
|
|
const toMediaItem = (id: string, date: Date): MediaItemType => {
|
|
return {
|
|
objectURL: id,
|
|
index: 0,
|
|
message: {
|
|
conversationId: '1234',
|
|
id: 'id',
|
|
receivedAt: date.getTime(),
|
|
receivedAtMs: date.getTime(),
|
|
attachments: [],
|
|
sentAt: date.getTime(),
|
|
},
|
|
attachment: fakeAttachment({
|
|
fileName: 'fileName',
|
|
contentType: IMAGE_JPEG,
|
|
url: 'url',
|
|
}),
|
|
};
|
|
};
|
|
|
|
describe('groupMediaItemsByDate', () => {
|
|
it('should group mediaItems', () => {
|
|
const referenceTime = testDate(2024, 4, 12, 18, 0, 0).getTime(); // Friday
|
|
const input: Array<MediaItemType> = shuffle([
|
|
toMediaItem('today-1', testDate(2024, 4, 12, 17, 59)), // Friday, one minute ago
|
|
toMediaItem('today-2', testDate(2024, 4, 12, 0, 1)), // Friday early morning
|
|
toMediaItem('yesterday-1', testDate(2024, 4, 11, 18, 0)), // Thursday
|
|
toMediaItem('yesterday-2', testDate(2024, 4, 11, 0, 1)), // Thursday early morning
|
|
toMediaItem('thisWeek-1', testDate(2024, 4, 10, 18, 0)), // Wednesday
|
|
toMediaItem('thisWeek-2', testDate(2024, 4, 8, 18, 0)), // Monday
|
|
toMediaItem('thisWeek-3', testDate(2024, 4, 5, 18, 0)), // Last Friday
|
|
toMediaItem('thisWeek-4', testDate(2024, 4, 5, 0, 1)), // Last Friday early morning
|
|
toMediaItem('thisMonth-1', testDate(2024, 4, 2, 18, 0)), // Second day of moth
|
|
toMediaItem('thisMonth-2', testDate(2024, 4, 1, 18, 0)), // First day of month
|
|
toMediaItem('mar2024-1', testDate(2024, 3, 31, 23, 59)),
|
|
toMediaItem('mar2024-2', testDate(2024, 3, 1, 0, 1)),
|
|
toMediaItem('feb2011-1', testDate(2011, 2, 28, 23, 59)),
|
|
toMediaItem('feb2011-2', testDate(2011, 2, 1, 0, 1)),
|
|
]);
|
|
|
|
const actual = groupMediaItemsByDate(referenceTime, input);
|
|
|
|
assert.strictEqual(actual[0].type, 'today');
|
|
assert.strictEqual(actual[0].mediaItems.length, 2, 'today');
|
|
assert.strictEqual(actual[0].mediaItems[0].objectURL, 'today-1');
|
|
assert.strictEqual(actual[0].mediaItems[1].objectURL, 'today-2');
|
|
|
|
assert.strictEqual(actual[1].type, 'yesterday');
|
|
assert.strictEqual(actual[1].mediaItems.length, 2, 'yesterday');
|
|
assert.strictEqual(actual[1].mediaItems[0].objectURL, 'yesterday-1');
|
|
assert.strictEqual(actual[1].mediaItems[1].objectURL, 'yesterday-2');
|
|
|
|
assert.strictEqual(actual[2].type, 'thisWeek');
|
|
assert.strictEqual(actual[2].mediaItems.length, 4, 'thisWeek');
|
|
assert.strictEqual(actual[2].mediaItems[0].objectURL, 'thisWeek-1');
|
|
assert.strictEqual(actual[2].mediaItems[1].objectURL, 'thisWeek-2');
|
|
assert.strictEqual(actual[2].mediaItems[2].objectURL, 'thisWeek-3');
|
|
assert.strictEqual(actual[2].mediaItems[3].objectURL, 'thisWeek-4');
|
|
|
|
assert.strictEqual(actual[3].type, 'thisMonth');
|
|
assert.strictEqual(actual[3].mediaItems.length, 2, 'thisMonth');
|
|
assert.strictEqual(actual[3].mediaItems[0].objectURL, 'thisMonth-1');
|
|
assert.strictEqual(actual[3].mediaItems[1].objectURL, 'thisMonth-2');
|
|
|
|
assert.strictEqual(actual[4].type, 'yearMonth');
|
|
assert.strictEqual(actual[4].mediaItems.length, 2, 'mar2024');
|
|
assert.strictEqual(actual[4].mediaItems[0].objectURL, 'mar2024-1');
|
|
assert.strictEqual(actual[4].mediaItems[1].objectURL, 'mar2024-2');
|
|
|
|
assert.strictEqual(actual[5].type, 'yearMonth');
|
|
assert.strictEqual(actual[5].mediaItems.length, 2, 'feb2011');
|
|
assert.strictEqual(actual[5].mediaItems[0].objectURL, 'feb2011-1');
|
|
assert.strictEqual(actual[5].mediaItems[1].objectURL, 'feb2011-2');
|
|
|
|
assert.strictEqual(actual.length, 6, 'total sections');
|
|
});
|
|
});
|