chore: clean up tests

Author: DanielRivers

lib/sessionManager/stores/expressSession.ts

@@ -0,0 +1,88 @@
+import { storageSettings } from "../index.js";
+import { SessionBase, StorageKeys, type SessionManager } from "../types.js";
+import { splitString } from "../utils.js";
+
+/**
+ * Provides a memory based session manager implementation for the browser.
+ * @class MemoryStorage
+ */
+export class MemoryStorage<V extends string = StorageKeys>
+  extends SessionBase<V>
+  implements SessionManager<V>
+{
+  private memCache: Record<string, unknown> = {};
+
+  /**
+   * Clears all items from session store.
+   * @returns {void}
+.04
+   */
+  async destroySession(): Promise<void> {
+    this.memCache = {};
+  }
+
+  /**
+   * Sets the provided key-value store to the memory cache.
+   * @param {string} itemKey
+   * @param {unknown} itemValue
+   * @returns {void}
+   */
+  async setSessionItem(
+    itemKey: V | StorageKeys,
+    itemValue: unknown,
+  ): Promise<void> {
+    // clear items first
+    await this.removeSessionItem(itemKey);
+
+    if (typeof itemValue === "string") {
+      splitString(itemValue, storageSettings.maxLength).forEach(
+        (splitValue, index) => {
+          this.memCache[`${storageSettings.keyPrefix}${itemKey}${index}`] =
+            splitValue;
+        },
+      );
+      return;
+    }
+    this.memCache[`${storageSettings.keyPrefix}${String(itemKey)}0`] =
+      itemValue;
+  }
+
+  /**
+   * Gets the item for the provided key from the memory cache.
+   * @param {string} itemKey
+   * @returns {unknown | null}
+   */
+  async getSessionItem(itemKey: V | StorageKeys): Promise<unknown | null> {
+    if (
+      this.memCache[`${storageSettings.keyPrefix}${String(itemKey)}0`] ===
+      undefined
+    ) {
+      return null;
+    }
+
+    let itemValue = "";
+    let index = 0;
+    let key = `${storageSettings.keyPrefix}${String(itemKey)}${index}`;
+    while (this.memCache[key] !== undefined) {
+      itemValue += this.memCache[key];
+      index++;
+      key = `${storageSettings.keyPrefix}${String(itemKey)}${index}`;
+    }
+
+    return itemValue;
+  }
+
+  /**
+   * Removes the item for the provided key from the memory cache.
+   * @param {string} itemKey
+   * @returns {void}
+   */
+  async removeSessionItem(itemKey: V | StorageKeys): Promise<void> {
+    // Remove all items with the key prefix
+    for (const key in this.memCache) {
+      if (key.startsWith(`${storageSettings.keyPrefix}${String(itemKey)}`)) {
+        delete this.memCache[key];
+      }
+    }
+  }
+}

lib/utils/exhangeAuthCode._ts

@@ -0,0 +1,57 @@
+import { getActiveStorage, StorageKeys } from "../main";
+
+export const exchangeAuthCode = async ({
+    domain,
+    clientId,
+    code,
+    getCallbackURLParams} : {
+    domain: string;
+    clientId: string;
+    code: string;
+    getCallbackURLParams: URL;
+  ): Promise<OAuth2CodeExchangeResponse> {
+    const activeStorage = getActiveStorage();
+    activeStorage.getSessionItem(StorageKeys.state);
+
+    
+    const [code, state] = super.getCallbackURLParams(callbackURL);
+    const storedStateKey = this.getCodeVerifierKey(state);
+    if (!storedStateKey?.endsWith(state)) {
+      throw new Error('Received state does not match stored state');
+    }
+
+    const getItem = isBrowserEnvironment()
+      ? (sessionManager as unknown as BrowserSessionManager).getSessionItemBrowser
+      : sessionManager.getSessionItem;
+
+    const storedState = (await getItem.call(sessionManager, storedStateKey)) as
+      | string
+      | null;
+    if (!storedState) {
+      throw new Error('Stored state not found');
+    }
+
+    const authFlowState = JSON.parse(storedState);
+    this.codeVerifier = authFlowState.codeVerifier;
+
+    const body = new URLSearchParams({
+      redirect_uri: this.config.redirectURL,
+      client_id: this.config.clientId,
+      code_verifier: this.codeVerifier!,
+      grant_type: 'authorization_code',
+      code,
+    });
+
+    const removeItem = isBrowserEnvironment()
+      ? (sessionManager as unknown as BrowserSessionManager).removeSessionItemBrowser
+      : sessionManager.removeSessionItem;
+
+    activeStorage.removeSessionItem(StorageKeys.state);
+
+
+    try {
+      return await this.fetchTokensFor(sessionManager, body);
+    } finally {
+      await removeItem.call(sessionManager, this.getCodeVerifierKey(state));
+    }
+  }

lib/utils/token/index.test.ts

@@ -1,4 +1,4 @@
-import { describe, expect, it, beforeEach, vi } from "vitest";
+import { describe, expect, it } from "vitest";
 import { MemoryStorage } from "../../sessionManager";
 import {
   getActiveStorage,