feat: add clearActiveStorage and tests

Author: DanielRivers

lib/main.test.ts

@@ -50,6 +50,7 @@ describe("index exports", () => {
       // token utils
       "getActiveStorage",
       "hasActiveStorage",
+      "clearActiveStorage",
       "getClaim",
       "getClaims",
       "getCurrentOrganization",

lib/utils/generateAuthUrl.test.ts

@@ -1,6 +1,8 @@
 import { describe, it, expect } from "vitest";
 import { IssuerRouteTypes, LoginOptions, Scopes } from "../types";
 import { generateAuthUrl } from "./generateAuthUrl";
+import { MemoryStorage, StorageKeys } from "../sessionManager";
+import { setActiveStorage } from "./token";
 
 describe("generateAuthUrl", () => {
   it("should generate the correct auth URL with required parameters", async () => {
@@ -121,4 +123,37 @@ describe("generateAuthUrl", () => {
     result.url.searchParams.delete("state");
     expect(result.url.toString()).toBe(expectedUrl);
   });
+
+  it("should update state when active state found", async () => {
+
+    const store = new MemoryStorage();
+    setActiveStorage(store);
+
+    const domain = "https://auth.example.com";
+    const options: LoginOptions = {
+      clientId: "client123",
+      responseType: "code",
+      scope: [Scopes.openid, Scopes.profile],
+      loginHint: "[email protected]",
+      isCreateOrg: true,
+      connectionId: "conn123",
+      redirectURL: "https://example.com",
+      audience: "audience123",
+      prompt: "login",
+    };
+
+    await generateAuthUrl(
+      domain,
+      IssuerRouteTypes.login,
+      options,
+    );
+
+    const state = await store.getSessionItem(StorageKeys.state)
+    const nonce = await store.getSessionItem(StorageKeys.nonce)
+    const codeVerifier = await store.getSessionItem(StorageKeys.state)
+
+    expect(state).toBeDefined();
+    expect(nonce).toBeDefined();
+    expect(codeVerifier).toBeDefined();
+  });
 });

lib/utils/token/index.test.ts

@@ -0,0 +1,21 @@
+import { describe, expect, it, beforeEach, vi } from "vitest";
+import { MemoryStorage } from "../../sessionManager";
+import { getActiveStorage, hasActiveStorage, setActiveStorage, clearActiveStorage } from ".";
+
+describe("token index", () => {
+  it("hasActiveStorage", async () => {
+    const storage = new MemoryStorage();
+    setActiveStorage(storage);
+    expect(hasActiveStorage()).toStrictEqual(true);
+
+  });
+});
+
+describe("token index", () => {
+  it("hasActiveStorage when not set", async () => {
+    clearActiveStorage();
+    console.log(getActiveStorage());  
+    expect(hasActiveStorage()).toStrictEqual(false);
+  });
+});
+

lib/utils/token/index.ts

@@ -39,10 +39,16 @@ const hasActiveStorage = (): boolean => {
   return storage.value !== null;
 };
 
+const clearActiveStorage = (): void => {
+  storage.value = null
+};
+
+
 export {
   setActiveStorage,
   getActiveStorage,
   hasActiveStorage,
+  clearActiveStorage,
   getClaim,
   getClaims,
   getCurrentOrganization,