106 lines
3.8 KiB
TypeScript
106 lines
3.8 KiB
TypeScript
/*
|
|
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 { getInitialScreenAfterLogin, onNewScreen } from "../../../src/vector/routing";
|
|
|
|
describe("onNewScreen", () => {
|
|
it("should replace history if stripping via fields", () => {
|
|
Object.defineProperty(window, "location", {
|
|
value: {
|
|
hash: "#/room/!room:server?via=abc",
|
|
replace: jest.fn(),
|
|
assign: jest.fn(),
|
|
},
|
|
writable: true,
|
|
});
|
|
onNewScreen("room/!room:server");
|
|
expect(window.location.assign).not.toHaveBeenCalled();
|
|
expect(window.location.replace).toHaveBeenCalled();
|
|
});
|
|
|
|
it("should not replace history if changing rooms", () => {
|
|
Object.defineProperty(window, "location", {
|
|
value: {
|
|
hash: "#/room/!room1:server?via=abc",
|
|
replace: jest.fn(),
|
|
assign: jest.fn(),
|
|
},
|
|
writable: true,
|
|
});
|
|
onNewScreen("room/!room2:server");
|
|
expect(window.location.assign).toHaveBeenCalled();
|
|
expect(window.location.replace).not.toHaveBeenCalled();
|
|
});
|
|
});
|
|
|
|
describe("getInitialScreenAfterLogin", () => {
|
|
beforeEach(() => {
|
|
jest.spyOn(sessionStorage.__proto__, "getItem").mockClear().mockReturnValue(null);
|
|
jest.spyOn(sessionStorage.__proto__, "setItem").mockClear();
|
|
});
|
|
|
|
const makeMockLocation = (hash = "") => {
|
|
const url = new URL("https://test.org");
|
|
url.hash = hash;
|
|
return url as unknown as Location;
|
|
};
|
|
|
|
describe("when current url has no hash", () => {
|
|
it("does not set an initial screen in session storage", () => {
|
|
getInitialScreenAfterLogin(makeMockLocation());
|
|
expect(sessionStorage.setItem).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it("returns undefined when there is no initial screen in session storage", () => {
|
|
expect(getInitialScreenAfterLogin(makeMockLocation())).toBeUndefined();
|
|
});
|
|
|
|
it("returns initial screen from session storage", () => {
|
|
const screen = {
|
|
screen: "/room/!test",
|
|
};
|
|
jest.spyOn(sessionStorage.__proto__, "getItem").mockReturnValue(JSON.stringify(screen));
|
|
expect(getInitialScreenAfterLogin(makeMockLocation())).toEqual(screen);
|
|
});
|
|
});
|
|
|
|
describe("when current url has a hash", () => {
|
|
it("sets an initial screen in session storage", () => {
|
|
const hash = "/room/!test";
|
|
getInitialScreenAfterLogin(makeMockLocation(hash));
|
|
expect(sessionStorage.setItem).toHaveBeenCalledWith(
|
|
"mx_screen_after_login",
|
|
JSON.stringify({
|
|
screen: "room/!test",
|
|
params: {},
|
|
}),
|
|
);
|
|
});
|
|
|
|
it("sets an initial screen in session storage with params", () => {
|
|
const hash = "/room/!test?param=test";
|
|
getInitialScreenAfterLogin(makeMockLocation(hash));
|
|
expect(sessionStorage.setItem).toHaveBeenCalledWith(
|
|
"mx_screen_after_login",
|
|
JSON.stringify({
|
|
screen: "room/!test",
|
|
params: { param: "test" },
|
|
}),
|
|
);
|
|
});
|
|
});
|
|
});
|