undeexcepturi/tests/features/cli/DatabaseSeedCommand.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 { SeedManager } from '@mikro-orm/seeder';
import { Configuration, MikroORM } from '@mikro-orm/core';
import { SqliteDriver } from '@mikro-orm/sqlite';
import { MongoDriver } from '@mikro-orm/mongodb';
import { CLIHelper } from '@mikro-orm/cli';

const config = new Configuration({ driver: MongoDriver } as any, false);
const showHelpMock = jest.spyOn(CLIHelper, 'showHelp');
showHelpMock.mockImplementation(() => void 0);
const closeSpy = jest.spyOn(MikroORM.prototype, 'close');
const getConfigMock = jest.spyOn(CLIHelper, 'getConfiguration');
getConfigMock.mockResolvedValue(config as any);
const dumpMock = jest.spyOn(CLIHelper, 'dump');
dumpMock.mockImplementation(() => void 0);
const seed = jest.spyOn(SeedManager.prototype, 'seedString');
seed.mockImplementation(async () => void 0);

(global as any).console.log = jest.fn();

import { DatabaseSeedCommand } from '../../../packages/cli/src/commands/DatabaseSeedCommand';
import { initORMSqlite } from '../../bootstrap';

describe('DatabaseSeedCommand', () => {

  let orm: MikroORM<SqliteDriver>;

  beforeAll(async () => {
    orm = await initORMSqlite();
    const getORMMock = jest.spyOn(CLIHelper, 'getORM');
    getORMMock.mockResolvedValue(orm);
  });

  afterAll(async () => await orm.close(true));

  test('handler', async () => {
    const cmd = new DatabaseSeedCommand();

    const mockOption = jest.fn();
    const args = { option: mockOption };
    cmd.builder(args as any);
    expect(mockOption).toHaveBeenCalledWith('c', {
      alias: 'class',
      type: 'string',
      desc: 'Seeder class to run',
    });
    await expect(cmd.handler({} as any)).resolves.toBeUndefined();
    expect(seed).toHaveBeenCalledTimes(1);
    expect(seed).toHaveBeenCalledWith((orm.config.get('seeder').defaultSeeder));
    expect(closeSpy).toHaveBeenCalledTimes(1);

    await expect(cmd.handler({ class: 'TestSeeder' } as any)).resolves.toBeUndefined();
    expect(seed).toHaveBeenCalledTimes(2);
    expect(seed).toHaveBeenCalledWith(('TestSeeder'));
    expect(closeSpy).toHaveBeenCalledTimes(2);
  });

});