UserRepository tests

UserRepositoryTests.ts

import { Api } from './Api';
import { UserRepository } from './UserRepository';

jest.mock('./Api', () => {
  return {
    Api: jest.fn().mockImplementation(() => {
      return {
        getUser: jest.fn((id: number) => {
          return id === 1 ? '[email protected]' : null;
        }),
      };
    })
  };
});

describe('UserRepository', () => {
  let api: Api;
  let userRepository: UserRepository;

  beforeEach(() => {
    api = new Api();
    userRepository = new UserRepository(api);
  });

  test('getUserEmail returns the correct email for a valid ID', async () => {
    const email = await userRepository.getUserEmail(1);
    expect(email).toBe('[email protected]');
    expect(api.getUser).toHaveBeenCalledWith(1);
  });

  test('getUserEmail should return null for an invalid ID', async () => {
    const email = await userRepository.getUserEmail(2);
    expect(email).toBeNull();
    expect(api.getUser).toHaveBeenCalledWith(2);
  });
});