mirror of
https://github.com/element-hq/element-web.git
synced 2024-12-17 02:21:45 +03:00
e8b92b308b
* Conform more code to strict null checking * delint * Iterate * delint * Fix bad test
134 lines
4.8 KiB
TypeScript
134 lines
4.8 KiB
TypeScript
/*
|
|
Copyright 2019 New Vector Ltd
|
|
Copyright 2022 The Matrix.org Foundation C.I.C.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
import React from "react";
|
|
import { fireEvent, render, screen, waitForElementToBeRemoved } from "@testing-library/react";
|
|
import { createClient, MatrixClient } from "matrix-js-sdk/src/matrix";
|
|
import { MatrixError } from "matrix-js-sdk/src/http-api/errors";
|
|
import { mocked } from "jest-mock";
|
|
import fetchMock from "fetch-mock-jest";
|
|
|
|
import SdkConfig, { DEFAULTS } from "../../../../src/SdkConfig";
|
|
import { mkServerConfig, mockPlatformPeg, unmockPlatformPeg } from "../../../test-utils";
|
|
import Registration from "../../../../src/components/structures/auth/Registration";
|
|
|
|
jest.mock("matrix-js-sdk/src/matrix");
|
|
jest.useFakeTimers();
|
|
|
|
describe("Registration", function () {
|
|
const registerRequest = jest.fn();
|
|
const mockClient = mocked({
|
|
registerRequest,
|
|
loginFlows: jest.fn(),
|
|
} as unknown as MatrixClient);
|
|
|
|
beforeEach(function () {
|
|
SdkConfig.put({
|
|
...DEFAULTS,
|
|
disable_custom_urls: true,
|
|
});
|
|
mockClient.registerRequest.mockRejectedValueOnce(
|
|
new MatrixError(
|
|
{
|
|
flows: [{ stages: [] }],
|
|
},
|
|
401,
|
|
),
|
|
);
|
|
mockClient.loginFlows.mockClear().mockResolvedValue({ flows: [{ type: "m.login.password" }] });
|
|
mocked(createClient).mockImplementation((opts) => {
|
|
mockClient.idBaseUrl = opts.idBaseUrl;
|
|
mockClient.baseUrl = opts.baseUrl;
|
|
return mockClient;
|
|
});
|
|
fetchMock.get("https://matrix.org/_matrix/client/versions", {
|
|
unstable_features: {},
|
|
versions: [],
|
|
});
|
|
mockPlatformPeg({
|
|
startSingleSignOn: jest.fn(),
|
|
});
|
|
});
|
|
|
|
afterEach(function () {
|
|
fetchMock.restore();
|
|
SdkConfig.unset(); // we touch the config, so clean up
|
|
unmockPlatformPeg();
|
|
});
|
|
|
|
const defaultProps = {
|
|
defaultDeviceDisplayName: "test-device-display-name",
|
|
makeRegistrationUrl: jest.fn(),
|
|
onLoggedIn: jest.fn(),
|
|
onLoginClick: jest.fn(),
|
|
onServerConfigChange: jest.fn(),
|
|
};
|
|
|
|
function getRawComponent(hsUrl = "https://matrix.org", isUrl = "https://vector.im") {
|
|
return <Registration {...defaultProps} serverConfig={mkServerConfig(hsUrl, isUrl)} />;
|
|
}
|
|
|
|
function getComponent(hsUrl?: string, isUrl?: string) {
|
|
return render(getRawComponent(hsUrl, isUrl));
|
|
}
|
|
|
|
it("should show server picker", async function () {
|
|
const { container } = getComponent();
|
|
expect(container.querySelector(".mx_ServerPicker")).toBeTruthy();
|
|
});
|
|
|
|
it("should show form when custom URLs disabled", async function () {
|
|
const { container } = getComponent();
|
|
await waitForElementToBeRemoved(() => screen.queryAllByLabelText("Loading…"));
|
|
expect(container.querySelector("form")).toBeTruthy();
|
|
});
|
|
|
|
it("should show SSO options if those are available", async () => {
|
|
mockClient.loginFlows.mockClear().mockResolvedValue({ flows: [{ type: "m.login.sso" }] });
|
|
const { container } = getComponent();
|
|
await waitForElementToBeRemoved(() => screen.queryAllByLabelText("Loading…"));
|
|
|
|
const ssoButton = container.querySelector(".mx_SSOButton");
|
|
expect(ssoButton).toBeTruthy();
|
|
});
|
|
|
|
it("should handle serverConfig updates correctly", async () => {
|
|
mockClient.loginFlows.mockResolvedValue({
|
|
flows: [
|
|
{
|
|
type: "m.login.sso",
|
|
},
|
|
],
|
|
});
|
|
|
|
const { container, rerender } = render(getRawComponent());
|
|
await waitForElementToBeRemoved(() => screen.queryAllByLabelText("Loading…"));
|
|
|
|
fireEvent.click(container.querySelector(".mx_SSOButton")!);
|
|
expect(registerRequest.mock.instances[0].baseUrl).toBe("https://matrix.org");
|
|
|
|
fetchMock.get("https://server2/_matrix/client/versions", {
|
|
unstable_features: {},
|
|
versions: [],
|
|
});
|
|
rerender(getRawComponent("https://server2"));
|
|
await waitForElementToBeRemoved(() => screen.queryAllByLabelText("Loading…"));
|
|
|
|
fireEvent.click(container.querySelector(".mx_SSOButton")!);
|
|
expect(registerRequest.mock.instances[1].baseUrl).toBe("https://server2");
|
|
});
|
|
});
|