more tests

test/components/views/beacon/RoomCallBanner-test.tsx

@@ -43,6 +43,7 @@ import { CallStore } from "../../../../src/stores/CallStore";
 import { WidgetMessagingStore } from "../../../../src/stores/widgets/WidgetMessagingStore";
 import { MatrixClientPeg } from "../../../../src/MatrixClientPeg";
 import { RoomViewStore } from "../../../../src/stores/RoomViewStore";
+import { ConnectionState } from "../../../../src/models/Call";
 
 describe("<RoomLiveShareWarning />", () => {
     let client: Mocked<MatrixClient>;
@@ -119,11 +120,19 @@ describe("<RoomLiveShareWarning />", () => {
             expect(videoCallLabel.innerHTML).toBe("Video call");
         });
 
-        it("show Join button if the user has not joined", async () => {
+        it("shows Join button if the user has not joined", async () => {
             await renderBanner();
             const videoCallLabel = await screen.findByText("Join");
             expect(videoCallLabel.innerHTML).toBe("Join");
         });
+
+        it("shows Leave button if the user has not joined", async () => {
+            call.setConnectionState(ConnectionState.Connected);
+            await renderBanner();
+            const videoCallLabel = await screen.findByText("Leave");
+            expect(videoCallLabel.innerHTML).toBe("Leave");
+        });
+
         it("dont show banner if the call is shown", async () => {
             jest.spyOn(RoomViewStore.instance, 'isViewingCall').mockReturnValue(false);
             await renderBanner();

test/test-utils/call.ts

@@ -20,7 +20,7 @@ import type { Room } from "matrix-js-sdk/src/models/room";
 import type { RoomMember } from "matrix-js-sdk/src/models/room-member";
 import type { MatrixEvent } from "matrix-js-sdk/src/models/event";
 import { mkEvent } from "./test-utils";
-import { Call, ElementCall, JitsiCall } from "../../src/models/Call";
+import { Call, ConnectionState, ElementCall, JitsiCall } from "../../src/models/Call";
 
 export class MockedCall extends Call {
     public static readonly EVENT_TYPE = "org.example.mocked_call";
@@ -68,6 +68,10 @@ export class MockedCall extends Call {
         super.participants = value;
     }
 
+    public setConnectionState(value: ConnectionState) {
+        super.connectionState = value;
+    }
+
     // No action needed for any of the following methods since this is just a mock
     protected getDevices(): string[] { return []; }
     protected async setDevices(): Promise<void> { }