Use localStorage with Web Storage API (#23172)

test/common/auth/token_storage/token_storage.test.ts

@@ -0,0 +1,110 @@
+import { describe, it, expect, test, vi, afterEach, beforeEach } from "vitest";
+import type { AuthData } from "home-assistant-js-websocket";
+import { FallbackStorage } from "../../../test_helper/local-storage-fallback";
+
+describe("token_storage", () => {
+  beforeEach(() => {
+    vi.stubGlobal(
+      "window.__tokenCache",
+      (window.__tokenCache = {
+        tokens: undefined,
+        writeEnabled: undefined,
+      })
+    );
+    window.localStorage = new FallbackStorage();
+  });
+
+  afterEach(() => {
+    vi.resetAllMocks();
+    vi.resetModules();
+  });
+
+  test("initialize tokenCache", async () => {
+    vi.stubGlobal(
+      "window.__tokenCache",
+      (window.__tokenCache = undefined as any)
+    );
+
+    await import("../../../../src/common/auth/token_storage");
+
+    expect(window.__tokenCache).toEqual({
+      tokens: undefined,
+      writeEnabled: undefined,
+    });
+  });
+
+  test("should load tokens", async () => {
+    const tokens: AuthData = {
+      access_token: "test",
+      expires: 1800,
+      expires_in: 1800,
+      hassUrl: "http://localhost",
+      refresh_token: "refresh",
+      clientId: "client",
+    };
+
+    const getItemSpy = vi.fn(() => JSON.stringify(tokens));
+    window.localStorage.getItem = getItemSpy;
+
+    const { loadTokens } = await import(
+      "../../../../src/common/auth/token_storage"
+    );
+
+    const loadedTokens = loadTokens();
+    expect(loadedTokens).toEqual(tokens);
+
+    expect(window.__tokenCache.tokens).toEqual(tokens);
+    expect(window.__tokenCache.writeEnabled).toBe(true);
+    expect(getItemSpy).toHaveBeenCalledOnce();
+    expect(getItemSpy).toHaveBeenCalledWith("hassTokens");
+  });
+
+  test("should load null tokens", async () => {
+    const getItemSpy = vi.fn(() => "hello");
+    window.localStorage.getItem = getItemSpy;
+
+    const { loadTokens } = await import(
+      "../../../../src/common/auth/token_storage"
+    );
+
+    const loadedTokens = loadTokens();
+    expect(loadedTokens).toEqual(null);
+
+    expect(window.__tokenCache.tokens).toEqual(null);
+    expect(window.__tokenCache.writeEnabled).toBe(undefined);
+    expect(getItemSpy).toHaveBeenCalledOnce();
+    expect(getItemSpy).toHaveBeenCalledWith("hassTokens");
+  });
+
+  it("should enable write", async () => {
+    const { enableWrite } = await import(
+      "../../../../src/common/auth/token_storage"
+    );
+    enableWrite();
+    expect(window.__tokenCache.writeEnabled).toBe(true);
+  });
+
+  it("should enable write with tokens", async () => {
+    vi.stubGlobal(
+      "window.__tokenCache",
+      (window.__tokenCache = {
+        tokens: "testToken" as any,
+      })
+    );
+
+    const setItemSpy = vi.fn();
+    window.localStorage.setItem = setItemSpy;
+
+    const { enableWrite } = await import(
+      "../../../../src/common/auth/token_storage"
+    );
+
+    enableWrite();
+    expect(window.__tokenCache.writeEnabled).toBe(true);
+    expect(setItemSpy).toHaveBeenCalledOnce();
+    expect(setItemSpy).toHaveBeenCalledWith(
+      "hassTokens",
+      JSON.stringify("testToken")
+    );
+  });
+});