firebird-qa/tests/bugs/core_3598_test.py

#coding:utf-8

"""
ID:          issue-3952
ISSUE:       3952
TITLE:       TRACE: add statistics of actions that were after transaction finished
DESCRIPTION:
  Three tables are created: permanent, GTT with on commit PRESERVE rows and on commit DELETE rows.

  Trace config is created with *prohibition* of any activity related to security<N>.fdb
  but allow to log transactions-related events (commits and rollbacks) for working database.
  Trace is started before furthe actions.

  Then we launch ISQL and apply two DML for each of these tables:
  1) insert row + commit;
  2) insert row + rollback.

  Finally (after ISQL will finish), we stop trace and parse its log.
  For *each* table TWO lines with performance statristics must exist: both for COMMIT and ROLLBACK events.
JIRA:        CORE-3598
FBTEST:      bugs.core_3598
"""

import pytest
from firebird.qa import *

init_script = """
    recreate table tfix(id int);
    recreate global temporary table gtt_ssn(id int) on commit preserve rows;
    recreate global temporary table gtt_tra(id int) on commit delete rows;
"""

db = db_factory(init=init_script)

act = python_act('db')

test_script = """
    set autoddl off;
    set echo on;
    set count on;
    set bail on;
    insert into tfix(id) values(1);
    commit;
    insert into tfix(id) values(2);
    rollback;
    insert into gtt_ssn(id) values(1);
    commit;
    insert into gtt_ssn(id) values(2);
    rollback;
    insert into gtt_tra(id) values(1);
    commit;
    insert into gtt_tra(id) values(2);
    rollback;
"""

expected_stdout = """
    Statement statistics detected for COMMIT
    Statement statistics detected for COMMIT
    Statement statistics detected for ROLLBACK
    Found performance block header
    Found table statistics for TFIX
    Statement statistics detected for COMMIT
    Statement statistics detected for ROLLBACK
    Found performance block header
    Found table statistics for GTT_SSN
    Statement statistics detected for COMMIT
    Statement statistics detected for ROLLBACK
"""

trace = ['log_transactions = true',
           'print_perf = true',
           'log_initfini = false',
           ]

@pytest.mark.version('>=3.0')
def test_1(act: Action, capsys):
    with act.trace(db_events=trace):
        act.isql(switches=[], input=test_script)
    # Output log of trace session, with filtering only interested info:
    # Performance header text (all excessive spaces will be removed before comparison - see below):
    perf_header='Table                             Natural     Index    Update    Insert    Delete   Backout     Purge   Expunge'
    checked_events= {') COMMIT_TRANSACTION': 'commit',
                     ') ROLLBACK_TRANSACTION': 'rollback',
                     ') EXECUTE_STATEMENT': 'execute_statement',
                     ') START_TRANSACTION': 'start_transaction'
                     }
    i, k = 0, 0
    watched_event = ''
    for line in act.trace_log:
        k += 1
        e = ''.join([v.upper() for x, v in checked_events.items() if x in line])
        watched_event = e if e else watched_event

        if ' ms,' in line and ('fetch' in line or 'mark' in line): # One of these *always* must be in trace statistics.
            print(f'Statement statistics detected for {watched_event}')
            i += 1
        if ' '.join(line.split()).upper() == ' '.join(perf_header.split()).upper():
            print('Found performance block header')
        if line.startswith('TFIX') or line.startswith('GTT_SSN') or line.startswith('GTT_TRA'):
            print(f'Found table statistics for {line.split()[0]}')
    # Check
    act.expected_stdout = expected_stdout
    act.stdout = capsys.readouterr().out
    assert act.clean_stdout == act.clean_expected_stdout