90 lines
2.3 KiB
JavaScript
90 lines
2.3 KiB
JavaScript
|
// 测试环境配置 - 使用SQLite内存数据库进行测试
|
|||
|
|
const { Sequelize } = require('sequelize');
|
|||
|
|
|
|||
|
|
// 设置测试环境变量
|
|||
|
|
process.env.NODE_ENV = 'test';
|
|||
|
|
|
|||
|
|
// 创建测试用的SQLite内存数据库
|
|||
|
|
const testSequelize = new Sequelize('sqlite::memory:', {
|
|||
|
|
logging: false, // 关闭SQL日志
|
|||
|
|
define: {
|
|||
|
|
timestamps: true,
|
|||
|
|
underscored: false,
|
|||
|
|
freezeTableName: false
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
// 重写数据库配置模块，让测试使用SQLite
|
|||
|
|
const originalRequire = require;
|
|||
|
|
require = function(id) {
|
|||
|
|
if (id === '../src/config/database' || id.endsWith('/config/database')) {
|
|||
|
|
return testSequelize;
|
|||
|
|
}
|
|||
|
|
return originalRequire.apply(this, arguments);
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
// 全局测试设置
|
|||
|
|
beforeAll(async () => {
|
|||
|
|
try {
|
|||
|
|
// 连接测试数据库
|
|||
|
|
await testSequelize.authenticate();
|
|||
|
|
console.log('测试数据库连接成功');
|
|||
|
|
|
|||
|
|
// 导入所有模型
|
|||
|
|
const User = require('../src/models/User');
|
|||
|
|
const Order = require('../src/models/Order');
|
|||
|
|
|
|||
|
|
// 同步数据库模型（仅在测试环境）
|
|||
|
|
await testSequelize.sync({ force: true });
|
|||
|
|
console.log('测试数据库同步完成');
|
|||
|
|
} catch (error) {
|
|||
|
|
console.error('测试数据库连接失败:', error);
|
|||
|
|
// 不要直接退出进程，让Jest处理错误
|
|||
|
|
throw error;
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
// 每个测试后清理数据
|
|||
|
|
afterEach(async () => {
|
|||
|
|
try {
|
|||
|
|
// 清理所有表数据，但保留表结构
|
|||
|
|
const models = Object.keys(testSequelize.models);
|
|||
|
|
for (const modelName of models) {
|
|||
|
|
await testSequelize.models[modelName].destroy({
|
|||
|
|
where: {},
|
|||
|
|
truncate: true,
|
|||
|
|
cascade: true,
|
|||
|
|
restartIdentity: true
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
} catch (error) {
|
|||
|
|
console.error('清理测试数据失败:', error);
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
// 测试后清理
|
|||
|
|
afterAll(async () => {
|
|||
|
|
try {
|
|||
|
|
// 关闭数据库连接
|
|||
|
|
await testSequelize.close();
|
|||
|
|
console.log('测试数据库连接已关闭');
|
|||
|
|
} catch (error) {
|
|||
|
|
console.error('测试清理失败:', error);
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
// 设置全局超时
|
|||
|
|
jest.setTimeout(30000);
|
|||
|
|
|
|||
|
|
// 导出测试数据库实例供测试文件使用
|
|||
|
|
global.testSequelize = testSequelize;
|
|||
|
|
|
|||
|
|
// 抑制控制台输出（可选）
|
|||
|
|
// global.console = {
|
|||
|
|
// ...console,
|
|||
|
|
// log: jest.fn(),
|
|||
|
|
// debug: jest.fn(),
|
|||
|
|
// info: jest.fn(),
|
|||
|
|
// warn: jest.fn(),
|
|||
|
|
// error: jest.fn(),
|
|||
|
|
// };
|