{
+ let client: Mocked;
+ let room: Room;
+ let store: MessagePreviewStore;
+
async function addEvent(
store: MessagePreviewStore,
room: Room,
event: MatrixEvent,
fireAction = true,
): Promise {
- room.timeline.push(event);
- mocked(room.findEventById).mockImplementation((eventId) => room.timeline.find((e) => e.getId() === eventId));
+ room.addLiveEvents([event]);
if (fireAction) {
// @ts-ignore private access
await store.onAction({
@@ -42,15 +46,17 @@ describe("MessagePreviewStore", () => {
}
}
- it("should ignore edits for events other than the latest one", async () => {
- const client = stubClient();
- const room = mkStubRoom("!roomId:server", "Room", client);
+ beforeEach(async () => {
+ client = mocked(stubClient());
+ room = new Room("!roomId:server", client, client.getSafeUserId());
mocked(client.getRoom).mockReturnValue(room);
- const store = MessagePreviewStore.testInstance();
+ store = MessagePreviewStore.testInstance();
await store.start();
await setupAsyncStoreWithClient(store, client);
+ });
+ it("should ignore edits for events other than the latest one", async () => {
const firstMessage = mkMessage({
user: "@sender:server",
event: true,
@@ -59,7 +65,7 @@ describe("MessagePreviewStore", () => {
});
await addEvent(store, room, firstMessage, false);
- await expect(store.getPreviewForRoom(room, DefaultTagID.Untagged)).resolves.toMatchInlineSnapshot(
+ expect((await store.getPreviewForRoom(room, DefaultTagID.Untagged))?.text).toMatchInlineSnapshot(
`"@sender:server: First message"`,
);
@@ -71,7 +77,7 @@ describe("MessagePreviewStore", () => {
});
await addEvent(store, room, secondMessage);
- await expect(store.getPreviewForRoom(room, DefaultTagID.Untagged)).resolves.toMatchInlineSnapshot(
+ expect((await store.getPreviewForRoom(room, DefaultTagID.Untagged))?.text).toMatchInlineSnapshot(
`"@sender:server: Second message"`,
);
@@ -93,7 +99,7 @@ describe("MessagePreviewStore", () => {
});
await addEvent(store, room, firstMessageEdit);
- await expect(store.getPreviewForRoom(room, DefaultTagID.Untagged)).resolves.toMatchInlineSnapshot(
+ expect((await store.getPreviewForRoom(room, DefaultTagID.Untagged))?.text).toMatchInlineSnapshot(
`"@sender:server: Second message"`,
);
@@ -115,21 +121,13 @@ describe("MessagePreviewStore", () => {
});
await addEvent(store, room, secondMessageEdit);
- await expect(store.getPreviewForRoom(room, DefaultTagID.Untagged)).resolves.toMatchInlineSnapshot(
+ expect((await store.getPreviewForRoom(room, DefaultTagID.Untagged))?.text).toMatchInlineSnapshot(
`"@sender:server: Second Message Edit"`,
);
});
it("should ignore edits to unknown events", async () => {
- const client = stubClient();
- const room = mkStubRoom("!roomId:server", "Room", client);
- mocked(client.getRoom).mockReturnValue(room);
-
- const store = MessagePreviewStore.testInstance();
- await store.start();
- await setupAsyncStoreWithClient(store, client);
-
- await expect(store.getPreviewForRoom(room, DefaultTagID.DM)).resolves.toMatchInlineSnapshot(`null`);
+ await expect(store.getPreviewForRoom(room, DefaultTagID.DM)).resolves.toBeNull();
const firstMessage = mkMessage({
user: "@sender:server",
@@ -139,7 +137,7 @@ describe("MessagePreviewStore", () => {
});
await addEvent(store, room, firstMessage, true);
- await expect(store.getPreviewForRoom(room, DefaultTagID.DM)).resolves.toMatchInlineSnapshot(
+ expect((await store.getPreviewForRoom(room, DefaultTagID.DM))?.text).toMatchInlineSnapshot(
`"@sender:server: First message"`,
);
@@ -161,22 +159,15 @@ describe("MessagePreviewStore", () => {
});
await addEvent(store, room, randomEdit);
- await expect(store.getPreviewForRoom(room, DefaultTagID.Untagged)).resolves.toMatchInlineSnapshot(
+ expect((await store.getPreviewForRoom(room, DefaultTagID.Untagged))?.text).toMatchInlineSnapshot(
`"@sender:server: First message"`,
);
});
it("should generate correct preview for message events in DMs", async () => {
- const client = stubClient();
- const room = mkStubRoom("!roomId:server", "Room", client);
- mocked(client.getRoom).mockReturnValue(room);
- room.currentState.getJoinedMemberCount = jest.fn().mockReturnValue(2);
+ room.getLiveTimeline().getState(EventTimeline.FORWARDS)!.getJoinedMemberCount = jest.fn().mockReturnValue(2);
- const store = MessagePreviewStore.testInstance();
- await store.start();
- await setupAsyncStoreWithClient(store, client);
-
- await expect(store.getPreviewForRoom(room, DefaultTagID.DM)).resolves.toMatchInlineSnapshot(`null`);
+ await expect(store.getPreviewForRoom(room, DefaultTagID.DM)).resolves.toBeNull();
const firstMessage = mkMessage({
user: "@sender:server",
@@ -186,7 +177,7 @@ describe("MessagePreviewStore", () => {
});
await addEvent(store, room, firstMessage);
- await expect(store.getPreviewForRoom(room, DefaultTagID.DM)).resolves.toMatchInlineSnapshot(
+ expect((await store.getPreviewForRoom(room, DefaultTagID.DM))?.text).toMatchInlineSnapshot(
`"@sender:server: First message"`,
);
@@ -198,8 +189,45 @@ describe("MessagePreviewStore", () => {
});
await addEvent(store, room, secondMessage);
- await expect(store.getPreviewForRoom(room, DefaultTagID.DM)).resolves.toMatchInlineSnapshot(
+ expect((await store.getPreviewForRoom(room, DefaultTagID.DM))?.text).toMatchInlineSnapshot(
`"@sender:server: Second message"`,
);
});
+
+ it("should generate the correct preview for a reaction", async () => {
+ const firstMessage = mkMessage({
+ user: "@sender:server",
+ event: true,
+ room: room.roomId,
+ msg: "First message",
+ });
+ await addEvent(store, room, firstMessage);
+
+ const reaction = mkReaction(firstMessage);
+ await addEvent(store, room, reaction);
+
+ const preview = await store.getPreviewForRoom(room, DefaultTagID.Untagged);
+ expect(preview).toBeDefined();
+ expect(preview.isThreadReply).toBe(false);
+ expect(preview.text).toMatchInlineSnapshot(`"@sender:server reacted 🙃 to First message"`);
+ });
+
+ it("should generate the correct preview for a reaction on a thread root", async () => {
+ const { rootEvent, thread } = mkThread({
+ room,
+ client,
+ authorId: client.getSafeUserId(),
+ participantUserIds: [client.getSafeUserId()],
+ });
+ await addEvent(store, room, rootEvent);
+
+ const reaction = mkReaction(rootEvent, { ts: 42 });
+ reaction.setThread(thread);
+ await addEvent(store, room, reaction);
+
+ const preview = await store.getPreviewForRoom(room, DefaultTagID.Untagged);
+ expect(preview).toBeDefined();
+ expect(preview.isThreadReply).toBe(false);
+ expect(preview.text).toContain("You reacted 🙃 to root event message");
+ });
});
diff --git a/test/test-utils/test-utils.ts b/test/test-utils/test-utils.ts
index 3d46c0ba5f..c9cf352c48 100644
--- a/test/test-utils/test-utils.ts
+++ b/test/test-utils/test-utils.ts
@@ -36,6 +36,7 @@ import {
ConditionKind,
PushRuleActionName,
IPushRules,
+ RelationType,
} from "matrix-js-sdk/src/matrix";
import { normalize } from "matrix-js-sdk/src/utils";
import { ReEmitter } from "matrix-js-sdk/src/ReEmitter";
@@ -471,6 +472,29 @@ export type MessageEventProps = MakeEventPassThruProps & {
msg?: string;
};
+/**
+ * Creates a "🙃" reaction for the given event.
+ * Uses the same room and user as for the event.
+ *
+ * @returns The reaction event
+ */
+export const mkReaction = (event: MatrixEvent, opts: Partial = {}): MatrixEvent => {
+ return mkEvent({
+ event: true,
+ room: event.getRoomId(),
+ type: EventType.Reaction,
+ user: event.getSender()!,
+ content: {
+ "m.relates_to": {
+ rel_type: RelationType.Annotation,
+ event_id: event.getId(),
+ key: "🙃",
+ },
+ },
+ ...opts,
+ });
+};
+
/**
* Create an m.room.message event.
* @param {Object} opts Values for the message
diff --git a/test/test-utils/threads.ts b/test/test-utils/threads.ts
index 9ce222cddb..8c7c658cfc 100644
--- a/test/test-utils/threads.ts
+++ b/test/test-utils/threads.ts
@@ -135,6 +135,11 @@ export const mkThread = ({
}
const thread = room.createThread(rootEvent.getId()!, rootEvent, events, true);
+
+ events.forEach((event) => {
+ thread.timeline.push(event);
+ });
+
// So that we do not have to mock the thread loading
thread.initialEventsFetched = true;