(global as any).process.env.FORCE_COLOR = 0;
(global as any).console.log = jest.fn();

import { MikroORM } from '@mikro-orm/core';
import { SqliteDriver } from '@mikro-orm/sqlite';
import { CLIHelper } from '@mikro-orm/cli';
import { SeedManager } from '@mikro-orm/seeder';
import { CreateSeederCommand } from '../../../packages/cli/src/commands/CreateSeederCommand';
import { initORMSqlite } from '../../bootstrap';

const closeSpy = jest.spyOn(MikroORM.prototype, 'close');
jest.spyOn(CLIHelper, 'showHelp').mockImplementation(() => void 0);
const createSeederMock = jest.spyOn(SeedManager.prototype, 'createSeeder');
createSeederMock.mockResolvedValue('database/seeders/database.seeder.ts');

describe('CreateSeederCommand', () => {

  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 CreateSeederCommand();
    const mockPositional = jest.fn();
    const mockDemand = jest.fn();
    const args = { positional: mockPositional, demandOption: mockDemand };
    cmd.builder(args as any);
    expect(mockPositional).toBeCalledWith('seeder', {
      describe: 'Name for the seeder class. (e.g. "test" will generate "TestSeeder" or "TestSeeder" will generate "TestSeeder")',
    });
    expect(mockDemand).toBeCalledWith('seeder');

    await expect(cmd.handler({ seeder: 'DatabaseSeeder' } as any)).resolves.toBeUndefined();
    expect(createSeederMock).toBeCalledTimes(1);
    expect(closeSpy).toBeCalledTimes(1);
  });

  test('should generate seeder class with all kind of names', async () => {
    const cmd = new CreateSeederCommand();
    // @ts-expect-error private method
    const spy = jest.spyOn(CreateSeederCommand, 'getSeederClassName');
    await cmd.handler({ seeder: 'DatabaseSeeder' } as any);
    expect(spy).toHaveLastReturnedWith('DatabaseSeeder');

    await cmd.handler({ seeder: 'database' } as any);
    expect(spy).toHaveLastReturnedWith('DatabaseSeeder');

    await cmd.handler({ seeder: 'Test' } as any);
    expect(spy).toHaveLastReturnedWith('TestSeeder');

    await cmd.handler({ seeder: 'project-names' } as any);
    expect(spy).toHaveLastReturnedWith('ProjectNamesSeeder');

    await cmd.handler({ seeder: 'project-names-seeder' } as any);
    expect(spy).toHaveLastReturnedWith('ProjectNamesSeeder');
  });
});