@oxyhq/core 5.5.0 → 7.0.0

package/src/session/__tests__/authStateStore.test.ts

@@ -0,0 +1,209 @@
+import {
+  createWebAuthStateStore,
+  createNativeAuthStateStore,
+  createMemoryAuthStateStore,
+  AUTH_STATE_STORAGE_KEY,
+  DEVICE_TOKEN_STORAGE_KEY,
+  type PersistedAuthState,
+  type NativeKeyValueStorage,
+} from '../authStateStore';
+
+const SAMPLE: PersistedAuthState = {
+  sessionId: 's-1',
+  refreshToken: 'r-abcdefghijklmnop',
+  userId: 'u-1',
+  deviceToken: 'd-1234567890',
+  accessToken: 'a-jwt',
+  expiresAt: '2030-01-01T00:00:00.000Z',
+};
+
+/** A minimal in-map Storage; setItem can be made to throw (quota simulation). */
+function makeFakeStorage(opts: { throwOnSet?: boolean } = {}): Storage {
+  const map = new Map<string, string>();
+  return {
+    getItem: (k: string) => (map.has(k) ? (map.get(k) as string) : null),
+    setItem: (k: string, v: string) => {
+      if (opts.throwOnSet) {
+        throw new DOMException('QuotaExceededError', 'QuotaExceededError');
+      }
+      map.set(k, v);
+    },
+    removeItem: (k: string) => {
+      map.delete(k);
+    },
+    clear: () => map.clear(),
+    key: (i: number) => Array.from(map.keys())[i] ?? null,
+    get length() {
+      return map.size;
+    },
+  } as Storage;
+}
+
+function installLocalStorage(storage: Storage): void {
+  Object.defineProperty(globalThis, 'localStorage', {
+    value: storage,
+    configurable: true,
+    writable: true,
+  });
+}
+
+afterEach(() => {
+  // Remove whatever localStorage the test installed (value or throwing getter).
+  if (Object.getOwnPropertyDescriptor(globalThis, 'localStorage')) {
+    delete (globalThis as { localStorage?: Storage }).localStorage;
+  }
+});
+
+describe('createWebAuthStateStore', () => {
+  it('round-trips a PersistedAuthState under the versioned key', async () => {
+    const storage = makeFakeStorage();
+    installLocalStorage(storage);
+    const store = createWebAuthStateStore();
+
+    await store.save(SAMPLE);
+    expect(storage.getItem(AUTH_STATE_STORAGE_KEY)).toBeTruthy();
+    expect(await store.load()).toEqual(SAMPLE);
+  });
+
+  it('clear() wipes the session but the deviceToken survives', async () => {
+    installLocalStorage(makeFakeStorage());
+    const store = createWebAuthStateStore();
+
+    await store.save(SAMPLE);
+    await store.saveDeviceToken('device-token-xyz');
+    await store.clear();
+
+    expect(await store.load()).toBeNull();
+    expect(await store.loadDeviceToken()).toBe('device-token-xyz');
+  });
+
+  it('persists the deviceToken under a separate long-lived key', async () => {
+    const storage = makeFakeStorage();
+    installLocalStorage(storage);
+    const store = createWebAuthStateStore();
+
+    await store.saveDeviceToken('dt');
+    expect(storage.getItem(DEVICE_TOKEN_STORAGE_KEY)).toBe('dt');
+    await store.clearDeviceToken();
+    expect(await store.loadDeviceToken()).toBeNull();
+  });
+
+  it('returns null for a malformed or incomplete blob', async () => {
+    const storage = makeFakeStorage();
+    installLocalStorage(storage);
+    const store = createWebAuthStateStore();
+
+    storage.setItem(AUTH_STATE_STORAGE_KEY, 'not-json');
+    expect(await store.load()).toBeNull();
+
+    storage.setItem(AUTH_STATE_STORAGE_KEY, JSON.stringify({ sessionId: 's', userId: 'u' }));
+    expect(await store.load()).toBeNull();
+  });
+
+  it('degrades to in-memory when accessing localStorage throws (sandboxed iframe SecurityError)', async () => {
+    Object.defineProperty(globalThis, 'localStorage', {
+      configurable: true,
+      get() {
+        throw new DOMException('The operation is insecure.', 'SecurityError');
+      },
+    });
+
+    // Construction must not throw, and the store must still function (in-memory).
+    const store = createWebAuthStateStore();
+    await expect(store.save(SAMPLE)).resolves.toBeUndefined();
+    expect(await store.load()).toEqual(SAMPLE);
+  });
+
+  it('swallows a write that throws (quota / private mode) without rejecting', async () => {
+    installLocalStorage(makeFakeStorage({ throwOnSet: true }));
+    const store = createWebAuthStateStore();
+
+    await expect(store.save(SAMPLE)).resolves.toBeUndefined();
+    // The write never reached storage...
+    expect(localStorage.getItem(AUTH_STATE_STORAGE_KEY)).toBeNull();
+    // ...but the in-memory mirror keeps the session live for this page's lifetime.
+    expect(await store.load()).toEqual(SAMPLE);
+  });
+
+  it('the mirror keeps the deviceToken live when the write throws', async () => {
+    installLocalStorage(makeFakeStorage({ throwOnSet: true }));
+    const store = createWebAuthStateStore();
+
+    await store.saveDeviceToken('dt-mirrored');
+    expect(await store.loadDeviceToken()).toBe('dt-mirrored');
+  });
+
+  it('a cleared session reads null even if storage later holds a stale blob (mirror wins)', async () => {
+    const storage = makeFakeStorage();
+    installLocalStorage(storage);
+    const store = createWebAuthStateStore();
+
+    await store.save(SAMPLE);
+    await store.clear();
+    // Something (another tab / a failed remove) leaves a stale blob behind.
+    storage.setItem(AUTH_STATE_STORAGE_KEY, JSON.stringify(SAMPLE));
+    // The authoritative in-memory mirror still reports the cleared state.
+    expect(await store.load()).toBeNull();
+  });
+});
+
+describe('createNativeAuthStateStore', () => {
+  function makeNativeStorage(): NativeKeyValueStorage & { map: Map<string, string> } {
+    const map = new Map<string, string>();
+    return {
+      map,
+      getItem: async (k) => (map.has(k) ? (map.get(k) as string) : null),
+      setItem: async (k, v) => {
+        map.set(k, v);
+      },
+      removeItem: async (k) => {
+        map.delete(k);
+      },
+    };
+  }
+
+  it('round-trips through the injected async storage', async () => {
+    const storage = makeNativeStorage();
+    const store = createNativeAuthStateStore(storage);
+
+    await store.save(SAMPLE);
+    expect(storage.map.get(AUTH_STATE_STORAGE_KEY)).toBeTruthy();
+    expect(await store.load()).toEqual(SAMPLE);
+  });
+
+  it('deviceToken survives a session clear', async () => {
+    const store = createNativeAuthStateStore(makeNativeStorage());
+    await store.save(SAMPLE);
+    await store.saveDeviceToken('dt-native');
+    await store.clear();
+    expect(await store.load()).toBeNull();
+    expect(await store.loadDeviceToken()).toBe('dt-native');
+  });
+
+  it('keeps the session live via the mirror when the injected storage throws (locked keychain)', async () => {
+    const store = createNativeAuthStateStore({
+      getItem: async () => {
+        throw new Error('secure store locked');
+      },
+      setItem: async () => {
+        throw new Error('secure store locked');
+      },
+      removeItem: async () => {
+        throw new Error('secure store locked');
+      },
+    });
+    await expect(store.save(SAMPLE)).resolves.toBeUndefined();
+    // The write threw, but the in-memory mirror preserves the session.
+    expect(await store.load()).toEqual(SAMPLE);
+  });
+});
+
+describe('createMemoryAuthStateStore', () => {
+  it('round-trips and clears in process memory', async () => {
+    const store = createMemoryAuthStateStore();
+    await store.save(SAMPLE);
+    expect(await store.load()).toEqual(SAMPLE);
+    await store.clear();
+    expect(await store.load()).toBeNull();
+  });
+});

