116 lines
4.4 KiB
TypeScript
116 lines
4.4 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.
|
|
*/
|
|
|
|
import { IdTokenClaims } from "oidc-client-ts";
|
|
import { decodeIdToken } from "matrix-js-sdk/src/matrix";
|
|
import { mocked } from "jest-mock";
|
|
|
|
import {
|
|
getStoredOidcClientId,
|
|
getStoredOidcIdToken,
|
|
getStoredOidcIdTokenClaims,
|
|
getStoredOidcTokenIssuer,
|
|
persistOidcAuthenticatedSettings,
|
|
} from "../../../src/utils/oidc/persistOidcSettings";
|
|
|
|
jest.mock("matrix-js-sdk/src/matrix");
|
|
|
|
describe("persist OIDC settings", () => {
|
|
jest.spyOn(Storage.prototype, "getItem");
|
|
jest.spyOn(Storage.prototype, "setItem");
|
|
|
|
beforeEach(() => {
|
|
localStorage.clear();
|
|
});
|
|
|
|
const clientId = "test-client-id";
|
|
const issuer = "https://auth.org/";
|
|
const idToken = "test-id-token";
|
|
const idTokenClaims: IdTokenClaims = {
|
|
// audience is this client
|
|
aud: "123",
|
|
// issuer matches
|
|
iss: issuer,
|
|
sub: "123",
|
|
exp: 123,
|
|
iat: 456,
|
|
};
|
|
|
|
describe("persistOidcAuthenticatedSettings", () => {
|
|
it("should set clientId and issuer in localStorage", () => {
|
|
persistOidcAuthenticatedSettings(clientId, issuer, idToken);
|
|
expect(localStorage.setItem).toHaveBeenCalledWith("mx_oidc_client_id", clientId);
|
|
expect(localStorage.setItem).toHaveBeenCalledWith("mx_oidc_token_issuer", issuer);
|
|
expect(localStorage.setItem).toHaveBeenCalledWith("mx_oidc_id_token", idToken);
|
|
});
|
|
});
|
|
|
|
describe("getStoredOidcTokenIssuer()", () => {
|
|
it("should return issuer from localStorage", () => {
|
|
localStorage.setItem("mx_oidc_token_issuer", issuer);
|
|
expect(getStoredOidcTokenIssuer()).toEqual(issuer);
|
|
expect(localStorage.getItem).toHaveBeenCalledWith("mx_oidc_token_issuer");
|
|
});
|
|
|
|
it("should return undefined when no issuer in localStorage", () => {
|
|
expect(getStoredOidcTokenIssuer()).toBeUndefined();
|
|
});
|
|
});
|
|
|
|
describe("getStoredOidcClientId()", () => {
|
|
it("should return clientId from localStorage", () => {
|
|
localStorage.setItem("mx_oidc_client_id", clientId);
|
|
expect(getStoredOidcClientId()).toEqual(clientId);
|
|
expect(localStorage.getItem).toHaveBeenCalledWith("mx_oidc_client_id");
|
|
});
|
|
it("should throw when no clientId in localStorage", () => {
|
|
expect(() => getStoredOidcClientId()).toThrow("Oidc client id not found in storage");
|
|
});
|
|
});
|
|
|
|
describe("getStoredOidcIdToken()", () => {
|
|
it("should return token from localStorage", () => {
|
|
localStorage.setItem("mx_oidc_id_token", idToken);
|
|
expect(getStoredOidcIdToken()).toEqual(idToken);
|
|
expect(localStorage.getItem).toHaveBeenCalledWith("mx_oidc_id_token");
|
|
});
|
|
|
|
it("should return undefined when no token in localStorage", () => {
|
|
expect(getStoredOidcIdToken()).toBeUndefined();
|
|
});
|
|
});
|
|
|
|
describe("getStoredOidcIdTokenClaims()", () => {
|
|
it("should return claims from localStorage", () => {
|
|
localStorage.setItem("mx_oidc_id_token_claims", JSON.stringify(idTokenClaims));
|
|
expect(getStoredOidcIdTokenClaims()).toEqual(idTokenClaims);
|
|
expect(localStorage.getItem).toHaveBeenCalledWith("mx_oidc_id_token_claims");
|
|
});
|
|
|
|
it("should return claims extracted from id_token in localStorage", () => {
|
|
localStorage.setItem("mx_oidc_id_token", idToken);
|
|
mocked(decodeIdToken).mockReturnValue(idTokenClaims);
|
|
expect(getStoredOidcIdTokenClaims()).toEqual(idTokenClaims);
|
|
expect(decodeIdToken).toHaveBeenCalledWith(idToken);
|
|
expect(localStorage.getItem).toHaveBeenCalledWith("mx_oidc_id_token_claims");
|
|
});
|
|
|
|
it("should return undefined when no claims in localStorage", () => {
|
|
expect(getStoredOidcIdTokenClaims()).toBeUndefined();
|
|
});
|
|
});
|
|
});
|