Avoid creating multiple call objects for the same widget (#30839)

* Extract some setup code out of the call tests * Don't force all rooms to be rechecked for calls when starting a call * Remove misleading unused group call callbacks The GroupCallEventHandler hasn't been relevant to our Element Call group calls for some time; instead we look at the state of the MatrixRTCSessionManager and WidgetStore to determine whether a call has been started. * Avoid creating multiple call objects for the same widget * fix test --------- Co-authored-by: Will Hunt <will@half-shot.uk>
2025-09-23 11:45:48 -04:00
parent c8d937655b
commit b5d8e63c6d
6 changed files with 162 additions and 107 deletions
--- a/test/test-utils/call.ts
+++ b/test/test-utils/call.ts
@@ -7,11 +7,29 @@ Please see LICENSE files in the repository root for full details.
 */

 import { MatrixWidgetType } from "matrix-widget-api";
+import {
+    type GroupCall,
+    Room,
+    type RoomMember,
+    type MatrixEvent,
+    type MatrixClient,
+    PendingEventOrdering,
+    KnownMembership,
+    RoomStateEvent,
+    type IContent,
+} from "matrix-js-sdk/src/matrix";
+import { mocked, type Mocked } from "jest-mock";
+import { type MatrixRTCSession } from "matrix-js-sdk/src/matrixrtc";

-import type { GroupCall, Room, RoomMember, MatrixEvent } from "matrix-js-sdk/src/matrix";
-import { mkEvent } from "./test-utils";
+import { mkEvent, mkRoomMember, setupAsyncStoreWithClient, stubClient } from "./test-utils";
 import { Call, type ConnectionState, ElementCall, JitsiCall } from "../../src/models/Call";
 import { CallStore } from "../../src/stores/CallStore";
+import { MatrixClientPeg } from "../../src/MatrixClientPeg";
+import DMRoomMap from "../../src/utils/DMRoomMap";
+import { MockEventEmitter } from "./client";
+import WidgetStore from "../../src/stores/WidgetStore";
+import { WidgetMessagingStore } from "../../src/stores/widgets/WidgetMessagingStore";
+import SettingsStore from "../../src/settings/SettingsStore";

 export class MockedCall extends Call {
    public static readonly EVENT_TYPE = "org.example.mocked_call";
@@ -105,8 +123,92 @@ export class MockedCall extends Call {
 /**
 * Sets up the call store to use mocked calls.
 */
-export const useMockedCalls = () => {
+export function useMockedCalls() {
    Call.get = (room) => MockedCall.get(room);
    JitsiCall.create = async (room) => MockedCall.create(room, "1");
    ElementCall.create = (room) => MockedCall.create(room, "1");
-};
+}
+
+/**
+ * Enables the feature flags required for call tests.
+ */
+export function enableCalls(): { enabledSettings: Set<string> } {
+    const enabledSettings = new Set(["feature_group_calls", "feature_video_rooms", "feature_element_call_video_rooms"]);
+    jest.spyOn(SettingsStore, "getValue").mockImplementation((settingName): any => {
+        if (settingName.startsWith("feature_")) return enabledSettings.has(settingName);
+        if (settingName === "activeCallRoomIds") return [];
+        return undefined;
+    });
+    return { enabledSettings };
+}
+
+export function setUpClientRoomAndStores(): {
+    client: Mocked<MatrixClient>;
+    room: Room;
+    alice: RoomMember;
+    bob: RoomMember;
+    carol: RoomMember;
+    roomSession: Mocked<MatrixRTCSession>;
+} {
+    stubClient();
+    const client = mocked<MatrixClient>(MatrixClientPeg.safeGet());
+    DMRoomMap.makeShared(client);
+
+    const room = new Room("!1:example.org", client, "@alice:example.org", {
+        pendingEventOrdering: PendingEventOrdering.Detached,
+    });
+
+    const alice = mkRoomMember(room.roomId, "@alice:example.org");
+    const bob = mkRoomMember(room.roomId, "@bob:example.org");
+    const carol = mkRoomMember(room.roomId, "@carol:example.org");
+    jest.spyOn(room, "getMember").mockImplementation((userId) => {
+        switch (userId) {
+            case alice.userId:
+                return alice;
+            case bob.userId:
+                return bob;
+            case carol.userId:
+                return carol;
+            default:
+                return null;
+        }
+    });
+
+    jest.spyOn(room, "getMyMembership").mockReturnValue(KnownMembership.Join);
+
+    client.getRoom.mockImplementation((roomId) => (roomId === room.roomId ? room : null));
+
+    const roomSession = new MockEventEmitter({
+        memberships: [],
+        getOldestMembership: jest.fn().mockReturnValue(undefined),
+        room,
+    }) as Mocked<MatrixRTCSession>;
+
+    client.matrixRTC.getRoomSession.mockReturnValue(roomSession);
+    client.getRooms.mockReturnValue([room]);
+    client.getUserId.mockReturnValue(alice.userId);
+    client.getDeviceId.mockReturnValue("alices_device");
+    client.reEmitter.reEmit(room, [RoomStateEvent.Events]);
+    client.sendStateEvent.mockImplementation(async (roomId, eventType, content, stateKey = "") => {
+        if (roomId !== room.roomId) throw new Error("Unknown room");
+        const event = mkEvent({
+            event: true,
+            type: eventType,
+            room: roomId,
+            user: alice.userId,
+            skey: stateKey,
+            content: content as IContent,
+        });
+        room.addLiveEvents([event], { addToState: true });
+        return { event_id: event.getId()! };
+    });
+
+    setupAsyncStoreWithClient(WidgetStore.instance, client);
+    setupAsyncStoreWithClient(WidgetMessagingStore.instance, client);
+
+    return { client, room, alice, bob, carol, roomSession };
+}
+
+export function cleanUpClientRoomAndStores(client: MatrixClient, room: Room) {
+    client.reEmitter.stopReEmitting(room, [RoomStateEvent.Events]);
+}
--- a/test/unit-tests/models/Call-test.ts
+++ b/test/unit-tests/models/Call-test.ts
@@ -11,12 +11,9 @@ import { mocked } from "jest-mock";
 import { waitFor } from "jest-matrix-react";
 import {
    RoomType,
-    Room,
+    type Room,
    RoomEvent,
    MatrixEvent,
-    RoomStateEvent,
-    PendingEventOrdering,
-    type IContent,
    type MatrixClient,
    type IMyDevice,
    type RoomMember,
@@ -41,15 +38,7 @@ import {
    ElementCall,
    ElementCallIntent,
 } from "../../../src/models/Call";
-import {
-    stubClient,
-    mkEvent,
-    mkRoomMember,
-    setupAsyncStoreWithClient,
-    mockPlatformPeg,
-    MockEventEmitter,
-} from "../../test-utils";
-import { MatrixClientPeg } from "../../../src/MatrixClientPeg";
+import { cleanUpClientRoomAndStores, enableCalls, mockPlatformPeg, setUpClientRoomAndStores } from "../../test-utils";
 import WidgetStore from "../../../src/stores/WidgetStore";
 import { WidgetMessagingStore } from "../../../src/stores/widgets/WidgetMessagingStore";
 import ActiveWidgetStore, { ActiveWidgetStoreEvent } from "../../../src/stores/ActiveWidgetStore";
@@ -60,80 +49,7 @@ import { type SettingKey } from "../../../src/settings/Settings.tsx";
 import SdkConfig from "../../../src/SdkConfig.ts";
 import DMRoomMap from "../../../src/utils/DMRoomMap.ts";

-const enabledSettings = new Set(["feature_group_calls", "feature_video_rooms", "feature_element_call_video_rooms"]);
-jest.spyOn(SettingsStore, "getValue").mockImplementation(
-    (settingName): any => enabledSettings.has(settingName) || undefined,
-);
-
-const setUpClientRoomAndStores = (): {
-    client: Mocked<MatrixClient>;
-    room: Room;
-    alice: RoomMember;
-    bob: RoomMember;
-    carol: RoomMember;
-    roomSession: Mocked<MatrixRTCSession>;
-} => {
-    stubClient();
-    const client = mocked<MatrixClient>(MatrixClientPeg.safeGet());
-    DMRoomMap.makeShared(client);
-
-    const room = new Room("!1:example.org", client, "@alice:example.org", {
-        pendingEventOrdering: PendingEventOrdering.Detached,
-    });
-
-    const alice = mkRoomMember(room.roomId, "@alice:example.org");
-    const bob = mkRoomMember(room.roomId, "@bob:example.org");
-    const carol = mkRoomMember(room.roomId, "@carol:example.org");
-    jest.spyOn(room, "getMember").mockImplementation((userId) => {
-        switch (userId) {
-            case alice.userId:
-                return alice;
-            case bob.userId:
-                return bob;
-            case carol.userId:
-                return carol;
-            default:
-                return null;
-        }
-    });
-
-    jest.spyOn(room, "getMyMembership").mockReturnValue(KnownMembership.Join);
-
-    client.getRoom.mockImplementation((roomId) => (roomId === room.roomId ? room : null));
-
-    const roomSession = new MockEventEmitter({
-        memberships: [],
-        getOldestMembership: jest.fn().mockReturnValue(undefined),
-    }) as Mocked<MatrixRTCSession>;
-
-    client.matrixRTC.getRoomSession.mockReturnValue(roomSession);
-    client.getRooms.mockReturnValue([room]);
-    client.getUserId.mockReturnValue(alice.userId);
-    client.getDeviceId.mockReturnValue("alices_device");
-    client.reEmitter.reEmit(room, [RoomStateEvent.Events]);
-    client.sendStateEvent.mockImplementation(async (roomId, eventType, content, stateKey = "") => {
-        if (roomId !== room.roomId) throw new Error("Unknown room");
-        const event = mkEvent({
-            event: true,
-            type: eventType,
-            room: roomId,
-            user: alice.userId,
-            skey: stateKey,
-            content: content as IContent,
-        });
-        room.addLiveEvents([event], { addToState: true });
-        return { event_id: event.getId()! };
-    });
-
-    setupAsyncStoreWithClient(WidgetStore.instance, client);
-    setupAsyncStoreWithClient(WidgetMessagingStore.instance, client);
-
-    return { client, room, alice, bob, carol, roomSession };
-};
-
-const cleanUpClientRoomAndStores = (client: MatrixClient, room: Room) => {
-    client.reEmitter.stopReEmitting(room, [RoomStateEvent.Events]);
-};
+const { enabledSettings } = enableCalls();

 const setUpWidget = (call: Call): { widget: Widget; messaging: Mocked<ClientWidgetApi> } => {
    call.widget.data = { ...call.widget, skipLobby: true };
--- a/test/unit-tests/stores/CallStore-test.ts
+++ b/test/unit-tests/stores/CallStore-test.ts
@@ -0,0 +1,38 @@
+/*
+ * Copyright 2025 New Vector Ltd.
+ *
+ * SPDX-License-Identifier: AGPL-3.0-only OR GPL-3.0-only OR LicenseRef-Element-Commercial
+ * Please see LICENSE files in the repository root for full details.
+ */
+
+import { type CallMembership, MatrixRTCSessionManagerEvents } from "matrix-js-sdk/src/matrixrtc";
+
+import { ElementCall } from "../../../src/models/Call";
+import { CallStore } from "../../../src/stores/CallStore";
+import {
+    setUpClientRoomAndStores,
+    cleanUpClientRoomAndStores,
+    setupAsyncStoreWithClient,
+    enableCalls,
+} from "../../test-utils";
+
+enableCalls();
+
+test("CallStore constructs one call for one MatrixRTC session", () => {
+    const { client, room } = setUpClientRoomAndStores();
+    try {
+        setupAsyncStoreWithClient(CallStore.instance, client);
+        const getSpy = jest.spyOn(ElementCall, "get");
+
+        // Simulate another user starting a new MatrixRTC session
+        const session = client.matrixRTC.getRoomSession(room);
+        session.memberships.push({} as CallMembership);
+        client.matrixRTC.emit(MatrixRTCSessionManagerEvents.SessionStarted, room.roomId, session);
+
+        expect(getSpy).toHaveBeenCalledTimes(1);
+        expect(getSpy).toHaveReturnedWith(expect.any(ElementCall));
+        expect(CallStore.instance.getCall(room.roomId)).not.toBe(null);
+    } finally {
+        cleanUpClientRoomAndStores(client, room);
+    }
+});