354 lines
10 KiB
JavaScript
354 lines
10 KiB
JavaScript
|
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest'
|
|||
|
|
import { api, clearCache, cancelAllRequests } from '@/services/api'
|
|||
|
|
import axios from 'axios'
|
|||
|
|
|
|||
|
|
// 模拟axios
|
|||
|
|
vi.mock('axios')
|
|||
|
|
const mockedAxios = vi.mocked(axios)
|
|||
|
|
|
|||
|
|
// 模拟localStorage
|
|||
|
|
const localStorageMock = {
|
|||
|
|
getItem: vi.fn(),
|
|||
|
|
setItem: vi.fn(),
|
|||
|
|
removeItem: vi.fn(),
|
|||
|
|
clear: vi.fn()
|
|||
|
|
}
|
|||
|
|
global.localStorage = localStorageMock
|
|||
|
|
|
|||
|
|
// 模拟console方法
|
|||
|
|
const consoleSpy = {
|
|||
|
|
error: vi.spyOn(console, 'error').mockImplementation(() => {}),
|
|||
|
|
warn: vi.spyOn(console, 'warn').mockImplementation(() => {}),
|
|||
|
|
log: vi.spyOn(console, 'log').mockImplementation(() => {})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 模拟window.performance
|
|||
|
|
Object.defineProperty(window, 'performance', {
|
|||
|
|
value: {
|
|||
|
|
now: vi.fn(() => Date.now())
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
describe('API Service', () => {
|
|||
|
|
beforeEach(() => {
|
|||
|
|
// 重置所有模拟
|
|||
|
|
vi.clearAllMocks()
|
|||
|
|
|
|||
|
|
// 模拟axios.create返回值
|
|||
|
|
mockedAxios.create.mockReturnValue({
|
|||
|
|
request: vi.fn(),
|
|||
|
|
interceptors: {
|
|||
|
|
request: {
|
|||
|
|
use: vi.fn()
|
|||
|
|
},
|
|||
|
|
response: {
|
|||
|
|
use: vi.fn()
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
afterEach(() => {
|
|||
|
|
// 清理所有请求
|
|||
|
|
cancelAllRequests()
|
|||
|
|
clearCache()
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
describe('API实例', () => {
|
|||
|
|
it('应该正确创建API实例', () => {
|
|||
|
|
expect(api).toBeDefined()
|
|||
|
|
expect(typeof api.get).toBe('function')
|
|||
|
|
expect(typeof api.post).toBe('function')
|
|||
|
|
expect(typeof api.put).toBe('function')
|
|||
|
|
expect(typeof api.delete).toBe('function')
|
|||
|
|
expect(typeof api.request).toBe('function')
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
it('应该设置正确的baseURL', () => {
|
|||
|
|
// 由于我们模拟了axios.create，这里我们检查调用参数
|
|||
|
|
expect(mockedAxios.create).toHaveBeenCalledWith(
|
|||
|
|
expect.objectContaining({
|
|||
|
|
baseURL: 'https://localhost:7001/api',
|
|||
|
|
timeout: 15000
|
|||
|
|
})
|
|||
|
|
)
|
|||
|
|
})
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
describe('请求拦截器', () => {
|
|||
|
|
it('应该在请求前添加时间戳', async () => {
|
|||
|
|
// 模拟请求
|
|||
|
|
const mockRequest = {
|
|||
|
|
url: '/test',
|
|||
|
|
method: 'get',
|
|||
|
|
headers: {}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 模拟请求拦截器
|
|||
|
|
const requestInterceptor = mockedAxios.create().interceptors.request.use.mock.calls[0][0]
|
|||
|
|
const processedRequest = requestInterceptor(mockRequest)
|
|||
|
|
|
|||
|
|
// 检查是否添加了时间戳
|
|||
|
|
expect(processedRequest.metadata).toBeDefined()
|
|||
|
|
expect(processedRequest.metadata.startTime).toBeDefined()
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
it('应该在请求前添加CSRF令牌', async () => {
|
|||
|
|
// 模拟localStorage中的CSRF令牌
|
|||
|
|
localStorageMock.getItem.mockReturnValue('test-csrf-token')
|
|||
|
|
|
|||
|
|
// 模拟请求
|
|||
|
|
const mockRequest = {
|
|||
|
|
url: '/test',
|
|||
|
|
method: 'post',
|
|||
|
|
headers: {}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 模拟请求拦截器
|
|||
|
|
const requestInterceptor = mockedAxios.create().interceptors.request.use.mock.calls[0][0]
|
|||
|
|
const processedRequest = requestInterceptor(mockRequest)
|
|||
|
|
|
|||
|
|
// 检查是否添加了CSRF令牌
|
|||
|
|
expect(processedRequest.headers['X-CSRF-Token']).toBe('test-csrf-token')
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
it('应该在请求前添加认证令牌', async () => {
|
|||
|
|
// 模拟localStorage中的认证令牌
|
|||
|
|
localStorageMock.getItem.mockImplementation((key) => {
|
|||
|
|
if (key === 'auth_token') return 'test-auth-token'
|
|||
|
|
return null
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
// 模拟请求
|
|||
|
|
const mockRequest = {
|
|||
|
|
url: '/test',
|
|||
|
|
method: 'get',
|
|||
|
|
headers: {}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 模拟请求拦截器
|
|||
|
|
const requestInterceptor = mockedAxios.create().interceptors.request.use.mock.calls[0][0]
|
|||
|
|
const processedRequest = requestInterceptor(mockRequest)
|
|||
|
|
|
|||
|
|
// 检查是否添加了认证令牌
|
|||
|
|
expect(processedRequest.headers.Authorization).toBe('Bearer test-auth-token')
|
|||
|
|
})
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
describe('响应拦截器', () => {
|
|||
|
|
it('应该正确处理成功响应', async () => {
|
|||
|
|
// 模拟响应
|
|||
|
|
const mockResponse = {
|
|||
|
|
data: { success: true, data: { id: 1, name: 'Test' } },
|
|||
|
|
status: 200,
|
|||
|
|
statusText: 'OK',
|
|||
|
|
headers: {},
|
|||
|
|
config: {
|
|||
|
|
url: '/test',
|
|||
|
|
method: 'get',
|
|||
|
|
metadata: { startTime: Date.now() - 100 }
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 模拟响应拦截器
|
|||
|
|
const responseInterceptor = mockedAxios.create().interceptors.response.use.mock.calls[0][0]
|
|||
|
|
const processedResponse = responseInterceptor(mockResponse)
|
|||
|
|
|
|||
|
|
// 检查响应处理
|
|||
|
|
expect(processedResponse.data).toEqual({ success: true, data: { id: 1, name: 'Test' } })
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
it('应该记录请求耗时', async () => {
|
|||
|
|
// 模拟响应
|
|||
|
|
const mockResponse = {
|
|||
|
|
data: { success: true, data: { id: 1, name: 'Test' } },
|
|||
|
|
status: 200,
|
|||
|
|
statusText: 'OK',
|
|||
|
|
headers: {},
|
|||
|
|
config: {
|
|||
|
|
url: '/test',
|
|||
|
|
method: 'get',
|
|||
|
|
metadata: { startTime: Date.now() - 100 }
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 模拟响应拦截器
|
|||
|
|
const responseInterceptor = mockedAxios.create().interceptors.response.use.mock.calls[0][0]
|
|||
|
|
responseInterceptor(mockResponse)
|
|||
|
|
|
|||
|
|
// 检查是否记录了请求耗时
|
|||
|
|
expect(consoleSpy.log).toHaveBeenCalledWith(
|
|||
|
|
expect.stringContaining('请求完成'),
|
|||
|
|
expect.anything()
|
|||
|
|
)
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
it('应该缓存GET请求响应', async () => {
|
|||
|
|
// 模拟响应
|
|||
|
|
const mockResponse = {
|
|||
|
|
data: { success: true, data: { id: 1, name: 'Test' } },
|
|||
|
|
status: 200,
|
|||
|
|
statusText: 'OK',
|
|||
|
|
headers: {},
|
|||
|
|
config: {
|
|||
|
|
url: '/test',
|
|||
|
|
method: 'get',
|
|||
|
|
metadata: { startTime: Date.now() - 100 }
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 模拟响应拦截器
|
|||
|
|
const responseInterceptor = mockedAxios.create().interceptors.response.use.mock.calls[0][0]
|
|||
|
|
responseInterceptor(mockResponse)
|
|||
|
|
|
|||
|
|
// 检查是否缓存了响应
|
|||
|
|
expect(localStorageMock.setItem).toHaveBeenCalledWith(
|
|||
|
|
expect.stringContaining('api_cache_'),
|
|||
|
|
expect.any(String)
|
|||
|
|
)
|
|||
|
|
})
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
describe('错误处理', () => {
|
|||
|
|
it('应该正确处理网络错误', async () => {
|
|||
|
|
// 模拟网络错误
|
|||
|
|
const networkError = new Error('Network Error')
|
|||
|
|
networkError.code = 'NETWORK_ERROR'
|
|||
|
|
|
|||
|
|
// 模拟错误拦截器
|
|||
|
|
const errorInterceptor = mockedAxios.create().interceptors.response.use.mock.calls[0][1]
|
|||
|
|
|
|||
|
|
// 处理错误
|
|||
|
|
const result = errorInterceptor(networkError)
|
|||
|
|
|
|||
|
|
// 检查错误处理
|
|||
|
|
expect(result).rejects.toThrow()
|
|||
|
|
expect(consoleSpy.error).toHaveBeenCalledWith(
|
|||
|
|
'网络错误',
|
|||
|
|
expect.any(Error)
|
|||
|
|
)
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
it('应该正确处理超时错误', async () => {
|
|||
|
|
// 模拟超时错误
|
|||
|
|
const timeoutError = new Error('Timeout')
|
|||
|
|
timeoutError.code = 'ECONNABORTED'
|
|||
|
|
|
|||
|
|
// 模拟错误拦截器
|
|||
|
|
const errorInterceptor = mockedAxios.create().interceptors.response.use.mock.calls[0][1]
|
|||
|
|
|
|||
|
|
// 处理错误
|
|||
|
|
const result = errorInterceptor(timeoutError)
|
|||
|
|
|
|||
|
|
// 检查错误处理
|
|||
|
|
expect(result).rejects.toThrow()
|
|||
|
|
expect(consoleSpy.error).toHaveBeenCalledWith(
|
|||
|
|
'请求超时',
|
|||
|
|
expect.any(Error)
|
|||
|
|
)
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
it('应该正确处理服务器错误', async () => {
|
|||
|
|
// 模拟服务器错误响应
|
|||
|
|
const serverError = {
|
|||
|
|
response: {
|
|||
|
|
status: 500,
|
|||
|
|
statusText: 'Internal Server Error',
|
|||
|
|
data: { message: '服务器内部错误' }
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 模拟错误拦截器
|
|||
|
|
const errorInterceptor = mockedAxios.create().interceptors.response.use.mock.calls[0][1]
|
|||
|
|
|
|||
|
|
// 处理错误
|
|||
|
|
const result = errorInterceptor(serverError)
|
|||
|
|
|
|||
|
|
// 检查错误处理
|
|||
|
|
expect(result).rejects.toThrow()
|
|||
|
|
expect(consoleSpy.error).toHaveBeenCalledWith(
|
|||
|
|
'服务器错误',
|
|||
|
|
expect.any(Object)
|
|||
|
|
)
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
it('应该正确处理401未授权错误', async () => {
|
|||
|
|
// 模拟401错误响应
|
|||
|
|
const unauthorizedError = {
|
|||
|
|
response: {
|
|||
|
|
status: 401,
|
|||
|
|
statusText: 'Unauthorized',
|
|||
|
|
data: { message: '未授权访问' }
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 模拟错误拦截器
|
|||
|
|
const errorInterceptor = mockedAxios.create().interceptors.response.use.mock.calls[0][1]
|
|||
|
|
|
|||
|
|
// 处理错误
|
|||
|
|
const result = errorInterceptor(unauthorizedError)
|
|||
|
|
|
|||
|
|
// 检查错误处理
|
|||
|
|
expect(result).rejects.toThrow()
|
|||
|
|
expect(consoleSpy.error).toHaveBeenCalledWith(
|
|||
|
|
'未授权访问',
|
|||
|
|
expect.any(Object)
|
|||
|
|
)
|
|||
|
|
})
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
describe('缓存管理', () => {
|
|||
|
|
it('应该正确缓存GET请求', () => {
|
|||
|
|
// 模拟缓存数据
|
|||
|
|
const cacheKey = 'api_cache_/test'
|
|||
|
|
const cacheData = {
|
|||
|
|
data: { id: 1, name: 'Test' },
|
|||
|
|
timestamp: Date.now(),
|
|||
|
|
expiry: Date.now() + 300000 // 5分钟后过期
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 设置缓存
|
|||
|
|
localStorageMock.setItem.mockReturnValue()
|
|||
|
|
localStorageMock.getItem.mockReturnValue(JSON.stringify(cacheData))
|
|||
|
|
|
|||
|
|
// 检查缓存设置
|
|||
|
|
localStorageMock.setItem(cacheKey, JSON.stringify(cacheData))
|
|||
|
|
expect(localStorageMock.setItem).toHaveBeenCalledWith(cacheKey, JSON.stringify(cacheData))
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
it('应该正确清除缓存', () => {
|
|||
|
|
// 调用清除缓存函数
|
|||
|
|
clearCache()
|
|||
|
|
|
|||
|
|
// 检查是否清除了所有缓存
|
|||
|
|
expect(localStorageMock.removeItem).toHaveBeenCalledWith('api_cache_/test')
|
|||
|
|
})
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
describe('请求取消', () => {
|
|||
|
|
it('应该正确取消重复请求', () => {
|
|||
|
|
// 模拟请求配置
|
|||
|
|
const requestConfig = {
|
|||
|
|
url: '/test',
|
|||
|
|
method: 'get'
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 模拟CancelToken
|
|||
|
|
const cancelToken = {
|
|||
|
|
token: 'test-token',
|
|||
|
|
cancel: vi.fn()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 检查是否创建了CancelToken
|
|||
|
|
expect(cancelToken.token).toBe('test-token')
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
it('应该正确取消所有请求', () => {
|
|||
|
|
// 调用取消所有请求函数
|
|||
|
|
cancelAllRequests()
|
|||
|
|
|
|||
|
|
// 检查是否取消了所有请求
|
|||
|
|
// 这里需要根据实际实现进行检查
|
|||
|
|
})
|
|||
|
|
})
|
|||
|
|
})
|