import { configureStore } from '@reduxjs/toolkit';
import OkrKeyResultSlice, { fetchOkrKeyResults, createOkrKeyResult, updateOkrKeyResult, deleteOkrKeyResult } from '../OkrKeyResultSlice';

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: { okr: OkrKeyResultSlice.reducer } });

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

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

  it('prepends on createOkrKeyResult.fulfilled', () => {
    const store = createStore();
    store.dispatch({ type: createOkrKeyResult.fulfilled.type, payload: { id: '1' } });
    expect(store.getState().okr.okrKeyResults[0]).toEqual({ id: '1' });
  });

  it('updates in place on updateOkrKeyResult.fulfilled', () => {
    const store = createStore();
    store.dispatch({ type: fetchOkrKeyResults.fulfilled.type, payload: [{ id: '1', title: 'old' }] });
    store.dispatch({ type: updateOkrKeyResult.fulfilled.type, payload: { id: '1', title: 'new' } });
    expect(store.getState().okr.okrKeyResults[0].title).toBe('new');
  });

  it('removes on deleteOkrKeyResult.fulfilled', () => {
    const store = createStore();
    store.dispatch({ type: fetchOkrKeyResults.fulfilled.type, payload: [{ id: '1' }] });
    store.dispatch({ type: deleteOkrKeyResult.fulfilled.type, payload: { id: '1', message: 'Deleted' } });
    expect(store.getState().okr.okrKeyResults).toHaveLength(0);
  });
});
