undeexcepturi/tests/features/seeder/seed-manager.test.ts

TypeScript ORM for Node.js based on Data Mapper, Unit of Work and Identity Map patterns. Supports MongoDB, MySQL, PostgreSQL and SQLite databases as well as usage with vanilla JavaScript.

import { readFile, remove } from 'fs-extra';
import { MikroORM, Utils } from '@mikro-orm/core';
import { SeedManager } from '@mikro-orm/seeder';
import { SchemaGenerator } from '@mikro-orm/sqlite';
import { initORMSqlite } from '../../bootstrap';
import { Book3Seeder } from '../../database/seeder/book3.seeder';
import { Author3Seeder } from '../../database/seeder/author3.seeder';

describe('Seeder', () => {

  let orm: MikroORM;

  beforeAll(async () => {
    orm = await initORMSqlite();
    const getORMMock = jest.spyOn(MikroORM, 'init');
    getORMMock.mockResolvedValue(orm);
    const createSchema = jest.spyOn(SchemaGenerator.prototype, 'createSchema');
    createSchema.mockImplementation();
    const dropSchema = jest.spyOn(SchemaGenerator.prototype, 'dropSchema');
    dropSchema.mockImplementation();
  });

  beforeEach(() => orm.config.resetServiceCache());

  afterAll(async () => {
    await orm.close(true);
    jest.restoreAllMocks();
  });

  test('seed', async () => {
    const bookRunMock = jest.spyOn(Book3Seeder.prototype, 'run');
    bookRunMock.mockImplementation(async () => void 0);
    const authorRunMock = jest.spyOn(Author3Seeder.prototype, 'run');
    authorRunMock.mockImplementation(async () => void 0);

    await orm.seeder.seed(Book3Seeder);
    expect(bookRunMock).toHaveBeenCalledTimes(1);
    await orm.seeder.seed(Book3Seeder, Author3Seeder);
    expect(bookRunMock).toHaveBeenCalledTimes(2);
    expect(authorRunMock).toHaveBeenCalledTimes(1);
  });

  test('seedString', async () => {
    const options = orm.config.get('seeder');
    options.path = './database/seeder';
    options.defaultSeeder = 'DatabaseSeeder';
    orm.config.set('seeder', options);
    const seedMock = jest.spyOn(SeedManager.prototype, 'seed');

    await orm.seeder.seedString('Book3Seeder');
    expect(seedMock).toHaveBeenCalledTimes(1);
    await orm.seeder.seedString('Book3Seeder', 'Author3Seeder');
    expect(seedMock).toHaveBeenCalledTimes(3);

    const re = 'Seeder class Unknown not found in ./tests/database/seeder/!(*.d).{js,ts}';
    await expect(orm.seeder.seedString('Unknown')).rejects.toThrow(re);
  });

  test('createSeeder (TS)', async () => {
    const options = orm.config.get('seeder');
    options.path = Utils.normalizePath(process.cwd()) + '/temp/seeders';
    options.defaultSeeder = 'DatabaseSeeder';
    orm.config.set('seeder', options);
    const seederFile = await orm.seeder.createSeeder('Publisher3Seeder');
    expect(seederFile).toBe(Utils.normalizePath(process.cwd()) + `/temp/seeders/Publisher3Seeder.ts`);
    const fileContents = await readFile(seederFile, 'utf8');
    expect(fileContents).toContain('export class Publisher3Seeder extends Seeder {');
    await remove(seederFile);
  });

  test('createSeeder (JS)', async () => {
    const options = orm.config.get('seeder');
    options.path = Utils.normalizePath(process.cwd()) + '/temp/seeders';
    options.emit = 'js';
    options.defaultSeeder = 'DatabaseSeeder';
    orm.config.set('seeder', options);
    const seederFile = await orm.seeder.createSeeder('Publisher3Seeder');
    expect(seederFile).toBe(Utils.normalizePath(process.cwd()) + `/temp/seeders/Publisher3Seeder.js`);
    const fileContents = await readFile(seederFile, 'utf8');
    expect(fileContents).toContain('exports.Publisher3Seeder = Publisher3Seeder;');
    await remove(seederFile);
  });

});