Write unit test

Author: justjanne

spec/unit/models/thread.spec.ts

@@ -14,17 +14,17 @@ See the License for the specific language governing permissions and
 limitations under the License.
 */
 
-import { MatrixClient, PendingEventOrdering } from "../../../src/client";
-import { Room } from "../../../src/models/room";
-import { Thread, THREAD_RELATION_TYPE, ThreadEvent } from "../../../src/models/thread";
-import { mkThread } from "../../test-utils/thread";
-import { TestClient } from "../../TestClient";
-import { emitPromise, mkMessage, mock } from "../../test-utils/test-utils";
-import { EventStatus, MatrixEvent } from "../../../src";
-import { ReceiptType } from "../../../src/@types/read_receipts";
-import { getMockClientWithEventEmitter, mockClientMethodsUser } from "../../test-utils/client";
-import { ReEmitter } from "../../../src/ReEmitter";
-import { Feature, ServerSupport } from "../../../src/feature";
+import {MatrixClient, PendingEventOrdering} from "../../../src/client";
+import {Room} from "../../../src/models/room";
+import {Thread, THREAD_RELATION_TYPE, ThreadEvent} from "../../../src/models/thread";
+import {mkThread} from "../../test-utils/thread";
+import {TestClient} from "../../TestClient";
+import {emitPromise, mkMessage, mock} from "../../test-utils/test-utils";
+import {Direction, EventStatus, MatrixEvent} from "../../../src";
+import {ReceiptType} from "../../../src/@types/read_receipts";
+import {getMockClientWithEventEmitter, mockClientMethodsUser} from "../../test-utils/client";
+import {ReEmitter} from "../../../src/ReEmitter";
+import {Feature, ServerSupport} from "../../../src/feature";
 
 describe("Thread", () => {
     describe("constructor", () => {
@@ -283,4 +283,84 @@ describe("Thread", () => {
             expect(thread2.getEventReadUpTo(myUserId)).toBe(null);
         });
     });
+
+    describe("resetLiveTimeline", () => {
+        it("correctly resets the live timeline", async () => {
+            const myUserId = "@bob:example.org";
+            const testClient = new TestClient(myUserId, "DEVICE", "ACCESS_TOKEN", undefined, { timelineSupport: false });
+            const client = testClient.client;
+            const room = new Room("123", client, myUserId, {
+                pendingEventOrdering: PendingEventOrdering.Detached,
+            });
+
+            jest.spyOn(client, "getRoom").mockReturnValue(room);
+
+            const { thread } = mkThread({
+                room,
+                client,
+                authorId: myUserId,
+                participantUserIds: ["@alice:example.org"],
+                length: 3,
+            });
+            await emitPromise(thread, ThreadEvent.Update);
+            expect(thread.length).toBe(2);
+
+            jest.spyOn(client, "createMessagesRequest").mockImplementation((_, token) => Promise.resolve({
+                chunk: [],
+                start: `${token}-new`,
+                end: `${token}-new`,
+            }));
+
+            function timelines(): [string | null, string | null][] {
+                return thread.timelineSet.getTimelines()
+                    .map(it => [it.getPaginationToken(Direction.Backward), it.getPaginationToken(Direction.Forward)]);
+            }
+
+            expect(timelines()).toEqual([[null, null]]);
+            const promise = thread.resetLiveTimeline("b1", "f1");
+            expect(timelines()).toEqual([[null, "f1"], ["b1", null]]);
+            await promise;
+            expect(timelines()).toEqual([[null, "f1-new"], ["b1-new", null]]);
+        });
+
+        it("does not modify changed tokens", async () => {
+            const myUserId = "@bob:example.org";
+            const testClient = new TestClient(myUserId, "DEVICE", "ACCESS_TOKEN", undefined, { timelineSupport: false });
+            const client = testClient.client;
+            const room = new Room("123", client, myUserId, {
+                pendingEventOrdering: PendingEventOrdering.Detached,
+            });
+
+            jest.spyOn(client, "getRoom").mockReturnValue(room);
+
+            const { thread } = mkThread({
+                room,
+                client,
+                authorId: myUserId,
+                participantUserIds: ["@alice:example.org"],
+                length: 3,
+            });
+            await emitPromise(thread, ThreadEvent.Update);
+            expect(thread.length).toBe(2);
+
+            jest.spyOn(client, "createMessagesRequest").mockImplementation((_, token) => Promise.resolve({
+                chunk: [],
+                start: `${token}-new`,
+                end: `${token}-new`,
+            }));
+
+            function timelines(): [string | null, string | null][] {
+                return thread.timelineSet.getTimelines()
+                    .map(it => [it.getPaginationToken(Direction.Backward), it.getPaginationToken(Direction.Forward)]);
+            }
+
+            expect(timelines()).toEqual([[null, null]]);
+            const promise = thread.resetLiveTimeline("b1", "f1");
+            expect(timelines()).toEqual([[null, "f1"], ["b1", null]]);
+            thread.timelineSet.getTimelines()[0].setPaginationToken("f2", Direction.Forward);
+            thread.timelineSet.getTimelines()[1].setPaginationToken("b2", Direction.Backward);
+            await promise;
+            expect(timelines()).toEqual([[null, "f2"], ["b2", null]]);
+        });
+    });
 });