Trek_CN/server/tests/unit/services/ephemeralTokens.test.ts

import { describe, it, expect, beforeEach, vi } from 'vitest';

// Reset module between tests that need a fresh token store
beforeEach(() => {
  vi.resetModules();
});

describe('ephemeralTokens', () => {
  async function getModule() {
    return import('../../../src/services/ephemeralTokens');
  }

  // AUTH-030 — Resource token creation (single-use)
  describe('createEphemeralToken', () => {
    it('AUTH-030: creates a token and returns a hex string', async () => {
      const { createEphemeralToken } = await getModule();
      const token = createEphemeralToken(1, 'download');
      expect(token).not.toBeNull();
      expect(typeof token).toBe('string');
      expect(token!.length).toBe(64); // 32 bytes hex
    });

    it('AUTH-030: different calls produce different tokens', async () => {
      const { createEphemeralToken } = await getModule();
      const t1 = createEphemeralToken(1, 'download');
      const t2 = createEphemeralToken(1, 'download');
      expect(t1).not.toBe(t2);
    });
  });

  // AUTH-029 — WebSocket token expiry (single-use)
  describe('consumeEphemeralToken', () => {
    it('AUTH-030: token is consumed and returns userId on first use', async () => {
      const { createEphemeralToken, consumeEphemeralToken } = await getModule();
      const token = createEphemeralToken(42, 'download')!;
      const userId = consumeEphemeralToken(token, 'download');
      expect(userId).toBe(42);
    });

    it('AUTH-030: token is single-use — second consume returns null', async () => {
      const { createEphemeralToken, consumeEphemeralToken } = await getModule();
      const token = createEphemeralToken(42, 'download')!;
      consumeEphemeralToken(token, 'download'); // first use
      const second = consumeEphemeralToken(token, 'download'); // second use
      expect(second).toBeNull();
    });

    it('AUTH-029: purpose mismatch returns null', async () => {
      const { createEphemeralToken, consumeEphemeralToken } = await getModule();
      const token = createEphemeralToken(42, 'ws')!;
      const result = consumeEphemeralToken(token, 'download');
      expect(result).toBeNull();
    });

    it('AUTH-029: expired token returns null', async () => {
      vi.useFakeTimers();
      const { createEphemeralToken, consumeEphemeralToken } = await getModule();
      const token = createEphemeralToken(42, 'ws')!; // 30s TTL
      vi.advanceTimersByTime(31_000); // advance past expiry
      const result = consumeEphemeralToken(token, 'ws');
      expect(result).toBeNull();
      vi.useRealTimers();
    });

    it('returns null for unknown token', async () => {
      const { consumeEphemeralToken } = await getModule();
      const result = consumeEphemeralToken('nonexistent-token', 'download');
      expect(result).toBeNull();
    });
  });

  describe('startTokenCleanup / stopTokenCleanup', () => {
    it('startTokenCleanup starts the interval (second call is no-op)', async () => {
      vi.useFakeTimers();
      const { createEphemeralToken, consumeEphemeralToken, startTokenCleanup, stopTokenCleanup } = await getModule();
      startTokenCleanup();
      startTokenCleanup(); // should be no-op, not throw
      // Token created while cleanup is running should still be consumable (interval hasn't fired)
      const token = createEphemeralToken(1, 'ws')!;
      expect(consumeEphemeralToken(token, 'ws')).toBe(1);
      stopTokenCleanup();
      vi.useRealTimers();
    });

    it('stopTokenCleanup clears the interval and allows restart', async () => {
      vi.useFakeTimers();
      const { createEphemeralToken, consumeEphemeralToken, startTokenCleanup, stopTokenCleanup } = await getModule();
      startTokenCleanup();
      stopTokenCleanup();
      stopTokenCleanup(); // calling stop twice should not throw
      startTokenCleanup(); // should be able to start again after stop
      stopTokenCleanup();
      // After stop, tokens should still be consumable (cleanup didn't run)
      const token = createEphemeralToken(2, 'download')!;
      expect(consumeEphemeralToken(token, 'download')).toBe(2);
      vi.useRealTimers();
    });

    it('cleanup interval removes expired tokens', async () => {
      vi.useFakeTimers();
      const { createEphemeralToken, consumeEphemeralToken, startTokenCleanup, stopTokenCleanup } = await getModule();
      startTokenCleanup();
      const token = createEphemeralToken(1, 'ws')!; // 30s TTL

      // Advance past TTL AND past cleanup interval (60s)
      vi.advanceTimersByTime(65_000);

      // Token should have been cleaned up by the interval
      const result = consumeEphemeralToken(token, 'ws');
      expect(result).toBeNull();

      stopTokenCleanup();
      vi.useRealTimers();
    });
  });
});