138 lines
3.6 KiB
TypeScript
138 lines
3.6 KiB
TypeScript
import { describe, it, expect, vi } from 'vitest'
|
|
import { successResponse, errorResponse } from '../response'
|
|
import type { Response } from 'express'
|
|
|
|
vi.mock('express', () => ({
|
|
default: {},
|
|
}))
|
|
|
|
describe('successResponse', () => {
|
|
it('应返回正确格式的成功响应（默认状态码）', () => {
|
|
const mockRes = {
|
|
status: vi.fn().mockReturnThis(),
|
|
json: vi.fn().mockReturnThis(),
|
|
} as unknown as Response
|
|
|
|
const data = { message: '操作成功' }
|
|
successResponse(mockRes, data)
|
|
|
|
expect(mockRes.status).toHaveBeenCalledWith(200)
|
|
expect(mockRes.json).toHaveBeenCalledWith({
|
|
success: true,
|
|
data: { message: '操作成功' },
|
|
})
|
|
})
|
|
|
|
it('应使用自定义状态码', () => {
|
|
const mockRes = {
|
|
status: vi.fn().mockReturnThis(),
|
|
json: vi.fn().mockReturnThis(),
|
|
} as unknown as Response
|
|
|
|
const data = { id: 123 }
|
|
successResponse(mockRes, data, 201)
|
|
|
|
expect(mockRes.status).toHaveBeenCalledWith(201)
|
|
expect(mockRes.json).toHaveBeenCalledWith({
|
|
success: true,
|
|
data: { id: 123 },
|
|
})
|
|
})
|
|
|
|
it('应正确处理数组数据', () => {
|
|
const mockRes = {
|
|
status: vi.fn().mockReturnThis(),
|
|
json: vi.fn().mockReturnThis(),
|
|
} as unknown as Response
|
|
|
|
const data = [1, 2, 3]
|
|
successResponse(mockRes, data)
|
|
|
|
expect(mockRes.json).toHaveBeenCalledWith({
|
|
success: true,
|
|
data: [1, 2, 3],
|
|
})
|
|
})
|
|
})
|
|
|
|
describe('errorResponse', () => {
|
|
it('应返回正确格式的错误响应', () => {
|
|
const mockRes = {
|
|
status: vi.fn().mockReturnThis(),
|
|
json: vi.fn().mockReturnThis(),
|
|
} as unknown as Response
|
|
|
|
errorResponse(mockRes, 400, 'BAD_REQUEST', '请求参数错误')
|
|
|
|
expect(mockRes.status).toHaveBeenCalledWith(400)
|
|
expect(mockRes.json).toHaveBeenCalledWith({
|
|
success: false,
|
|
error: {
|
|
code: 'BAD_REQUEST',
|
|
message: '请求参数错误',
|
|
},
|
|
})
|
|
})
|
|
|
|
it('应在非生产环境包含 details', () => {
|
|
const mockRes = {
|
|
status: vi.fn().mockReturnThis(),
|
|
json: vi.fn().mockReturnThis(),
|
|
} as unknown as Response
|
|
|
|
const details = { field: 'username', reason: 'required' }
|
|
errorResponse(mockRes, 422, 'VALIDATION_ERROR', '验证失败', details)
|
|
|
|
expect(mockRes.json).toHaveBeenCalledWith({
|
|
success: false,
|
|
error: {
|
|
code: 'VALIDATION_ERROR',
|
|
message: '验证失败',
|
|
details: { field: 'username', reason: 'required' },
|
|
},
|
|
})
|
|
})
|
|
|
|
it('应在生产环境不包含 details', () => {
|
|
const originalEnv = process.env.NODE_ENV
|
|
process.env.NODE_ENV = 'production'
|
|
|
|
const mockRes = {
|
|
status: vi.fn().mockReturnThis(),
|
|
json: vi.fn().mockReturnThis(),
|
|
} as unknown as Response
|
|
|
|
const details = { field: 'username', reason: 'required' }
|
|
errorResponse(mockRes, 422, 'VALIDATION_ERROR', '验证失败', details)
|
|
|
|
expect(mockRes.json).toHaveBeenCalledWith({
|
|
success: false,
|
|
error: {
|
|
code: 'VALIDATION_ERROR',
|
|
message: '验证失败',
|
|
details: undefined,
|
|
},
|
|
})
|
|
|
|
process.env.NODE_ENV = originalEnv
|
|
})
|
|
|
|
it('应正确处理不带 details 的错误响应', () => {
|
|
const mockRes = {
|
|
status: vi.fn().mockReturnThis(),
|
|
json: vi.fn().mockReturnThis(),
|
|
} as unknown as Response
|
|
|
|
errorResponse(mockRes, 404, 'NOT_FOUND', '资源不存在')
|
|
|
|
expect(mockRes.status).toHaveBeenCalledWith(404)
|
|
expect(mockRes.json).toHaveBeenCalledWith({
|
|
success: false,
|
|
error: {
|
|
code: 'NOT_FOUND',
|
|
message: '资源不存在',
|
|
},
|
|
})
|
|
})
|
|
})
|