import { configureStore } from '@reduxjs/toolkit';
import EmployeePensionEnrollmentsSlice, {
  fetchPensionEnrollments,
  createPensionEnrollment,
  updatePensionEnrollment,
  deletePensionEnrollment,
  clearEnrollments,
} from '../EmployeePensionEnrollmentsSlice';

jest.mock('@/services/api/api', () => ({
  apiGet: jest.fn(),
  apiPost: jest.fn(),
  apiPut: jest.fn(),
  apiDelete: jest.fn(),
}));

jest.mock('antd', () => ({
  message: { success: jest.fn(), error: jest.fn(), warning: jest.fn() },
}));

jest.mock('@/utilities/Helpers', () => {
  const mockHelpers = { handleServerError: jest.fn((e: any) => e?.message || 'Error') };
  return { __esModule: true, default: mockHelpers };
});

const { apiGet, apiPost, apiPut, apiDelete } = require('@/services/api/api');

const createTestStore = () =>
  configureStore({ reducer: { employeePensionEnrollments: EmployeePensionEnrollmentsSlice.reducer } });

const mockEnrollment = { id: '1', employee_id: 'e1', pension_pool_id: 'pp1', member_number: 'MEM001' };

describe('EmployeePensionEnrollmentsSlice', () => {
  beforeEach(() => jest.clearAllMocks());

  it('has correct initial state', () => {
    const store = createTestStore();
    const state = store.getState().employeePensionEnrollments;
    expect(state.enrollments).toEqual([]);
    expect(state.loading).toBe(false);
    expect(state.error).toBeNull();
  });

  describe('fetchPensionEnrollments', () => {
    it('populates enrollments on fulfilled', async () => {
      apiGet.mockResolvedValue({ data: [mockEnrollment] });
      const store = createTestStore();
      await store.dispatch(fetchPensionEnrollments('e1'));
      expect(store.getState().employeePensionEnrollments.enrollments).toEqual([mockEnrollment]);
    });
  });

  describe('createPensionEnrollment', () => {
    it('appends new enrollment on fulfilled', async () => {
      apiPost.mockResolvedValue({ data: { data: mockEnrollment } });
      const store = createTestStore();
      await store.dispatch(createPensionEnrollment({ employee_id: 'e1' } as any));
      expect(store.getState().employeePensionEnrollments.enrollments).toContainEqual(mockEnrollment);
    });
  });

  describe('updatePensionEnrollment', () => {
    it('updates enrollment in list on fulfilled', async () => {
      const updated = { ...mockEnrollment, member_number: 'MEM002' };
      apiPut.mockResolvedValue({ data: { data: updated } });
      const store = createTestStore();
      store.dispatch({ type: fetchPensionEnrollments.fulfilled.type, payload: [mockEnrollment] });
      await store.dispatch(updatePensionEnrollment({ id: '1', member_number: 'MEM002' } as any));
      expect(store.getState().employeePensionEnrollments.enrollments[0].member_number).toBe('MEM002');
    });
  });

  describe('deletePensionEnrollment', () => {
    it('removes enrollment by id on fulfilled', async () => {
      apiDelete.mockResolvedValue({});
      const store = createTestStore();
      store.dispatch({ type: fetchPensionEnrollments.fulfilled.type, payload: [mockEnrollment] });
      await store.dispatch(deletePensionEnrollment('1'));
      expect(store.getState().employeePensionEnrollments.enrollments).toEqual([]);
    });
  });

  describe('clearEnrollments', () => {
    it('clears all enrollments', () => {
      const store = createTestStore();
      store.dispatch({ type: fetchPensionEnrollments.fulfilled.type, payload: [mockEnrollment] });
      store.dispatch(clearEnrollments());
      expect(store.getState().employeePensionEnrollments.enrollments).toEqual([]);
    });
  });
});