123 lines
4.7 KiB
TypeScript
123 lines
4.7 KiB
TypeScript
/*
|
|
Copyright 2024 New Vector Ltd.
|
|
Copyright 2023 The Matrix.org Foundation C.I.C.
|
|
|
|
SPDX-License-Identifier: AGPL-3.0-only OR GPL-3.0-only
|
|
Please see LICENSE files in the repository root for full details.
|
|
*/
|
|
|
|
import { Page } from "@playwright/test";
|
|
|
|
import { test, expect } from "../../element-web-test";
|
|
import { doTokenRegistration } from "./utils";
|
|
import { Credentials } from "../../plugins/homeserver";
|
|
import { isDendrite } from "../../plugins/homeserver/dendrite";
|
|
|
|
test.describe("Soft logout", () => {
|
|
test.use({
|
|
displayName: "Alice",
|
|
startHomeserverOpts: ({ oAuthServer }, use) =>
|
|
use({
|
|
template: "default",
|
|
oAuthServerPort: oAuthServer.port,
|
|
}),
|
|
});
|
|
|
|
test.describe("with password user", () => {
|
|
test("shows the soft-logout page when a request fails, and allows a re-login", async ({ page, user }) => {
|
|
await interceptRequestsWithSoftLogout(page, user);
|
|
await expect(page.getByText("You're signed out")).toBeVisible();
|
|
await page.getByPlaceholder("Password").fill(user.password);
|
|
await page.getByPlaceholder("Password").press("Enter");
|
|
|
|
// back to the welcome page
|
|
await expect(page).toHaveURL(/\/#\/home/);
|
|
await expect(page.getByRole("heading", { name: `Welcome ${user.userId}`, exact: true })).toBeVisible();
|
|
});
|
|
|
|
test("still shows the soft-logout page when the page is reloaded after a soft-logout", async ({
|
|
page,
|
|
user,
|
|
}) => {
|
|
await interceptRequestsWithSoftLogout(page, user);
|
|
await expect(page.getByText("You're signed out")).toBeVisible();
|
|
await page.reload();
|
|
await expect(page.getByText("You're signed out")).toBeVisible();
|
|
});
|
|
});
|
|
|
|
test.describe("with SSO user", () => {
|
|
test.skip(isDendrite, "does not yet support SSO");
|
|
|
|
test.use({
|
|
user: async ({ page, homeserver }, use) => {
|
|
const user = await doTokenRegistration(page, homeserver);
|
|
|
|
// Eventually, we should end up at the home screen.
|
|
await expect(page).toHaveURL(/\/#\/home$/);
|
|
await expect(page.getByRole("heading", { name: "Welcome Alice", exact: true })).toBeVisible();
|
|
|
|
await use(user);
|
|
},
|
|
});
|
|
|
|
test("shows the soft-logout page when a request fails, and allows a re-login", async ({ page, user }) => {
|
|
await expect(page.getByRole("heading", { name: "Welcome Alice", exact: true })).toBeVisible();
|
|
|
|
await interceptRequestsWithSoftLogout(page, user);
|
|
|
|
await expect(page.getByText("You're signed out")).toBeVisible();
|
|
await page.getByRole("button", { name: "Continue with OAuth test" }).click();
|
|
|
|
// click the submit button
|
|
await page.getByRole("button", { name: "Submit" }).click();
|
|
|
|
// Synapse prompts us to grant permission to Element
|
|
await expect(page.getByRole("heading", { name: "Continue to your account" })).toBeVisible();
|
|
await page.getByRole("link", { name: "Continue" }).click();
|
|
|
|
// back to the welcome page
|
|
await expect(page).toHaveURL(/\/#\/home$/);
|
|
await expect(page.getByRole("heading", { name: "Welcome Alice", exact: true })).toBeVisible();
|
|
});
|
|
});
|
|
});
|
|
|
|
/**
|
|
* Intercept calls to /sync and have them fail with a soft-logout
|
|
*
|
|
* Any further requests to /sync with the same access token are blocked.
|
|
*/
|
|
async function interceptRequestsWithSoftLogout(page: Page, user: Credentials): Promise<void> {
|
|
await page.route("**/_matrix/client/*/sync*", async (route, req) => {
|
|
const accessToken = await req.headerValue("Authorization");
|
|
|
|
// now, if the access token on this request matches the expired one, block it
|
|
if (accessToken === `Bearer ${user.accessToken}`) {
|
|
console.log("Intercepting request with soft-logged-out access token");
|
|
await route.fulfill({
|
|
status: 401,
|
|
json: {
|
|
errcode: "M_UNKNOWN_TOKEN",
|
|
error: "Soft logout",
|
|
soft_logout: true,
|
|
},
|
|
});
|
|
return;
|
|
}
|
|
|
|
// otherwise, pass through as normal
|
|
await route.continue();
|
|
});
|
|
|
|
const promise = page.waitForResponse((resp) => resp.url().includes("/sync") && resp.status() === 401);
|
|
|
|
// do something to make the active /sync return: create a new room
|
|
await page.evaluate(() => {
|
|
// don't wait for this to complete: it probably won't, because of the broken sync
|
|
window.mxMatrixClientPeg.get().createRoom({});
|
|
});
|
|
|
|
await promise;
|
|
}
|