-- ============================================
-- rioCata - Tests de Índices
-- ============================================
-- Tests que verifican que los índices se están
-- usando correctamente con EXPLAIN
-- ============================================

\echo '=========================================='
\echo 'Tests de Uso de Índices'
\echo '=========================================='
\echo ''

-- ============================================
-- TEST: Índice GIN en tazas_defectuosas con operador @>
-- ============================================
\echo '[INDEX 1] Verificando uso de índice GIN en tazas_defectuosas...'

DO $$
DECLARE
  explain_output text;
  uses_index boolean := false;
BEGIN
  -- Obtener plan de ejecución
  SELECT string_agg(line, E'\n')
  INTO explain_output
  FROM (
    SELECT * FROM (
      EXPLAIN SELECT * FROM evaluacion
      WHERE tazas_defectuosas @> ARRAY[5]::smallint[]
    ) AS explain_lines(line)
  ) AS subq;

  -- Verificar que usa índice GIN
  IF explain_output LIKE '%idx_eval_tazas_defectuosas%' OR
     explain_output LIKE '%Bitmap Index Scan%' THEN
    uses_index := true;
  END IF;

  IF uses_index THEN
    RAISE NOTICE '  ✓ Query usa índice GIN para tazas_defectuosas con @>';
  ELSE
    RAISE WARNING '  ⚠ Query NO usa índice GIN (puede ser normal con tabla vacía)';
    RAISE NOTICE 'Plan: %', explain_output;
  END IF;

  RAISE NOTICE '✓ Verificación de índice GIN en tazas_defectuosas completada';
END $$;

\echo ''

-- ============================================
-- TEST: Índice funcional en acidez afectiva
-- ============================================
\echo '[INDEX 2] Verificando uso de índice funcional en acidez afectiva...'

DO $$
DECLARE
  explain_output text;
  uses_index boolean := false;
BEGIN
  -- Obtener plan de ejecución
  SELECT string_agg(line, E'\n')
  INTO explain_output
  FROM (
    SELECT * FROM (
      EXPLAIN SELECT * FROM evaluacion
      WHERE ((intensidades->'acidez'->>'afectiva')::int) >= 8
    ) AS explain_lines(line)
  ) AS subq;

  -- Verificar que usa índice funcional
  IF explain_output LIKE '%idx_eval_int_acidez_afectiva%' OR
     explain_output LIKE '%Index%' THEN
    uses_index := true;
  END IF;

  IF uses_index THEN
    RAISE NOTICE '  ✓ Query usa índice funcional para acidez afectiva';
  ELSE
    RAISE WARNING '  ⚠ Query NO usa índice funcional (puede ser normal con tabla vacía)';
  END IF;

  RAISE NOTICE '✓ Verificación de índice funcional en acidez completada';
END $$;

\echo ''

-- ============================================
-- TEST: Índice B-tree en puntaje_final
-- ============================================
\echo '[INDEX 3] Verificando uso de índice B-tree en puntaje_final...'

DO $$
DECLARE
  explain_output text;
  uses_index boolean := false;
BEGIN
  -- Obtener plan de ejecución para ORDER BY
  SELECT string_agg(line, E'\n')
  INTO explain_output
  FROM (
    SELECT * FROM (
      EXPLAIN SELECT * FROM evaluacion
      ORDER BY puntaje_final DESC
      LIMIT 10
    ) AS explain_lines(line)
  ) AS subq;

  -- Verificar que usa índice
  IF explain_output LIKE '%idx_eval_puntaje_final%' OR
     explain_output LIKE '%Index%puntaje%' THEN
    uses_index := true;
  END IF;

  IF uses_index THEN
    RAISE NOTICE '  ✓ Query usa índice B-tree para puntaje_final';
  ELSE
    RAISE WARNING '  ⚠ Query NO usa índice B-tree (puede ser normal con tabla vacía)';
  END IF;

  RAISE NOTICE '✓ Verificación de índice B-tree en puntaje_final completada';
END $$;

\echo ''

-- ============================================
-- TEST: Índice GIN en intensidades JSONB
-- ============================================
\echo '[INDEX 4] Verificando uso de índice GIN en intensidades JSONB...'

DO $$
DECLARE
  explain_output text;
  uses_index boolean := false;
BEGIN
  -- Obtener plan de ejecución
  SELECT string_agg(line, E'\n')
  INTO explain_output
  FROM (
    SELECT * FROM (
      EXPLAIN SELECT * FROM evaluacion
      WHERE intensidades @> '{"dulzor":{"afectiva":10}}'::jsonb
    ) AS explain_lines(line)
  ) AS subq;

  -- Verificar que usa índice GIN
  IF explain_output LIKE '%idx_eval_json_intensidades%' OR
     explain_output LIKE '%Bitmap Index Scan%intensidades%' THEN
    uses_index := true;
  END IF;

  IF uses_index THEN
    RAISE NOTICE '  ✓ Query usa índice GIN para búsqueda en JSONB';
  ELSE
    RAISE WARNING '  ⚠ Query NO usa índice GIN JSONB (puede ser normal con tabla vacía)';
  END IF;

  RAISE NOTICE '✓ Verificación de índice GIN en JSONB completada';
