61 lines
2.2 KiB
JavaScript
61 lines
2.2 KiB
JavaScript
|
const { SeedManager } = require('./src/database/seed-orders');
|
|||
|
|
const UsersGenerator = require('./src/mock-generators/users');
|
|||
|
|
const OrdersGenerator = require('./src/mock-generators/orders');
|
|||
|
|
|
|||
|
|
async function test() {
|
|||
|
|
console.log('Testing dependency resolution...');
|
|||
|
|
|
|||
|
|
const seedManager = new SeedManager();
|
|||
|
|
|
|||
|
|
// 生成 users
|
|||
|
|
const usersGen = new UsersGenerator();
|
|||
|
|
usersGen.setSeedManager(seedManager);
|
|||
|
|
const usersData = usersGen.generate(10);
|
|||
|
|
|
|||
|
|
// 注册生成的 users 数据
|
|||
|
|
seedManager.registerGeneratedData('users', usersData);
|
|||
|
|
|
|||
|
|
// 模拟插入 users（分配数据库 ID）
|
|||
|
|
const insertedUsersData = usersData.map((record, index) => ({
|
|||
|
|
...record,
|
|||
|
|
id: index + 1000 // 模拟数据库 ID
|
|||
|
|
}));
|
|||
|
|
|
|||
|
|
// 注册插入后的 users 数据
|
|||
|
|
seedManager.registerInsertedData('users', insertedUsersData);
|
|||
|
|
|
|||
|
|
// 注册 ID 映射
|
|||
|
|
usersData.forEach((record, index) => {
|
|||
|
|
if (record.temp_id && insertedUsersData[index].id) {
|
|||
|
|
seedManager.registerIdMapping('users', record.temp_id, insertedUsersData[index].id);
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
console.log('\nUsers generated and inserted.');
|
|||
|
|
console.log('Users data sample:', JSON.stringify(insertedUsersData[0], null, 2));
|
|||
|
|
|
|||
|
|
// 生成 orders
|
|||
|
|
const ordersGen = new OrdersGenerator();
|
|||
|
|
ordersGen.setSeedManager(seedManager);
|
|||
|
|
const ordersData = ordersGen.generate(5);
|
|||
|
|
|
|||
|
|
console.log('\nOrders generated:');
|
|||
|
|
ordersData.forEach((order, index) => {
|
|||
|
|
console.log(`Order ${index + 1}: user_id = ${order.user_id}, shipping_addr_id = ${order.shipping_addr_id}, billing_addr_id = ${order.billing_addr_id}`);
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
// 检查是否能够获取随机用户 ID
|
|||
|
|
const randomUserId = ordersGen.getRandomIdFromTable('users');
|
|||
|
|
console.log(`\nRandom user ID from getRandomIdFromTable: ${randomUserId}`);
|
|||
|
|
|
|||
|
|
// 解析依赖
|
|||
|
|
const ordersConfig = seedManager.getSeedConfig('orders');
|
|||
|
|
const resolvedOrders = seedManager.resolveDependencies(ordersData, ordersConfig.dependencies);
|
|||
|
|
|
|||
|
|
console.log('\nResolved orders:');
|
|||
|
|
resolvedOrders.forEach((order, index) => {
|
|||
|
|
console.log(`Order ${index + 1}: user_id = ${order.user_id}, shipping_addr_id = ${order.shipping_addr_id}, billing_addr_id = ${order.billing_addr_id}`);
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
test().catch(console.error);
|