package/src/session/__tests__/refresh.test.ts

@@ -0,0 +1,256 @@
+import type { OxyServices } from '../../OxyServices';
+import type { TokenRefreshResponse } from '@oxyhq/contracts';
+import type { SessionLoginResponse } from '../../models/session';
+import { refreshPersistedSession, startTokenRefreshScheduler } from '../refresh';
+import { createMemoryAuthStateStore, type PersistedAuthState } from '../authStateStore';
+
+const STORED: PersistedAuthState = {
+  sessionId: 'sess-old',
+  refreshToken: 'refresh-old-abcdefghij',
+  userId: 'user-1',
+  deviceToken: 'device-abcdefghij',
+};
+
+const ROTATED: TokenRefreshResponse = {
+  accessToken: 'access-new',
+  refreshToken: 'refresh-new-abcdefghij',
+  expiresAt: '2030-01-01T00:00:00.000Z',
+  sessionId: 'sess-new',
+};
+
+interface RefreshMockOverrides {
+  refreshWithToken?: OxyServices['refreshWithToken'];
+  signInWithSharedIdentity?: OxyServices['signInWithSharedIdentity'];
+}
+
+function makeOxy(overrides: RefreshMockOverrides = {}): { oxy: OxyServices; setTokens: jest.Mock } {
+  const setTokens = jest.fn();
+  const oxy = {
+    setTokens,
+    refreshWithToken: overrides.refreshWithToken ?? (async () => ROTATED),
+    signInWithSharedIdentity: overrides.signInWithSharedIdentity ?? (async () => null),
+  } as unknown as OxyServices;
+  return { oxy, setTokens };
+}
+
+describe('refreshPersistedSession — arm 1 (refresh-token rotation)', () => {
+  it('rotates, plants, persists the new family head, and returns the new token', async () => {
+    const store = createMemoryAuthStateStore();
+    await store.save(STORED);
+    const { oxy, setTokens } = makeOxy();
+
+    const token = await refreshPersistedSession({ oxy, store, allowSharedKeyFallback: false });
+
+    expect(token).toBe('access-new');
+    expect(setTokens).toHaveBeenCalledWith('access-new');
+    expect(await store.load()).toEqual({
+      sessionId: 'sess-new',
+      refreshToken: 'refresh-new-abcdefghij',
+      userId: 'user-1',
+      deviceToken: 'device-abcdefghij',
+      accessToken: 'access-new',
+      expiresAt: '2030-01-01T00:00:00.000Z',
+    });
+  });
+
+  it('clears the store on a family-revoked (401) error', async () => {
+    const store = createMemoryAuthStateStore();
+    await store.save(STORED);
+    const { oxy } = makeOxy({
+      refreshWithToken: async () => {
+        throw Object.assign(new Error('revoked'), { status: 401 });
+      },
+    });
+
+    const token = await refreshPersistedSession({ oxy, store, allowSharedKeyFallback: false });
+
+    expect(token).toBeNull();
+    expect(await store.load()).toBeNull();
+  });
+
+  it('clears the store on an invalid_grant code', async () => {
+    const store = createMemoryAuthStateStore();
+    await store.save(STORED);
+    const { oxy } = makeOxy({
+      refreshWithToken: async () => {
+        throw Object.assign(new Error('bad'), { code: 'invalid_grant' });
+      },
+    });
+
+    expect(await refreshPersistedSession({ oxy, store, allowSharedKeyFallback: false })).toBeNull();
+    expect(await store.load()).toBeNull();
+  });
+
+  it('KEEPS the store on a transient (500 / network) error', async () => {
+    const store = createMemoryAuthStateStore();
+    await store.save(STORED);
+    const { oxy } = makeOxy({
+      refreshWithToken: async () => {
+        throw Object.assign(new Error('server'), { status: 500 });
+      },
+    });
+
+    expect(await refreshPersistedSession({ oxy, store, allowSharedKeyFallback: false })).toBeNull();
+    expect(await store.load()).toEqual(STORED);
+  });
+});
+
+describe('refreshPersistedSession — arm 2 (native shared-key fallback)', () => {
+  const SHARED_SESSION: SessionLoginResponse = {
+    sessionId: 'sess-shared',
+    deviceId: 'dev-1',
+    expiresAt: '2030-01-01T00:00:00.000Z',
+    user: { id: 'user-1', username: 'u', name: {}, avatar: undefined },
+    accessToken: 'access-shared',
+  };
+
+  it('re-mints via shared identity when there is no refresh token', async () => {
+    const store = createMemoryAuthStateStore(); // no stored refresh token
+    const signInWithSharedIdentity = jest.fn(async () => SHARED_SESSION);
+    const { oxy } = makeOxy({ signInWithSharedIdentity });
+
+    const token = await refreshPersistedSession({ oxy, store, allowSharedKeyFallback: true });
+
+    expect(token).toBe('access-shared');
+    expect(signInWithSharedIdentity).toHaveBeenCalledTimes(1);
+  });
+
+  it('falls back to shared-key after a revoked refresh token', async () => {
+    const store = createMemoryAuthStateStore();
+    await store.save(STORED);
+    const signInWithSharedIdentity = jest.fn(async () => SHARED_SESSION);
+    const { oxy } = makeOxy({
+      refreshWithToken: async () => {
+        throw Object.assign(new Error('revoked'), { status: 403 });
+      },
+      signInWithSharedIdentity,
+    });
+
+    const token = await refreshPersistedSession({ oxy, store, allowSharedKeyFallback: true });
+
+    expect(token).toBe('access-shared');
+    expect(await store.load()).toBeNull(); // revoked family was cleared
+    expect(signInWithSharedIdentity).toHaveBeenCalledTimes(1);
+  });
+
+  it('does NOT try shared-key when the fallback is disabled (web)', async () => {
+    const store = createMemoryAuthStateStore();
+    const signInWithSharedIdentity = jest.fn(async () => SHARED_SESSION);
+    const { oxy } = makeOxy({ signInWithSharedIdentity });
+
+    expect(await refreshPersistedSession({ oxy, store, allowSharedKeyFallback: false })).toBeNull();
+    expect(signInWithSharedIdentity).not.toHaveBeenCalled();
+  });
+});
+
+describe('startTokenRefreshScheduler', () => {
+  function makeSchedulerOxy(expiresInSeconds: number | null): {
+    oxy: OxyServices;
+    refreshAccessToken: jest.Mock;
+  } {
+    let cleared = false;
+    const refreshAccessToken = jest.fn(async () => {
+      cleared = true; // stop the reschedule loop after the first fire
+      return null;
+    });
+    const nowSec = Math.floor(Date.now() / 1000);
+    const oxy = {
+      getAccessToken: () => (cleared ? null : 'tok'),
+      getAccessTokenExpiry: () => (expiresInSeconds === null ? null : nowSec + expiresInSeconds),
+      onTokensChanged: () => () => undefined,
+      httpService: { refreshAccessToken },
+    } as unknown as OxyServices;
+    return { oxy, refreshAccessToken };
+  }
+
+  afterEach(() => {
+    jest.useRealTimers();
+  });
+
+  it('fires a refresh ~60s before expiry', async () => {
+    jest.useFakeTimers();
+    const { oxy, refreshAccessToken } = makeSchedulerOxy(120);
+    const handle = startTokenRefreshScheduler(oxy);
+
+    expect(refreshAccessToken).not.toHaveBeenCalled();
+    await jest.advanceTimersByTimeAsync(60_000);
+    expect(refreshAccessToken).toHaveBeenCalledWith('preflight');
+
+    handle.dispose();
+  });
+
+  it('no-ops cleanly for an opaque token with no exp', () => {
+    jest.useFakeTimers();
+    const { oxy, refreshAccessToken } = makeSchedulerOxy(null);
+    const handle = startTokenRefreshScheduler(oxy);
+    jest.advanceTimersByTime(10 * 60_000);
+    expect(refreshAccessToken).not.toHaveBeenCalled();
+    handle.dispose();
+  });
+
+  it('backs off on repeated failure — bounded attempts, never a zero-delay busy loop', async () => {
+    jest.useFakeTimers();
+    // Already-expired token + a refresh that ALWAYS fails. The old code floored
+    // the delay to 0 and re-armed in the finally block → a tight loop that would
+    // exhaust jest's 100k-timer guard. The backoff schedule fires only a handful
+    // of times per simulated minute.
+    const refreshAccessToken = jest.fn(async () => null);
+    const nowSec = Math.floor(Date.now() / 1000);
+    const oxy = {
+      getAccessToken: () => 'tok',
+      getAccessTokenExpiry: () => nowSec - 10,
+      onTokensChanged: () => () => undefined,
+      httpService: { refreshAccessToken },
+    } as unknown as OxyServices;
+
+    const handle = startTokenRefreshScheduler(oxy);
+    await jest.advanceTimersByTimeAsync(60_000);
+
+    expect(refreshAccessToken.mock.calls.length).toBeGreaterThan(0);
+    expect(refreshAccessToken.mock.calls.length).toBeLessThan(10);
+    handle.dispose();
+  });
+
+  it('resets the backoff after a successful refresh', async () => {
+    jest.useFakeTimers();
+    // Expiry is always ~1h from NOW, so a successful refresh genuinely pushes
+    // the token out (as a real rotation would). Fail once, then succeed.
+    const refreshAccessToken = jest
+      .fn<Promise<string | null>, [unknown]>()
+      .mockResolvedValueOnce(null)
+      .mockResolvedValue('fresh');
+    const oxy = {
+      getAccessToken: () => 'tok',
+      getAccessTokenExpiry: () => Math.floor(Date.now() / 1000) + 3600,
+      onTokensChanged: () => () => undefined,
+      httpService: { refreshAccessToken },
+    } as unknown as OxyServices;
+
+    const handle = startTokenRefreshScheduler(oxy);
+    // First fire ~ (3600-60)s out (fails → 5s backoff), then the backoff fire
+    // succeeds and re-arms ~1h out from the (now-fresh) expiry.
+    await jest.advanceTimersByTimeAsync((3600 - 60) * 1000);
+    await jest.advanceTimersByTimeAsync(5_000);
+    const callsAfterSuccess = refreshAccessToken.mock.calls.length;
+    expect(callsAfterSuccess).toBe(2); // one failure + one success
+    // A further minute must NOT produce a burst — the healthy re-arm is ~1h out.
+    await jest.advanceTimersByTimeAsync(60_000);
+    expect(refreshAccessToken.mock.calls.length).toBe(callsAfterSuccess);
+    handle.dispose();
+  });
+
+  it('calls unref() on its timer so it never holds the event loop', () => {
+    const unref = jest.fn();
+    const setTimeoutSpy = jest
+      .spyOn(global, 'setTimeout')
+      .mockReturnValue({ unref } as unknown as ReturnType<typeof setTimeout>);
+    const { oxy } = makeSchedulerOxy(120);
+
+    const handle = startTokenRefreshScheduler(oxy);
+    expect(setTimeoutSpy).toHaveBeenCalled();
+    expect(unref).toHaveBeenCalled();
+
+    handle.dispose();
+    setTimeoutSpy.mockRestore();
+  });
+});