import { configureStore } from '@reduxjs/toolkit';
import AiQuestionsSlice, { clearGeneratedQuestions } from '../AiQuestionsSlice';

jest.mock('@/services/api/api', () => ({ apiPost: 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: { ai: AiQuestionsSlice.reducer } });

describe('AiQuestionsSlice', () => {
  it('has correct initial state', () => {
    const store = createStore();
    const state = store.getState().ai;
    expect(state.generatedQuestions).toEqual([]);
    expect(state.generating).toBe(false);
    expect(state.saving).toBe(false);
    expect(state.error).toBeNull();
  });

  it('sets generating on generate pending', () => {
    const store = createStore();
    store.dispatch({ type: 'assessment/aiQuestions/generate/pending' });
    expect(store.getState().ai.generating).toBe(true);
  });

  it('populates generatedQuestions on generate fulfilled', () => {
    const store = createStore();
    const questions = [{ question_text: 'Q1', type: 'mcq', points: 1, explanation: null, category_id: 'c1', level_id: 'l1', options: [], matches: [] }];
    store.dispatch({
      type: 'assessment/aiQuestions/generate/fulfilled',
      payload: { data: { questions }, message: 'Generated' },
    });
    expect(store.getState().ai.generatedQuestions).toEqual(questions);
    expect(store.getState().ai.generating).toBe(false);
  });

  it('sets saving on saveGenerated pending', () => {
    const store = createStore();
    store.dispatch({ type: 'assessment/aiQuestions/saveGenerated/pending' });
    expect(store.getState().ai.saving).toBe(true);
  });

  it('clears generatedQuestions on saveGenerated fulfilled', () => {
    const store = createStore();
    store.dispatch({
      type: 'assessment/aiQuestions/generate/fulfilled',
      payload: { data: { questions: [{ question_text: 'Q1' }] }, message: 'ok' },
    });
    store.dispatch({ type: 'assessment/aiQuestions/saveGenerated/fulfilled', payload: { data: [], message: 'Saved' } });
    expect(store.getState().ai.generatedQuestions).toEqual([]);
    expect(store.getState().ai.saving).toBe(false);
  });

  it('clears via clearGeneratedQuestions', () => {
    const store = createStore();
    store.dispatch({
      type: 'assessment/aiQuestions/generate/fulfilled',
      payload: { data: { questions: [{ question_text: 'Q1' }] }, message: 'ok' },
    });
    store.dispatch(clearGeneratedQuestions());
    expect(store.getState().ai.generatedQuestions).toEqual([]);
    expect(store.getState().ai.error).toBeNull();
  });
});