-- More AM-agnostic infrastructure tests, using more complex queries

-- Remove all the hypothetical indexes if any
SELECT hypopg_reset();

-- Check that custom objects cannot override hypopg internal hypothetical oid
-- threshold retrieval.  This is done during the first hypothetical index
-- creation in a backend so it needs to be first
CREATE FUNCTION public.max_state(oid, oid) RETURNS oid AS
$$
BEGIN
    RAISE EXCEPTION 'custom aggregate used';
END;
$$
LANGUAGE plpgsql STABLE;
CREATE AGGREGATE public.max(oid) (
    SFUNC = public.max_state,
    STYPE = oid,
    INITCOND = '0'
);

CREATE FUNCTION public.lt(oid, oid) RETURNS bool AS
$$
BEGIN
    RAISE EXCEPTION 'custom operator used';
END;
$$
LANGUAGE plpgsql
SET search_path TO pg_catalog, pg_temp;
CREATE OPERATOR public.< (
    PROCEDURE = public.lt,
    LEFTARG = oid,
    RIGHTARG = oid
);

SET search_path = public, pg_catalog;
SELECT count(*)
FROM public.hypopg_create_index('CREATE INDEX ON public.hypo (id)');

RESET search_path;
SELECT hypopg_reset();

-------------------
-- JOIN with an SRF
-------------------

-- Query should do a seq scan on hypo
SELECT COUNT(*) FROM do_explain($$
    SELECT *
    FROM hypo h
    JOIN generate_series(1, 10) f(g) ON h.id = f.g
$$) e
WHERE e ~ 'Seq Scan on hypo';

SELECT COUNT(*) AS NB
FROM hypopg_create_index('CREATE INDEX ON hypo(id)');

-- Should use the hypothetical index
SELECT COUNT(*) FROM do_explain($$
    SELECT *
    FROM hypo h
    JOIN generate_series(1, 10) f(g) ON h.id = f.g
$$) e
WHERE e ~ 'Index.*<\d+>btree_hypo';

-- check that an error doesn't leak the "is_explain" state
SET enable_seqscan = off;
DO $$
BEGIN
    EXECUTE 'EXPLAIN SELECT 1 / 0';
EXCEPTION WHEN division_by_zero THEN
    NULL;
END
$$;
SELECT COUNT(*) = 1 AS ordinary_query_ok FROM hypo WHERE id = 1;
RESET enable_seqscan;
SELECT hypopg_reset();

------------------------------------------------------------
-- Table with hypothetical partial indexes in various places
------------------------------------------------------------

-- an extra index is necessary to force get_actual_variable_range() being run
CREATE INDEX real_idx ON hypo (id);

SELECT count(*)
FROM public.hypopg_create_index('CREATE INDEX ON public.hypo (id) WHERE id > 0');

SELECT COUNT(*) FROM do_explain($$
    SELECT *
    FROM generate_series(1, 10) f(g)
    JOIN hypo h ON h.id = f.g
    WHERE id > 0
$$) e
WHERE e ~ 'Index.*<\d+>btree_hypo';

CREATE VIEW v_hypo AS
SELECT *
FROM hypo
WHERE id = 1;

SELECT COUNT(*) FROM do_explain($$
    SELECT *
    FROM v_hypo
    WHERE id = 1
$$) e
WHERE e ~ 'Index.*<\d+>btree_hypo';

DROP INDEX real_idx;
DROP VIEW v_hypo;