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

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";

const HASS_URL = `${location.protocol}//${location.host}`;

describe("token_storage", () => {
  beforeEach(() => {
    vi.stubGlobal(
      "window.__tokenCache",
      (window.__tokenCache = {
        tokens: undefined,
        writeEnabled: undefined,
      })
    );
    vi.stubGlobal("__HASS_URL__", HASS_URL);
    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")
    );
  });
});