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 },
 | |
|         );
 | |
|     });
 | |
| });
 |