firebird-qa/tests/bugs/core_4161_test.py

#coding:utf-8
#
# id:           bugs.core_4161
# title:        User can not insert records into table with column "generated by default as identity" in its DDL
# decription:   
# tracker_id:   CORE-4161
# min_versions: ['3.0']
# versions:     3.0
# qmid:         

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

# version: 3.0
# resources: None

substitutions_1 = []

init_script_1 = """"""

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

test_script_1 = """
    set wng off;
    set list on;

    create or alter user tmp$core4161 password '123';
    commit;
    revoke all on all from tmp$core4161;
    commit;
    
    recreate table autoid_test(id int generated by default as identity primary key, f01 int);
    insert into autoid_test(f01) values(100);
    insert into autoid_test(f01) values(200);
    insert into autoid_test(f01) values(300);
    commit;

    recreate table backup_data(id int, f01 int);
    insert into backup_data(id, f01) select id, f01 from autoid_test;
    commit;

    grant insert,update,delete,select on autoid_test to tmp$core4161;
    grant select on backup_data to tmp$core4161;
    commit;

    set term ^;
    execute block returns( who_am_i type of column sec$users.sec$user_name, mine_id int, mine_f01 int )
    as
      declare v_dbname type of column mon$database.mon$database_name;
      declare v_usr type of column sec$users.sec$user_name = 'tmp$core4161';
      declare v_pwd varchar(20) = '123';
      declare v_connect varchar(255);
      declare v_dummy int;
    begin
      select 'localhost:' || d.mon$database_name
      from mon$database d
      into v_dbname;
    
      execute statement 'select count(*) from autoid_test'
      on external (v_dbname)
      as user (v_usr) password (v_pwd)
      into v_dummy;
    
      execute statement ('update autoid_test set f01=-f01 where id = :x') (x := 1)
      on external (v_dbname)
      as user (v_usr) password (v_pwd);
    
      execute statement ('delete from autoid_test where id >= :x') (x := 1)
      on external (v_dbname)
      as user (v_usr) password (v_pwd);
    
      execute statement ('insert into autoid_test(f01) values(:x)') (x := 555)
      on external (v_dbname)
      as user (v_usr) password (v_pwd);
    
      execute statement ('insert into autoid_test(id, f01) values(:i, :x)') (i := -1,  x := -555)
      on external (v_dbname)
      as user (v_usr) password (v_pwd);

      execute statement ('insert into autoid_test(f01) select f01 from backup_data')
      on external (v_dbname)
      as user (v_usr) password (v_pwd);
      
      for
        execute statement 'select current_user, id, f01 from autoid_test'
        on external (v_dbname)
        as user (v_usr) password (v_pwd)
        into who_am_i, mine_id, mine_f01
      do 
        suspend;
    
    end
    ^
    set term ;^
    rollback;

    select current_user as who_am_i, id as mine_id, f01 as mine_f01
    from autoid_test;
    commit;

    --                                    ||||||||||||||||||||||||||||
    -- ###################################|||  FB 4.0+, SS and SC  |||##############################
    --                                    ||||||||||||||||||||||||||||
    -- If we check SS or SC and ExtConnPoolLifeTime > 0 (config parameter FB 4.0+) then current
    -- DB (bugs.core_NNNN.fdb) will be 'captured' by firebird.exe process and fbt_run utility
    -- will not able to drop this database at the final point of test.
    -- Moreover, DB file will be hold until all activity in firebird.exe completed and AFTER this
    -- we have to wait for <ExtConnPoolLifeTime> seconds after it (discussion and small test see
    -- in the letter to hvlad and dimitr 13.10.2019 11:10).
    -- This means that one need to kill all connections to prevent from exception on cleanup phase:
    -- SQLCODE: -901 / lock time-out on wait transaction / object <this_test_DB> is in use
    -- #############################################################################################
    delete from mon$attachments where mon$attachment_id != current_connection;
    commit;

    -- Do not forget about cleanup - remember that other tests can query sec$users or run `show users`!
    drop user tmp$core4161;
    commit;
  """

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

expected_stdout_1 = """
    WHO_AM_I                        TMP$CORE4161
    MINE_ID                         4
    MINE_F01                        555

    WHO_AM_I                        TMP$CORE4161
    MINE_ID                         -1
    MINE_F01                        -555

    WHO_AM_I                        TMP$CORE4161
    MINE_ID                         5
    MINE_F01                        100

    WHO_AM_I                        TMP$CORE4161
    MINE_ID                         6
    MINE_F01                        200

    WHO_AM_I                        TMP$CORE4161
    MINE_ID                         7
    MINE_F01                        300



    WHO_AM_I                        SYSDBA
    MINE_ID                         1
    MINE_F01                        100

    WHO_AM_I                        SYSDBA
    MINE_ID                         2
    MINE_F01                        200

    WHO_AM_I                        SYSDBA
    MINE_ID                         3
    MINE_F01                        300
  """

@pytest.mark.version('>=3.0')
def test_1(act_1: Action):
    act_1.expected_stdout = expected_stdout_1
    act_1.execute()
    assert act_1.clean_expected_stdout == act_1.clean_stdout