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

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: { ta: TrainingAttendanceSlice.reducer } });

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

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

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

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

  it('updates in-place on update fulfilled', () => {
    const store = createStore();
    store.dispatch({ type: 'trainingAttendance/fetch/fulfilled', payload: [{ id: '1', status: 'present' }] });
    store.dispatch({ type: 'trainingAttendance/update/fulfilled', payload: { id: '1', status: 'absent' } });
    expect(store.getState().ta.attendance[0].status).toBe('absent');
  });

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

  it('handles bulk attendance by updating existing and adding new', () => {
    const store = createStore();
    store.dispatch({ type: 'trainingAttendance/fetch/fulfilled', payload: [{ id: '1', status: 'absent' }] });
    store.dispatch({
      type: 'trainingAttendance/markBulkAttendance/fulfilled',
      payload: { message: 'Done', data: [{ id: '1', status: 'present' }, { id: '2', status: 'present' }] },
    });
    expect(store.getState().ta.attendance).toHaveLength(2);
    expect(store.getState().ta.attendance[0].status).toBe('present');
  });
});