vendure/modules/testing/mock-data.service.ts

import * as faker from "faker/locale/en_GB";
import { Connection, createConnection } from "typeorm";
import { CustomerEntity } from "../core/entity/customer/customer.entity";
import { ProductVariantEntity } from "../core/entity/product-variant/product-variant.entity";
import { ProductEntity } from "../core/entity/product/product.entity";
import { ProductVariantTranslationEntity } from "../core/entity/product-variant/product-variant-translation.entity";
import { AddressEntity } from "../core/entity/address/address.entity";
import { Role } from "../core/auth/roles";
import { ProductTranslationEntity } from "../core/entity/product/product-translation.entity";
import { PasswordService } from "../core/auth/password.service";
import { UserEntity } from "../core/entity/user/user.entity";
import { AdministratorEntity } from "../core/entity/administrator/administrator.entity";

/**
 * A Class used for generating mock data.
 */
export class MockDataService {
    connection: Connection;

    populate(): Promise<any> {
        return createConnection({
            type: 'mysql',
            entities: ['./**/entity/**/*.entity.ts'],
            synchronize: true,
            logging: false,
            host: '192.168.99.100',
            port: 3306,
            username: 'root',
            password: '',
            database: 'test',
        }).then(async connection => {
            this.connection = connection;

            await this.clearAllTables();
            await this.populateCustomersAndAddresses();
            await this.populateProducts();
            await this.populateAdministrators();
        });
    }

    async clearAllTables() {
        await this.connection.synchronize(true);
        console.log('Cleared all tables');
    }

    async populateProducts() {
        for (let i = 0; i < 5; i++) {
            const product = new ProductEntity();
            product.image = faker.image.imageUrl();

            const name = faker.commerce.productName();
            const slug = name.toLowerCase().replace(/\s+/g, '-');
            const description = faker.lorem.sentence();

            const translation1 = this.makeProductTranslation('en', name, slug, description);
            const translation2 = this.makeProductTranslation('de', name, slug, description);
            await this.connection.manager.save(translation1);
            await this.connection.manager.save(translation2);

            // 1 - 4 variants
            const variantCount = Math.floor(Math.random() * 4) + 1;
            let variants = [];
            for (let j = 0; j < variantCount; j++) {
                const variant = new ProductVariantEntity();
                const variantName = `${name} variant ${j + 1}`;
                variant.image = faker.image.imageUrl();
                variant.price = faker.commerce.price(100, 12000, 0);

                const variantTranslation1 = this.makeProductVariantTranslation('en', variantName);
                const variantTranslation2 = this.makeProductVariantTranslation('de', variantName);
                await this.connection.manager.save(variantTranslation1);
                await this.connection.manager.save(variantTranslation2);

                variant.translations = [variantTranslation1, variantTranslation2];
                await this.connection.manager.save(variant);
                console.log(`${j + 1}. created product variant ${variantName}`);
                variants.push(variant);
            }

            product.variants = variants;
            product.translations = [translation1, translation2];
            await this.connection.manager.save(product);
            console.log(`${i + 1}. created product & translations for ${translation1.name}`);
        }
    }

    async populateCustomersAndAddresses() {
        const passwordService = new PasswordService();

        for (let i = 0; i < 5; i++) {
            const customer = new CustomerEntity();
            customer.firstName = faker.name.firstName();
            customer.lastName = faker.name.lastName();
            customer.emailAddress = faker.internet.email(customer.firstName, customer.lastName);
            customer.phoneNumber = faker.phone.phoneNumber();

            const user = new UserEntity();
            user.passwordHash = await passwordService.hash('test');
            user.identifier = customer.emailAddress;
            user.roles = [Role.Customer];

            await this.connection.manager.save(user);

            const address = new AddressEntity();
            address.fullName = `${customer.firstName} ${customer.lastName}`;
            address.streetLine1 = faker.address.streetAddress();
            address.city = faker.address.city();
            address.province = faker.address.county();
            address.postalCode = faker.address.zipCode();
            address.country = faker.address.countryCode();

            await this.connection.manager.save(address);

            customer.addresses = [address];
            customer.user = user;
            await this.connection.manager.save(customer);
            console.log('created customer, user and address for ' + customer.firstName + ' ' + customer.lastName);
        }
    }

    async populateAdministrators() {
        const passwordService = new PasswordService();

        const user = new UserEntity();
        user.passwordHash = await passwordService.hash('admin');
        user.identifier = 'admin';
        user.roles = [Role.Superadmin];

        await this.connection.manager.save(user);

        const administrator = new AdministratorEntity();
        administrator.emailAddress = 'admin@test.com';
        administrator.firstName = 'Super';
        administrator.lastName = 'Admin';
        administrator.user = user;

        await this.connection.manager.save(administrator);
    }

    private makeProductTranslation(
        langCode: string,
        name: string,
        slug: string,
        description: string,
    ): ProductTranslationEntity {
        const productTranslation = new ProductTranslationEntity();
        productTranslation.languageCode = langCode;
        productTranslation.name = `${langCode} ${name}`;
        productTranslation.slug = `${langCode} ${slug}`;
        productTranslation.description = `${langCode} ${description}`;
        return productTranslation;
    }

    private makeProductVariantTranslation(langCode: string, name: string): ProductVariantTranslationEntity {
        const productVariantTranslation = new ProductVariantTranslationEntity();
        productVariantTranslation.languageCode = langCode;
        productVariantTranslation.name = `${langCode} ${name}`;;
        return productVariantTranslation;
    }
}