#coding:utf-8

"""
ID:          issue-4641
ISSUE:       4641
TITLE:       Regression: Predicates involving PSQL variables/parameters are not pushed inside the aggregation
DESCRIPTION:
JIRA:        CORE-4318
FBTEST:      bugs.core_4318
"""

import pytest
from firebird.qa import *

init_script = """
    recreate table t2 (
      id integer not null,
      t1_id integer
    );
    commit;

    recreate table t1 (
      id integer not null
    );
    commit;

    set term ^;

    execute block
    as
    declare variable i integer = 0;
    begin
      while (i < 1000) do begin
        i = i + 1;

        insert into t2(id, t1_id) values(:i, mod(:i, 10));

        merge into t1 using (
          select mod(:i, 10) as f from rdb$database
        ) src on t1.id = src.f
        when not matched then
           insert (id) values(src.f);

      end -- while (i < 1000) do begin

    end^
    set term ;^
    commit;

    alter table t1 add constraint pk_t1 primary key (id);
    alter table t2 add constraint pk_t2 primary key (id);
    alter table t2 add constraint fk_t2_ref_t1 foreign key (t1_id) references t1(id);
    commit;
"""

db = db_factory(init=init_script)

test_script = """
    set explain on;
    set planonly;

    set term ^;
    execute block returns (s integer) as
        declare v integer = 1;
    begin
      with t as (
          select t1_id as t1_id, sum(id) as s -- FB 5.x: "Select Expression (line NNN, column MMM)"
          from t2
          group by 1
      )
      select s
      from t
      where t1_id = :v
      into :s;

      suspend;
    end
    ^
    set term ;^
    -- In 3.0.0.30837 plan was:
    -- Select Expression
    --    -> Singularity Check
    --        -> Filter
    --            -> Aggregate
    --                -> Table "T T2" Access By ID
    --                    -> Index "FK_T2_REF_T1" Scan
    -- (i.e. there was NO "Filter" between "Aggregate" and "Table "T T2" Access By ID")
"""

act = isql_act('db', test_script, substitutions = [('line \\d+, col(umn)? \\d+', 'line, col')])

fb3x_expected_out = """
    Select Expression
        -> Singularity Check
            -> Filter
                -> Aggregate
                    -> Filter
                        -> Table "T2" as "T T2" Access By ID
                            -> Index "FK_T2_REF_T1" Range Scan (full match)
"""

fb5x_expected_out = """
    Select Expression (line 8, column 7)
        -> Singularity Check
            -> Filter
                -> Aggregate
                    -> Filter
                        -> Table "T2" as "T T2" Access By ID
                            -> Index "FK_T2_REF_T1" Range Scan (full match)
"""

@pytest.mark.version('>=3.0')
def test_1(act: Action):
    act.expected_stdout = fb3x_expected_out if act.is_version('<5') else fb5x_expected_out
    act.execute()
    #assert act.stdout == act.clean_expected_stdout
    assert act.clean_stdout == act.clean_expected_stdout