add admin user mock and integration test for admin override authority

Author: Antigravity Assistant

server/test/integration/socket-admin-override.test.ts

@@ -0,0 +1,111 @@
+import { describe, it, expect, beforeAll, afterAll, beforeEach, vi } from 'vitest';
+import { Server as HTTPServer } from 'http';
+import { Server } from 'socket.io';
+import express from 'express';
+import { createServer } from 'http';
+import { CONFIG } from '../../src/config.js';
+import { LobbyController } from '../../src/controllers/lobby.controller.js';
+import { requireLobbyOwner } from '../../src/middlewares/permissionMiddleware.js';
+import { DISCONNECT_REASONS } from '../../src/constants/disconnect.constants.js';
+import {
+  createAndJoinClient,
+  mockLobbyId,
+  tokenA,
+  userA,
+  tokenAdmin,
+  adminUser,
+  teardownTestServer,
+  setupTestMocks,
+  setupTestLobby
+} from './utils/socket-test-utils.js';
+import { setupSocket } from '../../src/sockets/index.js';
+
+vi.mock('../../src/models/Lobby.js', () => ({
+  Lobby: {
+    findById: vi.fn().mockImplementation(async (id) => {
+      if (id === mockLobbyId) return { _id: mockLobbyId, owner: userA.id };
+      return null;
+    })
+  }
+}));
+
+describe('Admin Override Authority Integration', () => {
+  let io: Server;
+  let httpServer: HTTPServer;
+  let port: number;
+
+  beforeAll(async () => {
+    setupTestMocks();
+
+    const app = express();
+    app.use(express.json());
+
+    httpServer = createServer(app);
+    io = new Server(httpServer);
+    setupSocket(io);
+
+    app.use((req, res, next) => {
+      (req as any).io = io;
+      (req as any).user = adminUser;
+      next();
+    });
+
+    app.post('/api/lobbies/:id/kick', requireLobbyOwner, LobbyController.kickUser);
+
+    await new Promise<void>((resolve, reject) => {
+      httpServer.listen(() => {
+        const address = httpServer.address();
+        if (!address || typeof address === 'string') {
+          return reject(new Error('Failed to gracefully acquire a port number for the test server.'));
+        }
+        port = address.port;
+        resolve();
+      });
+    });
+  });
+
+  beforeEach(() => {
+    setupTestLobby();
+  });
+
+  afterAll(async () => {
+    await teardownTestServer(io, httpServer);
+  });
+
+  it('should allow admin to clear canvas and kick the original owner', async () => {
+    const ownerClient = await createAndJoinClient(port, tokenA);
+    const adminClient = await createAndJoinClient(port, tokenAdmin);
+
+    await new Promise(resolve => setTimeout(resolve, 50));
+
+    const clearCanvasPromise = new Promise<void>((resolve) => {
+      ownerClient.on(CONFIG.EVENTS.SERVER.CANVAS_CLEARED, () => {
+        resolve();
+      });
+    });
+
+    adminClient.emit(CONFIG.EVENTS.CLIENT.CLEAR_CANVAS, mockLobbyId);
+
+    await clearCanvasPromise;
+
+
+    const forceDisconnectPromise = new Promise<void>((resolve) => {
+      ownerClient.on(CONFIG.EVENTS.SERVER.FORCE_DISCONNECT, (data) => {
+        expect(data.reason).toBe(DISCONNECT_REASONS.KICKED);
+        resolve();
+      });
+    });
+
+    const response = await fetch(`http://localhost:${port}/api/lobbies/${mockLobbyId}/kick`, {
+      method: 'POST',
+      headers: { 'Content-Type': 'application/json' },
+      body: JSON.stringify({ targetUserId: userA.id })
+    });
+
+    expect(response.status).toBe(200);
+
+    await forceDisconnectPromise;
+    ownerClient.close();
+    adminClient.close();
+  });
+});

server/test/integration/utils/socket-test-utils.ts

@@ -12,8 +12,10 @@ import { canvasStore } from '../../../src/store/canvas.store.js';
 export const mockLobbyId = '507f1f77bcf86cd799439011';
 export const userA = { id: '507f1f77bcf86cd799439012', username: 'Alice' };
 export const userB = { id: '507f1f77bcf86cd799439013', username: 'Bob' };
+export const adminUser = { id: '507f1f77bcf86cd799439014', username: 'Admin', isAdmin: true };
 export const tokenA = 'token-a';
 export const tokenB = 'token-b';
+export const tokenAdmin = 'token-admin';
 
 export const setupTestLobby = (lobbyId = mockLobbyId, width = 100, height = 100) => {
   canvasStore.loadLobbyToMemory(lobbyId, width, height, ['#000000', '#ffffff', '#ff0000', '#00ff00', '#0000ff'], new Uint8Array(width * height).fill(0));
@@ -28,6 +30,7 @@ export const setupTestMocks = () => {
   ) => {
     if (token === tokenA) callback(null, userA);
     else if (token === tokenB) callback(null, userB);
+    else if (token === tokenAdmin) callback(null, adminUser);
     else callback(new JsonWebTokenError('Invalid token'), undefined);
   }) as typeof jwt.verify);