112 lines
		
	
	
		
			3.9 KiB
		
	
	
	
		
			TypeScript
		
	
	
			
		
		
	
	
			112 lines
		
	
	
		
			3.9 KiB
		
	
	
	
		
			TypeScript
		
	
	
| /*
 | |
| Copyright 2023 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.
 | |
| */
 | |
| 
 | |
| // eslint-disable-next-line deprecate/import
 | |
| import { mount } from "enzyme";
 | |
| import { sleep } from "matrix-js-sdk/src/utils";
 | |
| import React from "react";
 | |
| import { act } from "react-dom/test-utils";
 | |
| import { mocked } from "jest-mock";
 | |
| import { SlidingSync } from "matrix-js-sdk/src/sliding-sync";
 | |
| import { Room } from "matrix-js-sdk/src/matrix";
 | |
| 
 | |
| import { SlidingSyncRoomSearchOpts, useSlidingSyncRoomSearch } from "../../src/hooks/useSlidingSyncRoomSearch";
 | |
| import { MockEventEmitter, stubClient } from "../test-utils";
 | |
| import { SlidingSyncManager } from "../../src/SlidingSyncManager";
 | |
| 
 | |
| type RoomSearchHook = {
 | |
|     loading: boolean;
 | |
|     rooms: Room[];
 | |
|     search(opts: SlidingSyncRoomSearchOpts): Promise<boolean>;
 | |
| };
 | |
| 
 | |
| // hooks must be inside a React component else you get:
 | |
| // "Invalid hook call. Hooks can only be called inside of the body of a function component."
 | |
| function RoomSearchComponent(props: { onClick: (h: RoomSearchHook) => void }) {
 | |
|     const roomSearch = useSlidingSyncRoomSearch();
 | |
|     return <div onClick={() => props.onClick(roomSearch)} />;
 | |
| }
 | |
| 
 | |
| describe("useSlidingSyncRoomSearch", () => {
 | |
|     it("should display rooms when searching", async () => {
 | |
|         const client = stubClient();
 | |
|         const roomA = new Room("!a:localhost", client, client.getUserId()!);
 | |
|         const roomB = new Room("!b:localhost", client, client.getUserId()!);
 | |
|         const slidingSync = mocked(
 | |
|             new MockEventEmitter({
 | |
|                 getListData: jest.fn(),
 | |
|             }) as unknown as SlidingSync,
 | |
|         );
 | |
|         jest.spyOn(SlidingSyncManager.instance, "ensureListRegistered").mockResolvedValue({
 | |
|             ranges: [[0, 9]],
 | |
|         });
 | |
|         SlidingSyncManager.instance.slidingSync = slidingSync;
 | |
|         mocked(slidingSync.getListData).mockReturnValue({
 | |
|             joinedCount: 2,
 | |
|             roomIndexToRoomId: {
 | |
|                 0: roomA.roomId,
 | |
|                 1: roomB.roomId,
 | |
|             },
 | |
|         });
 | |
|         mocked(client.getRoom).mockImplementation((roomId) => {
 | |
|             switch (roomId) {
 | |
|                 case roomA.roomId:
 | |
|                     return roomA;
 | |
|                 case roomB.roomId:
 | |
|                     return roomB;
 | |
|                 default:
 | |
|                     return null;
 | |
|             }
 | |
|         });
 | |
| 
 | |
|         // first check that everything is empty and then do the search
 | |
|         let executeHook = (roomSearch: RoomSearchHook) => {
 | |
|             expect(roomSearch.loading).toBe(false);
 | |
|             expect(roomSearch.rooms).toEqual([]);
 | |
|             roomSearch.search({
 | |
|                 limit: 10,
 | |
|                 query: "foo",
 | |
|             });
 | |
|         };
 | |
|         const wrapper = mount(
 | |
|             <RoomSearchComponent
 | |
|                 onClick={(roomSearch: RoomSearchHook) => {
 | |
|                     executeHook(roomSearch);
 | |
|                 }}
 | |
|             />,
 | |
|         );
 | |
| 
 | |
|         // run the query
 | |
|         await act(async () => {
 | |
|             await sleep(1);
 | |
|             wrapper.simulate("click");
 | |
|             return act(() => sleep(1));
 | |
|         });
 | |
|         // now we expect there to be rooms
 | |
|         executeHook = (roomSearch) => {
 | |
|             expect(roomSearch.loading).toBe(false);
 | |
|             expect(roomSearch.rooms).toEqual([roomA, roomB]);
 | |
|         };
 | |
| 
 | |
|         // run the query
 | |
|         await act(async () => {
 | |
|             await sleep(1);
 | |
|             wrapper.simulate("click");
 | |
|             return act(() => sleep(1));
 | |
|         });
 | |
|     });
 | |
| });
 |