feat: 实现认证辅助工具

- 创建 Token 管理器 RoleAuthManager - 创建认证辅助类 AuthHelper - 支持 Token 注入和真实登录两种模式 - 实现 Token 缓存机制 - 添加完整的单元测试（5个测试用例全部通过）
2026-04-04 20:54:03 +08:00
parent 54ea704f27
commit 68070886d9
3 changed files with 210 additions and 0 deletions
@@ -0,0 +1,79 @@
+import { describe, it, expect, beforeEach, vi } from 'vitest';
+import { RoleAuthManager } from '../role-auth-manager';
+
+// Mock fetch
+global.fetch = vi.fn();
+
+describe('RoleAuthManager', () => {
+  beforeEach(() => {
+    RoleAuthManager.clearCache();
+    vi.clearAllMocks();
+  });
+
+  it('should authenticate and cache token', async () => {
+    const mockToken = 'mock-jwt-token-12345';
+    (global.fetch as any).mockResolvedValueOnce({
+      ok: true,
+      json: async () => ({ data: { token: mockToken } })
+    });
+
+    const token = await RoleAuthManager.getRoleToken('admin');
+    
+    expect(token).toBe(mockToken);
+    expect(global.fetch).toHaveBeenCalledWith(
+      expect.stringContaining('/api/auth/login'),
+      expect.objectContaining({
+        method: 'POST',
+        body: expect.stringContaining('admin')
+      })
+    );
+  });
+
+  it('should return cached token on second call', async () => {
+    const mockToken = 'cached-token';
+    (global.fetch as any).mockResolvedValueOnce({
+      ok: true,
+      json: async () => ({ data: { token: mockToken } })
+    });
+
+    const token1 = await RoleAuthManager.getRoleToken('admin');
+    const token2 = await RoleAuthManager.getRoleToken('admin');
+
+    expect(token1).toBe(token2);
+    expect(global.fetch).toHaveBeenCalledTimes(1);
+  });
+
+  it('should throw error for unknown role', async () => {
+    await expect(RoleAuthManager.getRoleToken('unknown')).rejects.toThrow("Role 'unknown' not found");
+  });
+
+  it('should throw error on authentication failure', async () => {
+    (global.fetch as any).mockResolvedValueOnce({
+      ok: false,
+      statusText: 'Unauthorized'
+    });
+
+    await expect(RoleAuthManager.getRoleToken('admin')).rejects.toThrow('Authentication failed');
+  });
+
+  it('should clear specific role token', async () => {
+    const mockToken = 'token-to-clear';
+    (global.fetch as any).mockResolvedValueOnce({
+      ok: true,
+      json: async () => ({ data: { token: mockToken } })
+    });
+
+    await RoleAuthManager.getRoleToken('admin');
+    RoleAuthManager.clearRoleToken('admin');
+    
+    // 再次获取应该重新认证
+    (global.fetch as any).mockResolvedValueOnce({
+      ok: true,
+      json: async () => ({ data: { token: 'new-token' } })
+    });
+    
+    const newToken = await RoleAuthManager.getRoleToken('admin');
+    expect(newToken).toBe('new-token');
+    expect(global.fetch).toHaveBeenCalledTimes(2);
+  });
+});