import { configureStore } from '@reduxjs/toolkit';
import LeaveBalancesSlice, { fetchLeaveBalances, fetchMyBalances } from '../LeaveBalancesSlice';

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: { leaveBalances: LeaveBalancesSlice.reducer } });

describe('LeaveBalancesSlice', () => {
  it('has correct initial state', () => {
    const state = createStore().getState().leaveBalances;
    expect(state.balances).toEqual([]);
    expect(state.myBalances).toEqual([]);
    expect(state.employeeBalances).toEqual([]);
    expect(state.loading).toBe(false);
    expect(state.actionLoading).toBe(false);
    expect(state.error).toBeNull();
    expect(state.importResult).toBeNull();
  });

  it('sets loading on fetchLeaveBalances.pending', () => {
    const store = createStore();
    store.dispatch(fetchLeaveBalances.pending('req'));
    expect(store.getState().leaveBalances.loading).toBe(true);
  });

  it('populates balances on fetchLeaveBalances.fulfilled', () => {
    const store = createStore();
    const data = [{ id: '1', balance: 10 }] as any;
    store.dispatch(fetchLeaveBalances.fulfilled(data, 'req'));
    expect(store.getState().leaveBalances.balances).toEqual(data);
    expect(store.getState().leaveBalances.loading).toBe(false);
  });

  it('populates my balances on fetchMyBalances.fulfilled', () => {
    const store = createStore();
    const data = [{ id: '2', balance: 5 }] as any;
    store.dispatch(fetchMyBalances.fulfilled(data, 'req'));
    expect(store.getState().leaveBalances.myBalances).toEqual(data);
  });
});