create-dp-koa 1.0.0

package/template/src/utils/testDataInitializer.ts

@@ -0,0 +1,296 @@
+import { getDataSource } from "@src/framework/utils/db";
+import { logger } from "@src/framework/utils/logger";
+import { YtUserEntity, YtUserTypeEnum, YtUserStatusEnum } from "@src/entity/ytUser.entity";
+import { ShopEntity, ShopStatusEnum } from "@src/entity/shop.entity";
+import { ShopAndUserEntity } from "@src/entity/shopUser.entity";
+import { YtGoodsEntity, YtGoodsStatusEnum } from "@src/entity/ytGoods.entity";
+import { GoodsImagesUnlockKeyEntity } from "@src/entity/goodsImagesUnlockKey.entity";
+import * as bcrypt from 'bcryptjs';
+
+/**
+ * 测试数据初始化工具
+ * 仅在内存数据库模式下使用
+ */
+export class TestDataInitializer {
+    private static instance: TestDataInitializer;
+
+    private constructor() {}
+
+    static getInstance(): TestDataInitializer {
+        if (!TestDataInitializer.instance) {
+            TestDataInitializer.instance = new TestDataInitializer();
+        }
+        return TestDataInitializer.instance;
+    }
+
+    /**
+     * 检查是否为内存数据库模式
+     */
+    private isMemoryDatabase(): boolean {
+        const dataSource = getDataSource();
+        if (!dataSource || !dataSource.isInitialized) {
+            return false;
+        }
+
+        const driver = (dataSource as any).driver;
+        return driver && driver.options && driver.options.database === ':memory:';
+    }
+
+    /**
+     * 初始化测试数据
+     */
+    async initializeTestData(): Promise<void> {
+        if (!this.isMemoryDatabase()) {
+            logger.info("非内存数据库模式，跳过测试数据初始化");
+            return;
+        }
+
+        logger.info("开始初始化内存数据库测试数据...");
+
+        try {
+            const dataSource = getDataSource();
+            if (!dataSource) {
+                throw new Error("数据库连接未初始化");
+            }
+
+            // 创建测试用户
+            const testUsers = await this.createTestUsers(dataSource);
+            logger.info(`创建了 ${testUsers.length} 个测试用户`);
+
+            // 创建测试店铺
+            const testShops = await this.createTestShops(dataSource, testUsers);
+            logger.info(`创建了 ${testShops.length} 个测试店铺`);
+
+            // 创建店铺用户关联
+            await this.createShopUserRelations(dataSource, testUsers, testShops);
+            logger.info("创建了店铺用户关联");
+
+            // 创建测试商品
+            const testGoods = await this.createTestGoods(dataSource, testShops);
+            logger.info(`创建了 ${testGoods.length} 个测试商品`);
+
+            // 创建测试解锁密钥
+            await this.createTestUnlockKeys(dataSource, testGoods);
+            logger.info("创建了测试解锁密钥");
+
+            logger.info("内存数据库测试数据初始化完成");
+
+        } catch (error) {
+            logger.error("初始化测试数据失败:", error as Error);
+            throw error;
+        }
+    }
+
+    /**
+     * 创建测试用户
+     */
+    private async createTestUsers(dataSource: any): Promise<YtUserEntity[]> {
+        // 加密密码 (使用符合强度要求的密码)
+        const hashedPassword = await bcrypt.hash('Password123!', 10);
+        
+        const users = [
+            {
+                nickName: "测试用户1",
+                email: "test1@example.com",
+                telnumber: "13800138001",
+                password: hashedPassword,
+                userType: YtUserTypeEnum.CUSTOMER,
+                status: YtUserStatusEnum.NORMAL,
+                avatar: "https://example.com/avatar1.jpg",
+                gender: 1,
+                age: 25,
+                constellation: "白羊座"
+            },
+            {
+                nickName: "测试用户2",
+                email: "test2@example.com",
+                telnumber: "13800138002",
+                password: hashedPassword,
+                userType: YtUserTypeEnum.CUSTOMER,
+                status: YtUserStatusEnum.NORMAL,
+                avatar: "https://example.com/avatar2.jpg",
+                gender: 2,
+                age: 28,
+                constellation: "金牛座"
+            },
+            {
+                nickName: "店铺管理员",
+                email: "admin@example.com",
+                telnumber: "13800138003",
+                password: hashedPassword,
+                userType: YtUserTypeEnum.SHOP,
+                status: YtUserStatusEnum.NORMAL,
+                avatar: "https://example.com/avatar3.jpg",
+                gender: 1,
+                age: 30,
+                constellation: "双子座"
+            }
+        ];
+
+        const userRepository = dataSource.getRepository(YtUserEntity);
+        const createdUsers: YtUserEntity[] = [];
+
+        for (const userData of users) {
+            const user = userRepository.create(userData);
+            const savedUser = await userRepository.save(user);
+            createdUsers.push(savedUser);
+        }
+
+        return createdUsers;
+    }
+
+    /**
+     * 创建测试店铺
+     */
+    private async createTestShops(dataSource: any, users: YtUserEntity[]): Promise<ShopEntity[]> {
+        const shops = [
+            {
+                name: "测试店铺1",
+                description: "这是一个测试店铺",
+                status: ShopStatusEnum.NORMAL
+            },
+            {
+                name: "测试店铺2",
+                description: "这是另一个测试店铺",
+                status: ShopStatusEnum.NORMAL
+            }
+        ];
+
+        const shopRepository = dataSource.getRepository(ShopEntity);
+        const createdShops: ShopEntity[] = [];
+
+        for (const shopData of shops) {
+            const shop = shopRepository.create(shopData);
+            const savedShop = await shopRepository.save(shop);
+            createdShops.push(savedShop);
+        }
+
+        return createdShops;
+    }
+
+    /**
+     * 创建店铺用户关联
+     */
+    private async createShopUserRelations(dataSource: any, users: YtUserEntity[], shops: ShopEntity[]): Promise<void> {
+        const relations = [
+            { userId: users[0].id, shopId: shops[0].id },
+            { userId: users[1].id, shopId: shops[1].id },
+            { userId: users[2].id, shopId: shops[0].id } // 管理员用户关联到第一个店铺
+        ];
+
+        const relationRepository = dataSource.getRepository(ShopAndUserEntity);
+
+        for (const relationData of relations) {
+            const relation = relationRepository.create(relationData);
+            await relationRepository.save(relation);
+        }
+    }
+
+    /**
+     * 创建测试商品
+     */
+    private async createTestGoods(dataSource: any, shops: ShopEntity[]): Promise<YtGoodsEntity[]> {
+        const goods = [
+            {
+                name: "测试商品1",
+                description: "这是一个测试商品",
+                price: 99.99,
+                albums: ["https://example.com/goods1-1.jpg", "https://example.com/goods1-2.jpg"],
+                tags: ["电子产品", "数码", "测试"],
+                content: "这是测试商品1的详细描述内容",
+                imagesContent: ["https://example.com/content1-1.jpg", "https://example.com/content1-2.jpg"],
+                shopId: shops[0].id,
+                status: YtGoodsStatusEnum.NORMAL
+            },
+            {
+                name: "测试商品2",
+                description: "这是另一个测试商品",
+                price: 199.99,
+                albums: ["https://example.com/goods2-1.jpg"],
+                tags: ["服装", "时尚", "测试"],
+                content: "这是测试商品2的详细描述内容",
+                imagesContent: ["https://example.com/content2-1.jpg"],
+                shopId: shops[1].id,
+                status: YtGoodsStatusEnum.NORMAL
+            },
+            {
+                name: "测试商品3",
+                description: "第三个测试商品",
+                price: 299.99,
+                albums: ["https://example.com/goods3-1.jpg", "https://example.com/goods3-2.jpg", "https://example.com/goods3-3.jpg"],
+                tags: ["家居", "装饰", "测试"],
+                content: "这是测试商品3的详细描述内容",
+                imagesContent: ["https://example.com/content3-1.jpg", "https://example.com/content3-2.jpg"],
+                shopId: shops[0].id,
+                status: YtGoodsStatusEnum.NORMAL
+            }
+        ];
+
+        const goodsRepository = dataSource.getRepository(YtGoodsEntity);
+        const createdGoods: YtGoodsEntity[] = [];
+
+        for (const goodsData of goods) {
+            const goods = goodsRepository.create(goodsData);
+            const savedGoods = await goodsRepository.save(goods);
+            createdGoods.push(savedGoods);
+        }
+
+        return createdGoods;
+    }
+
+    /**
+     * 创建测试解锁密钥
+     */
+    private async createTestUnlockKeys(dataSource: any, goods: YtGoodsEntity[]): Promise<void> {
+        const unlockKeys = [
+            { key: "TEST_KEY_001", goodsId: goods[0].id, isUse: false },
+            { key: "TEST_KEY_002", goodsId: goods[0].id, isUse: false },
+            { key: "TEST_KEY_003", goodsId: goods[1].id, isUse: false },
+            { key: "USED_KEY_001", goodsId: goods[2].id, isUse: true }, // 已使用的密钥
+            { key: "TEST_KEY_004", goodsId: goods[2].id, isUse: false }
+        ];
+
+        const unlockKeyRepository = dataSource.getRepository(GoodsImagesUnlockKeyEntity);
+
+        for (const keyData of unlockKeys) {
+            const unlockKey = unlockKeyRepository.create(keyData);
+            await unlockKeyRepository.save(unlockKey);
+        }
+    }
+
+    /**
+     * 清空所有测试数据
+     */
+    async clearTestData(): Promise<void> {
+        if (!this.isMemoryDatabase()) {
+            logger.info("非内存数据库模式，跳过测试数据清理");
+            return;
+        }
+
+        logger.info("开始清理内存数据库测试数据...");
+
+        try {
+            const dataSource = getDataSource();
+            if (!dataSource) {
+                throw new Error("数据库连接未初始化");
+            }
+
+            // 按依赖关系顺序删除数据
+            await dataSource.query("DELETE FROM goods_images_unlock_key");
+            await dataSource.query("DELETE FROM yt_goods_unlock");
+            await dataSource.query("DELETE FROM yt_goods");
+            await dataSource.query("DELETE FROM shop_and_user");
+            await dataSource.query("DELETE FROM shop");
+            await dataSource.query("DELETE FROM yt_user");
+
+            logger.info("内存数据库测试数据清理完成");
+
+        } catch (error) {
+            logger.error("清理测试数据失败:", error as Error);
+            throw error;
+        }
+    }
+}
+
+// 导出单例实例
+export const testDataInitializer = TestDataInitializer.getInstance();