import { configureStore } from '@reduxjs/toolkit';
import TrainingEnrollmentsSlice from '../TrainingEnrollmentsSlice';

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

const createStore = () => configureStore({ reducer: { te: TrainingEnrollmentsSlice.reducer } });

describe('TrainingEnrollmentsSlice', () => {
  it('has correct initial state', () => {
    const store = createStore();
    const state = store.getState().te;
    expect(state.enrollments).toEqual([]);
    expect(state.myTraining).toEqual([]);
    expect(state.meta).toBeNull();
    expect(state.loading).toBe(false);
    expect(state.error).toBeNull();
  });

  it('sets loading on fetch pending', () => {
    const store = createStore();
    store.dispatch({ type: 'trainingEnrollments/fetch/pending' });
    expect(store.getState().te.loading).toBe(true);
  });

  it('populates enrollments on fetch fulfilled', () => {
    const store = createStore();
    store.dispatch({ type: 'trainingEnrollments/fetch/fulfilled', payload: [{ id: '1' }] });
    expect(store.getState().te.enrollments).toEqual([{ id: '1' }]);
    expect(store.getState().te.loading).toBe(false);
  });

  it('prepends on create fulfilled', () => {
    const store = createStore();
    store.dispatch({ type: 'trainingEnrollments/fetch/fulfilled', payload: [{ id: '1' }] });
    store.dispatch({ type: 'trainingEnrollments/create/fulfilled', payload: { id: '2' } });
    expect(store.getState().te.enrollments[0]).toEqual({ id: '2' });
  });

  it('removes on delete fulfilled', () => {
    const store = createStore();
    store.dispatch({ type: 'trainingEnrollments/fetch/fulfilled', payload: [{ id: '1' }, { id: '2' }] });
    store.dispatch({ type: 'trainingEnrollments/delete/fulfilled', payload: { id: '1', message: 'Deleted' } });
    expect(store.getState().te.enrollments).toEqual([{ id: '2' }]);
  });

  it('populates myTraining on fetchMyTraining fulfilled', () => {
    const store = createStore();
    store.dispatch({ type: 'trainingEnrollments/fetchMyTraining/fulfilled', payload: [{ id: 'm1' }] });
    expect(store.getState().te.myTraining).toEqual([{ id: 'm1' }]);
  });

  it('updates enrollment in-place on approve fulfilled', () => {
    const store = createStore();
    store.dispatch({ type: 'trainingEnrollments/fetch/fulfilled', payload: [{ id: '1', status: 'pending' }] });
    store.dispatch({ type: 'trainingEnrollments/approve/fulfilled', payload: { id: '1', status: 'approved' } });
    expect(store.getState().te.enrollments[0].status).toBe('approved');
  });
});