import { configureStore } from '@reduxjs/toolkit';
import PensionPoolsSlice, {
  fetchPensionPools,
  createPensionPool,
  updatePensionPool,
  deletePensionPool,
} from '../PensionPoolsSlice';

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: { pensionPools: PensionPoolsSlice.reducer } });

const mockPool = { id: '1', name: 'Tier 1', country_code: 'GH', is_active: true };

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

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

  describe('fetchPensionPools', () => {
    it('populates pools from array response on fulfilled', async () => {
      apiGet.mockResolvedValue({ data: [mockPool] });
      const store = createTestStore();
      await store.dispatch(fetchPensionPools());
      expect(store.getState().pensionPools.pensionPools).toEqual([mockPool]);
    });

    it('handles paginated response format', async () => {
      apiGet.mockResolvedValue({ data: { data: [mockPool] } });
      const store = createTestStore();
      await store.dispatch(fetchPensionPools());
      expect(store.getState().pensionPools.pensionPools).toEqual([mockPool]);
    });
  });

  describe('createPensionPool', () => {
    it('prepends new pool on fulfilled', async () => {
      apiPost.mockResolvedValue({ data: { data: mockPool } });
      const store = createTestStore();
      await store.dispatch(createPensionPool({ name: 'Tier 1' } as any));
      expect(store.getState().pensionPools.pensionPools[0]).toEqual(mockPool);
    });
  });

  describe('updatePensionPool', () => {
    it('updates pool in list on fulfilled', async () => {
      const updated = { ...mockPool, name: 'Tier 2' };
      apiPut.mockResolvedValue({ data: { data: updated } });
      const store = createTestStore();
      store.dispatch({ type: fetchPensionPools.fulfilled.type, payload: [mockPool] });
      await store.dispatch(updatePensionPool({ id: '1', name: 'Tier 2' } as any));
      expect(store.getState().pensionPools.pensionPools[0].name).toBe('Tier 2');
    });
  });

  describe('deletePensionPool', () => {
    it('removes pool by id on fulfilled', async () => {
      apiDelete.mockResolvedValue({});
      const store = createTestStore();
      store.dispatch({ type: fetchPensionPools.fulfilled.type, payload: [mockPool] });
      await store.dispatch(deletePensionPool('1'));
      expect(store.getState().pensionPools.pensionPools).toEqual([]);
    });
  });
});