mirror of https://github.com/vector-im/riot-web
				
				
				
			
		
			
				
	
	
		
			103 lines
		
	
	
		
			3.7 KiB
		
	
	
	
		
			TypeScript
		
	
	
			
		
		
	
	
			103 lines
		
	
	
		
			3.7 KiB
		
	
	
	
		
			TypeScript
		
	
	
/*
 | 
						|
Copyright 2016 OpenMarket Ltd
 | 
						|
Copyright 2022 The Matrix.org Foundation C.I.C.
 | 
						|
Copyright 2022 Callum Brown
 | 
						|
 | 
						|
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, RenderResult } from "@testing-library/react";
 | 
						|
 | 
						|
import InteractiveAuthComponent from "../../../../src/components/structures/InteractiveAuth";
 | 
						|
import { flushPromises, getMockClientWithEventEmitter, unmockClientPeg } from "../../../test-utils";
 | 
						|
 | 
						|
describe("InteractiveAuthComponent", function () {
 | 
						|
    const mockClient = getMockClientWithEventEmitter({
 | 
						|
        generateClientSecret: jest.fn().mockReturnValue("t35tcl1Ent5ECr3T"),
 | 
						|
    });
 | 
						|
 | 
						|
    const defaultProps = {
 | 
						|
        matrixClient: mockClient,
 | 
						|
        makeRequest: jest.fn().mockResolvedValue(undefined),
 | 
						|
        onAuthFinished: jest.fn(),
 | 
						|
    };
 | 
						|
    const getComponent = (props = {}) => render(<InteractiveAuthComponent {...defaultProps} {...props} />);
 | 
						|
 | 
						|
    beforeEach(function () {
 | 
						|
        jest.clearAllMocks();
 | 
						|
    });
 | 
						|
 | 
						|
    afterAll(() => {
 | 
						|
        unmockClientPeg();
 | 
						|
    });
 | 
						|
 | 
						|
    const getSubmitButton = ({ container }: RenderResult) =>
 | 
						|
        container.querySelector(".mx_AccessibleButton_kind_primary");
 | 
						|
    const getRegistrationTokenInput = ({ container }: RenderResult) =>
 | 
						|
        container.querySelector('input[name="registrationTokenField"]');
 | 
						|
 | 
						|
    it("Should successfully complete a registration token flow", async () => {
 | 
						|
        const onAuthFinished = jest.fn();
 | 
						|
        const makeRequest = jest.fn().mockResolvedValue({ a: 1 });
 | 
						|
 | 
						|
        const authData = {
 | 
						|
            session: "sess",
 | 
						|
            flows: [{ stages: ["m.login.registration_token"] }],
 | 
						|
        };
 | 
						|
 | 
						|
        const wrapper = getComponent({ makeRequest, onAuthFinished, authData });
 | 
						|
 | 
						|
        const registrationTokenNode = getRegistrationTokenInput(wrapper);
 | 
						|
        const submitNode = getSubmitButton(wrapper);
 | 
						|
        const formNode = wrapper.container.querySelector("form");
 | 
						|
 | 
						|
        expect(registrationTokenNode).toBeTruthy();
 | 
						|
        expect(submitNode).toBeTruthy();
 | 
						|
        expect(formNode).toBeTruthy();
 | 
						|
 | 
						|
        // submit should be disabled
 | 
						|
        expect(submitNode).toHaveAttribute("disabled");
 | 
						|
        expect(submitNode).toHaveAttribute("aria-disabled", "true");
 | 
						|
 | 
						|
        // put something in the registration token box
 | 
						|
        fireEvent.change(registrationTokenNode!, { target: { value: "s3kr3t" } });
 | 
						|
 | 
						|
        expect(getRegistrationTokenInput(wrapper)).toHaveValue("s3kr3t");
 | 
						|
        expect(submitNode).not.toHaveAttribute("disabled");
 | 
						|
        expect(submitNode).not.toHaveAttribute("aria-disabled", "true");
 | 
						|
 | 
						|
        // hit enter; that should trigger a request
 | 
						|
        fireEvent.submit(formNode!);
 | 
						|
 | 
						|
        // wait for auth request to resolve
 | 
						|
        await flushPromises();
 | 
						|
 | 
						|
        expect(makeRequest).toHaveBeenCalledTimes(1);
 | 
						|
        expect(makeRequest).toHaveBeenCalledWith(
 | 
						|
            expect.objectContaining({
 | 
						|
                session: "sess",
 | 
						|
                type: "m.login.registration_token",
 | 
						|
                token: "s3kr3t",
 | 
						|
            }),
 | 
						|
        );
 | 
						|
 | 
						|
        expect(onAuthFinished).toHaveBeenCalledTimes(1);
 | 
						|
        expect(onAuthFinished).toHaveBeenCalledWith(
 | 
						|
            true,
 | 
						|
            { a: 1 },
 | 
						|
            { clientSecret: "t35tcl1Ent5ECr3T", emailSid: undefined },
 | 
						|
        );
 | 
						|
    });
 | 
						|
});
 |