organisation-tariff.service.spec.ts

import { Test, TestingModule } from '@nestjs/testing';
import { OrganisationTariffService } from './organisation-tariff.service';
import { getRepositoryToken } from '@nestjs/typeorm';
import { OrganisationTariff } from './organisation-tariff.entity';
import { Repository } from 'typeorm';

// @ts-ignore
export const repositoryMockFactory: () => MockType<Repository<OrganisationTariff>> = jest.fn(() => ({
  manager: { connection: { close: () => Promise.resolve() } },
  findOneOrFail: jest.fn(entity => entity),
}));

describe('OrganisationTariffService', () => {
  let service: OrganisationTariffService;
  // @ts-ignore
  let repositoryMock: MockType<Repository<OrganisationTariff>>;

  beforeEach(async () => {
    const module: TestingModule = await Test.createTestingModule({
      providers: [OrganisationTariffService, {
        provide: getRepositoryToken(OrganisationTariff),
        useFactory: repositoryMockFactory,
      }],
    }).compile();

    service = module.get<OrganisationTariffService>(OrganisationTariffService);
    repositoryMock = module.get(getRepositoryToken(OrganisationTariff));
  });

  afterEach(async () => { await service.repository.manager.connection.close(); });

  it('should be defined', async () => {
    expect(service).toBeDefined();

    const orgTariff = {id: 1, name: 'TimeOfUse' };
    repositoryMock.findOneOrFail.mockReturnValue(orgTariff);
    expect(await service.findById(orgTariff.id)).toEqual(orgTariff);
    expect(repositoryMock.findOneOrFail).toHaveBeenCalledWith(orgTariff.id);
  });
});