firebird-qa/tests/bugs/core_4403_test.py

#coding:utf-8
#
# id:           bugs.core_4403
# title:        Allow referencing cursors as record variables in PSQL
# decription:   
# tracker_id:   CORE-4403
# min_versions: ['3.0']
# versions:     3.0, 4.0
# qmid:         None

import pytest
from firebird.qa import db_factory, isql_act, Action

# version: 3.0
# resources: None

substitutions_1 = []

init_script_1 = """
    recreate table t1(id int primary key, x int, y int);
    recreate table t2(id int primary key, x int, y int);
    recreate table t3(id int primary key, x int, y int);
    commit;
    insert into t1 values(1, 10, 11);
    commit;
    insert into t2 values(2, 10, 22);
    commit;
    insert into t3 values(3, 10, 33);
    commit;
    set term ^;
    create or alter procedure sp_test(a_x int) returns(o_y int) as
    begin
        o_y = 2 * a_x;
        suspend;
    end
    ^
    set term ;^
    commit;
  """

db_1 = db_factory(sql_dialect=3, init=init_script_1)

test_script_1 = """
    set list on;
    set term ^;
    execute block returns(
       t1_id int, t1_x int, t1_y int
      ,t2_id int, t2_x int, t2_y int
      ,t3_id int, t3_x int, t3_y int
    ) as
    begin
      for
        select id, x, y from t1 as cursor c1
      do begin
        for select id, x, y from t2 where x = :c1.x as cursor c2 do
        begin
          for select id, x, y from t3 where x = :c1.x as cursor c3 do
          begin
    
            t1_id = c1.id;
            t1_x  = c1.x;
            t1_y  = c1.y;
    
            t2_id = c2.id;
            t2_x  = c2.x;
            t2_y  = c2.y;
    
            t3_id = c3.id;
            t3_x  = c3.x;
            t3_y  = c3.y;
    
            suspend;
          end
        end
      end
    end 
    ^

    -- This should raise exception "attempted update of read-only column", sample has been taken from:
    -- sql.ru/forum/actualutils.aspx?action=gotomsg&tid=1158905&msg=17704102 
    execute block as
    begin
      for select x, y from t1 as cursor ce do ce.x = ce.y + 1;
    end
    ^

    --/********************
    --### 29.05.2015. TODO ### UNCOMMENT LATER, AFTER FIX CORE-4819. CURRENTLY IT LEADS FB TO HANG / CRASH.

    -- Uncomment 06.08.2018:
    execute block returns(old_y int, new_y int) as
    begin
      for 
          select x, y from t1 
          as cursor ce 
      do begin
          old_y = ce.y;
          execute procedure sp_test(ce.x) returning_values(ce.y);
          new_y = ce.y;
          suspend;
      end
    end
    ^
    -- ********************/    

    set term ;^
    commit;
    set  list off;
  """

act_1 = isql_act('db_1', test_script_1, substitutions=substitutions_1)

expected_stdout_1 = """
   T1_ID                           1
   T1_X                            10
   T1_Y                            11
   T2_ID                           2
   T2_X                            10
   T2_Y                            22
   T3_ID                           3
   T3_X                            10
   T3_Y                            33  
  """
expected_stderr_1 = """
    Statement failed, SQLSTATE = 42000
    attempted update of read-only column

    Statement failed, SQLSTATE = 42000
    attempted update of read-only column
  """

@pytest.mark.version('>=3.0,<4.0')
def test_1(act_1: Action):
    act_1.expected_stdout = expected_stdout_1
    act_1.expected_stderr = expected_stderr_1
    act_1.execute()
    assert act_1.clean_expected_stderr == act_1.clean_stderr
    assert act_1.clean_expected_stdout == act_1.clean_stdout

# version: 4.0
# resources: None

substitutions_2 = []

init_script_2 = """
    recreate table t1(id int primary key, x int, y int);
    recreate table t2(id int primary key, x int, y int);
    recreate table t3(id int primary key, x int, y int);
    commit;
    insert into t1 values(1, 10, 11);
    commit;
    insert into t2 values(2, 10, 22);
    commit;
    insert into t3 values(3, 10, 33);
    commit;
    set term ^;
    create or alter procedure sp_test(a_x int) returns(o_y int) as
    begin
        o_y = 2 * a_x;
        suspend;
    end
    ^
    set term ;^
    commit;
  """

db_2 = db_factory(sql_dialect=3, init=init_script_2)

test_script_2 = """
    set list on;
    set term ^;
    execute block returns(
       t1_id int, t1_x int, t1_y int
      ,t2_id int, t2_x int, t2_y int
      ,t3_id int, t3_x int, t3_y int
    ) as
    begin
      for
        select id, x, y from t1 as cursor c1
      do begin
        for select id, x, y from t2 where x = :c1.x as cursor c2 do
        begin
          for select id, x, y from t3 where x = :c1.x as cursor c3 do
          begin
    
            t1_id = c1.id;
            t1_x  = c1.x;
            t1_y  = c1.y;
    
            t2_id = c2.id;
            t2_x  = c2.x;
            t2_y  = c2.y;
    
            t3_id = c3.id;
            t3_x  = c3.x;
            t3_y  = c3.y;
    
            suspend;
          end
        end
      end
    end 
    ^

    -- This should raise exception "attempted update of read-only column", sample has been taken from:
    -- sql.ru/forum/actualutils.aspx?action=gotomsg&tid=1158905&msg=17704102 
    execute block as
    begin
      for select x, y from t1 as cursor ce do ce.x = ce.y + 1;
    end
    ^

    --/********************
    --### 29.05.2015. TODO ### UNCOMMENT LATER, AFTER FIX CORE-4819. CURRENTLY IT LEADS FB TO HANG / CRASH.

    -- Uncomment 06.08.2018:
    execute block returns(old_y int, new_y int) as
    begin
      for 
          select x, y from t1 
          as cursor ce 
      do begin
          old_y = ce.y;
          execute procedure sp_test(ce.x) returning_values(ce.y);
          new_y = ce.y;
          suspend;
      end
    end
    ^
    --********************/    

    set term ;^
    commit;
    set  list off;
  """

act_2 = isql_act('db_2', test_script_2, substitutions=substitutions_2)

expected_stdout_2 = """
   T1_ID                           1
   T1_X                            10
   T1_Y                            11
   T2_ID                           2
   T2_X                            10
   T2_Y                            22
   T3_ID                           3
   T3_X                            10
   T3_Y                            33  
  """
expected_stderr_2 = """
    Statement failed, SQLSTATE = 42000
    attempted update of read-only column CE.X

    Statement failed, SQLSTATE = 42000
    attempted update of read-only column CE.Y
  """

@pytest.mark.version('>=4.0')
def test_2(act_2: Action):
    act_2.expected_stdout = expected_stdout_2
    act_2.expected_stderr = expected_stderr_2
    act_2.execute()
    assert act_2.clean_expected_stderr == act_2.clean_stderr
    assert act_2.clean_expected_stdout == act_2.clean_stdout