Code-MCP

import { describe, it, expect, beforeEach, afterEach, vi } from "vitest"; import request from "supertest"; // ============================================ // Unit Test Examples // ============================================ // Math Utilities Tests describe("Math Utilities", () => { describe("add", () => { it("adds two positive numbers", () => { expect(add(2, 3)).toBe(5); }); it("adds negative numbers", () => { expect(add(-1, -1)).toBe(-2); }); it("handles zero", () => { expect(add(0, 5)).toBe(5); }); }); describe("divide", () => { it("divides two numbers", () => { expect(divide(10, 2)).toBe(5); }); it("throws error on division by zero", () => { expect(() => divide(10, 0)).toThrow("Cannot divide by zero"); }); }); }); // Async Function Tests describe("Async Operations", () => { it("fetches data successfully", async () => { const result = await fetchData("123"); expect(result).toHaveProperty("id", "123"); }); it("handles errors gracefully", async () => { await expect(fetchData("invalid")).rejects.toThrow("Not found"); }); }); // ============================================ // Mock Examples // ============================================ describe("User Service with Mocks", () => { const mockUserRepo = { findById: vi.fn(), create: vi.fn(), update: vi.fn(), delete: vi.fn(), }; beforeEach(() => { vi.clearAllMocks(); }); it("finds user by id", async () => { const mockUser = { id: 1, name: "John" }; mockUserRepo.findById.mockResolvedValue(mockUser); const userService = new UserService(mockUserRepo); const result = await userService.getUser(1); expect(mockUserRepo.findById).toHaveBeenCalledWith(1); expect(result).toEqual(mockUser); }); it("creates user with hashed password", async () => { const input = { email: "test@example.com", password: "secret123" }; mockUserRepo.create.mockResolvedValue({ id: 1, email: input.email }); const userService = new UserService(mockUserRepo); const result = await userService.createUser(input); expect(mockUserRepo.create).toHaveBeenCalled(); expect(result.email).toBe(input.email); expect(mockUserRepo.create.mock.calls[0][0].password).not.toBe( input.password, ); }); }); // ============================================ // Integration Test Examples // ============================================ describe("API Integration Tests", () => { let app: Express; beforeEach(async () => { app = createApp(); await setupTestDatabase(); }); afterEach(async () => { await cleanupTestDatabase(); }); describe("POST /api/users", () => { it("creates a new user", async () => { const response = await request(app) .post("/api/users") .send({ email: "new@example.com", name: "New User", password: "password123", }) .expect(201); expect(response.body.success).toBe(true); expect(response.body.data).toHaveProperty("id"); expect(response.body.data.email).toBe("new@example.com"); }); it("rejects duplicate email", async () => { // Create first user await request(app) .post("/api/users") .send({ email: "test@example.com", name: "Test", password: "pass" }); // Try to create duplicate const response = await request(app) .post("/api/users") .send({ email: "test@example.com", name: "Test2", password: "pass" }) .expect(400); expect(response.body.error).toContain("already exists"); }); it("validates required fields", async () => { const response = await request(app) .post("/api/users") .send({ email: "test@example.com" }) .expect(400); expect(response.body.error).toBeDefined(); }); }); describe("GET /api/users/:id", () => { it("returns user by id", async () => { const createRes = await request(app) .post("/api/users") .send({ email: "test@example.com", name: "Test", password: "pass" }); const response = await request(app) .get(`/api/users/${createRes.body.data.id}`) .expect(200); expect(response.body.data.email).toBe("test@example.com"); }); it("returns 404 for non-existent user", async () => { await request(app).get("/api/users/99999").expect(404); }); }); }); // ============================================ // E2E Test Examples // ============================================ describe("Authentication Flow E2E", () => { let app: Express; let authToken: string; beforeAll(async () => { app = createApp(); await setupTestDatabase(); }); afterAll(async () => { await cleanupTestDatabase(); }); it("registers a new user", async () => { const response = await request(app) .post("/api/auth/register") .send({ email: "e2e@example.com", name: "E2E User", password: "securePassword123", }) .expect(201); expect(response.body.user.email).toBe("e2e@example.com"); }); it("logs in with credentials", async () => { const response = await request(app) .post("/api/auth/login") .send({ email: "e2e@example.com", password: "securePassword123", }) .expect(200); expect(response.body.token).toBeDefined(); authToken = response.body.token; }); it("accesses protected route with token", async () => { const response = await request(app) .get("/api/profile") .set("Authorization", `Bearer ${authToken}`) .expect(200); expect(response.body.email).toBe("e2e@example.com"); }); it("rejects access without token", async () => { await request(app).get("/api/profile").expect(401); }); }); // ============================================ // Test Utilities // ============================================ export function createTestUser(overrides = {}) { return { email: `test-${Date.now()}@example.com`, name: "Test User", password: "testPassword123", ...overrides, }; } export async function authenticateAs( app: Express, role: "user" | "admin" = "user", ) { const user = createTestUser({ role }); await request(app).post("/api/auth/register").send(user); const response = await request(app) .post("/api/auth/login") .send({ email: user.email, password: user.password }); return response.body.token; } // Stubs for examples (would be real implementations) function add(a: number, b: number) { return a + b; } function divide(a: number, b: number) { if (b === 0) throw new Error("Cannot divide by zero"); return a / b; } async function fetchData(id: string) { if (id === "invalid") throw new Error("Not found"); return { id }; } class UserService { constructor(private repo: any) {} async getUser(id: number) { return this.repo.findById(id); } async createUser(data: any) { return this.repo.create({ ...data, password: "hashed" }); } } function createApp(): any { return {}; } async function setupTestDatabase() {} async function cleanupTestDatabase() {} type Express = any;