import { configureStore } from '@reduxjs/toolkit';
import PipMilestoneSlice, { clearMilestones, fetchPipMilestones, createPipMilestone, reviewPipMilestone, deletePipMilestone } from '../PipMilestoneSlice';

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', () => ({
  __esModule: true, default: { handleServerError: jest.fn((e) => e?.message || 'Error') } }));

const createStore = () => configureStore({ reducer: { pm: PipMilestoneSlice.reducer } });

describe('PipMilestoneSlice', () => {
  it('has correct initial state', () => {
    const store = createStore();
    const state = store.getState().pm;
    expect(state.milestones).toEqual([]);
    expect(state.currentMilestone).toBeNull();
    expect(state.loading).toBe(false);
    expect(state.error).toBeNull();
  });

  it('populates on fetchPipMilestones.fulfilled', () => {
    const store = createStore();
    store.dispatch({ type: fetchPipMilestones.fulfilled.type, payload: [{ id: '1' }] });
    expect(store.getState().pm.milestones).toEqual([{ id: '1' }]);
  });

  it('appends on createPipMilestone.fulfilled', () => {
    const store = createStore();
    store.dispatch({ type: createPipMilestone.fulfilled.type, payload: { id: '1' } });
    expect(store.getState().pm.milestones).toHaveLength(1);
  });

  it('updates on reviewPipMilestone.fulfilled', () => {
    const store = createStore();
    store.dispatch({ type: fetchPipMilestones.fulfilled.type, payload: [{ id: '1', status: 'pending' }] });
    store.dispatch({ type: reviewPipMilestone.fulfilled.type, payload: { id: '1', status: 'approved' } });
    expect(store.getState().pm.milestones[0].status).toBe('approved');
  });

  it('removes on deletePipMilestone.fulfilled', () => {
    const store = createStore();
    store.dispatch({ type: fetchPipMilestones.fulfilled.type, payload: [{ id: '1' }] });
    store.dispatch({ type: deletePipMilestone.fulfilled.type, payload: { id: '1', message: 'Deleted' } });
    expect(store.getState().pm.milestones).toHaveLength(0);
  });

  it('clears via clearMilestones', () => {
    const store = createStore();
    store.dispatch({ type: fetchPipMilestones.fulfilled.type, payload: [{ id: '1' }] });
    store.dispatch(clearMilestones());
    expect(store.getState().pm.milestones).toEqual([]);
    expect(store.getState().pm.currentMilestone).toBeNull();
  });
});