1
0
mirror of https://github.com/bitwarden/browser.git synced 2024-12-21 16:18:28 +01:00

[PM-4273] Add tests for TotpService (#7058)

Adds tests for the TOTP service in preparation for migrating it to the SDK.
This commit is contained in:
Oscar Hinton 2023-12-05 18:32:49 +01:00 committed by GitHub
parent 7bbdee9daa
commit 2eebf890b5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -0,0 +1,63 @@
import { mock } from "jest-mock-extended";
import { LogService } from "../../platform/abstractions/log.service";
import { WebCryptoFunctionService } from "../../platform/services/web-crypto-function.service";
import { TotpService } from "./totp.service";
describe("TotpService", () => {
let totpService: TotpService;
const logService = mock<LogService>();
beforeEach(() => {
totpService = new TotpService(new WebCryptoFunctionService(global), logService);
// TOTP is time-based, so we need to mock the current time
jest.useFakeTimers({
now: new Date("2023-01-01T00:00:00.000Z"),
});
});
afterEach(() => {
jest.clearAllMocks();
jest.useRealTimers();
});
it("should return null if key is null", async () => {
const result = await totpService.getCode(null);
expect(result).toBeNull();
});
it("should return a code if key is not null", async () => {
const result = await totpService.getCode("WQIQ25BRKZYCJVYP");
expect(result).toBe("194506");
});
it("should handle otpauth keys", async () => {
const key = "otpauth://totp/test-account?secret=WQIQ25BRKZYCJVYP";
const result = await totpService.getCode(key);
expect(result).toBe("194506");
const period = totpService.getTimeInterval(key);
expect(period).toBe(30);
});
it("should handle otpauth different period", async () => {
const key = "otpauth://totp/test-account?secret=WQIQ25BRKZYCJVYP&period=60";
const result = await totpService.getCode(key);
expect(result).toBe("730364");
const period = totpService.getTimeInterval(key);
expect(period).toBe(60);
});
it("should handle steam keys", async () => {
const key = "steam://HXDMVJECJJWSRB3HWIZR4IFUGFTMXBOZ";
const result = await totpService.getCode(key);
expect(result).toBe("7W6CJ");
const period = totpService.getTimeInterval(key);
expect(period).toBe(30);
});
});