END $$;

\echo ''

-- ============================================
-- TEST: Índices con datos reales
-- ============================================
\echo '[INDEX 5] Verificando uso de índices con datos reales...'

DO $$
DECLARE
  test_sesion_id uuid := gen_random_uuid();
  test_user_id uuid := gen_random_uuid();
  test_part_id uuid := gen_random_uuid();
  test_muestra_id uuid;
  explain_output text;
  i int;
BEGIN
  -- Crear datos de prueba (100 evaluaciones)
  INSERT INTO auth.users (id, email, nombre)
  VALUES (test_user_id, 'test_index@test.com', 'Test User');

  INSERT INTO sesion (id, codigo, fecha)
  VALUES (test_sesion_id, 'TEST-INDEX', CURRENT_DATE);

  INSERT INTO sesion_participante (id, sesion_id, catador_id)
  VALUES (test_part_id, test_sesion_id, test_user_id);

  -- Insertar 100 muestras y evaluaciones
  FOR i IN 1..100 LOOP
    test_muestra_id := gen_random_uuid();

    INSERT INTO muestra (id, sesion_id, codigo)
    VALUES (test_muestra_id, test_sesion_id, 'M-' || i);

    INSERT INTO evaluacion (
      muestra_id, sesion_participante_id, intensidades,
      tazas_defectuosas, puntaje_final
    ) VALUES (
      test_muestra_id, test_part_id,
      jsonb_build_object(
        'acidez', jsonb_build_object('descriptiva', (i % 15) + 1, 'afectiva', (i % 10) + 1),
        'dulzor', jsonb_build_object('descriptiva', ((i * 2) % 15) + 1, 'afectiva', ((i * 2) % 10) + 1)
      ),
      CASE WHEN i % 10 = 0 THEN ARRAY[5]::smallint[] ELSE ARRAY[]::smallint[] END,
      (i % 80) + 1
    );
  END LOOP;

  RAISE NOTICE '  ✓ Creadas 100 evaluaciones para testing de índices';

  -- Test 1: Índice en puntaje_final con datos
  SELECT string_agg(line, E'\n')
  INTO explain_output
  FROM (
    SELECT * FROM (
      EXPLAIN SELECT * FROM evaluacion
      WHERE puntaje_final > 50
      ORDER BY puntaje_final DESC
    ) AS explain_lines(line)
  ) AS subq;

  IF explain_output LIKE '%Index%' THEN
    RAISE NOTICE '  ✓ Índice usado en puntaje_final con 100 registros';
  ELSE
    RAISE NOTICE '  - Plan para puntaje_final: Seq Scan (normal con 100 registros)';
  END IF;

  -- Test 2: Índice GIN en arrays con datos
  SELECT string_agg(line, E'\n')
  INTO explain_output
  FROM (
    SELECT * FROM (
      EXPLAIN SELECT * FROM evaluacion
      WHERE tazas_defectuosas @> ARRAY[5]::smallint[]
    ) AS explain_lines(line)
  ) AS subq;

  IF explain_output LIKE '%Index%' OR explain_output LIKE '%Bitmap%' THEN
    RAISE NOTICE '  ✓ Índice GIN usado en tazas_defectuosas con 100 registros';
  ELSE
    RAISE NOTICE '  - Plan para tazas_defectuosas: Seq Scan (normal con 100 registros)';
  END IF;

  -- Cleanup
  DELETE FROM sesion WHERE id = test_sesion_id;
  DELETE FROM auth.users WHERE id = test_user_id;

  RAISE NOTICE '✓ Verificación con datos reales completada';
  RAISE NOTICE '';
  RAISE NOTICE 'NOTA: Con pocos registros PostgreSQL puede preferir Seq Scan sobre índices.';
  RAISE NOTICE '      Los índices se usan más efectivamente con miles de registros.';
END $$;

\echo ''

-- ============================================
-- TEST: Listado de todos los índices
-- ============================================
\echo '[INDEX 6] Listando todos los índices de rioCata...'

SELECT
  schemaname AS schema,
  tablename AS tabla,
  indexname AS indice,
  indexdef AS definicion
FROM pg_indexes
WHERE schemaname = 'public'
  AND tablename IN ('sesion', 'sesion_participante', 'muestra', 'evaluacion')
ORDER BY tablename, indexname;

\echo ''
\echo '=========================================='
\echo 'Tests de Índices completados'
\echo '=========================================='
\echo ''
\echo 'NOTA IMPORTANTE:'
\echo 'Los índices GIN y funcionales se aprovechan mejor con grandes'
\echo 'volúmenes de datos (miles de registros). Con datos de prueba'
\echo 'pequeños, PostgreSQL puede optar por Seq Scan que es más eficiente.'
\echo ''