firebird-qa/tests/bugs/gh_7979_test.py

#coding:utf-8

"""
ID:          issue-7979
ISSUE:       Hang when database with disconnect trigger using MON$ tables is shutting down
TITLE:       
DESCRIPTION:
NOTES:
    [02.02.2024] pzotov
    ### ACHTUNG ###
    Bug could NOT be reproduced on Windows.
    Bug can be reproduced only when appropriate code runs OUTSIDE current firebird-QA framework, i.e. directly from OS.
    Because of that, this test creates temporary .py script which is launched further using subprocess.run( [ sys.executable ...] )
    Confirmed problem on 6.0.0.219: 'gfix -shut full -force 0' hangs.
    Checked on 6.0.0.244.
"""
import sys
import subprocess
from pathlib import Path
import pytest
from firebird.qa import *

init_sql = f"""
    create table logger (dts timestamp default 'now', att_cnt int);
    set term ^;
    execute block as
    begin
        rdb$set_context('USER_SESSION', 'INITIAL_DDL', 1);
    end
    ^
    create trigger logger active on disconnect as
        declare c int;
    begin
        if ( rdb$get_context('USER_SESSION', 'INITIAL_DDL') is null ) then
        begin
            select count(*) from mon$attachments where mon$attachment_id = current_connection into :c;
            insert into logger(att_cnt) values(:c);
        end
    end
    ^
    set term ;^
    commit;
"""

db = db_factory(init = init_sql)
act = python_act('db', substitutions = [ ('^((?!(Attributes|ATT_CNT|Records affected)).)*$', ''), ('[ \t]+', ' ') ])
tmp_run_py = temp_file('tmp_7979_run_external.py')
tmp_log_py = temp_file('tmp_7979_run_external.log')
tmp_sql_py = temp_file('tmp_7979_check_result.sql')

@pytest.mark.version('>=6.0')
def test_1(act: Action, tmp_run_py: Path, tmp_log_py: Path, tmp_sql_py: Path, capsys):
    if act.platform == 'Windows':
        pytest.skip('Could not reproduce bug on Windows')
    if act.get_server_architecture() != 'SuperServer':
        pytest.skip('Applies only to SuperServer')

    py_run_ext = ' '.join( [ sys.executable, '-u', f'{str(tmp_run_py)}'] )
    py_source = f"""# -*- coding: utf-8 -*-
# {py_run_ext}
import os
import sys
import argparse as ap
from pathlib import Path
import subprocess
import datetime as py_dt

import firebird.driver
from firebird.driver import *

os.environ["ISC_USER"] = '{act.db.user}'
os.environ["ISC_PASSWORD"] = '{act.db.password}'
driver_config.fb_client_library.value = "{act.vars['fbclient']}"

with connect_server('localhost:service_mgr') as srv:
    bin_isql="{act.vars['isql']}"
    bin_gfix="{act.vars['gfix']}"
    bin_gstat="{act.vars['gstat']}"
    
with connect('{act.db.dsn}') as con:
    #print(f"Trying to run gfix -shut full -force 0 '{act.db.dsn}'")
    subprocess.run( [bin_gfix, '-shut', 'full', '-force', '0', '{act.db.dsn}'] )
    subprocess.run( [bin_gstat, '-h', '{act.db.db_path}'] )
    #print(f"Trying to run gfix -online '{act.db.dsn}'")
    subprocess.run( [bin_gfix, '-online', '{act.db.dsn}'] )
    subprocess.run( [bin_gstat, '-h', '{act.db.db_path}'] )

chk_sql='''
    set list on;
    set count on;
    select att_cnt from logger;
'''
with open('{str(tmp_sql_py)}', 'w') as f:
    f.write(chk_sql)

#subprocess.run( [bin_isql, '-q', '-i', '{str(tmp_sql_py)}', '-user', os.environ["ISC_USER"], '-pas',  os.environ["ISC_PASSWORD"], '{act.db.dsn}'] )
subprocess.run( [bin_isql, '-q', '-i', '{str(tmp_sql_py)}', '{act.db.dsn}'] )
"""

    tmp_run_py.write_text(py_source)
    with open(tmp_log_py, 'w') as f:
        subprocess.run( [ sys.executable, '-u', f'{str(tmp_run_py)}'], stdout = f, stderr = subprocess.STDOUT )
    
    with open(tmp_log_py, 'r') as f:
        print(f.read())
    act.expected_stdout = """
	Attributes full shutdown
	Attributes
	ATT_CNT 1
	Records affected: 1
    """
    act.stdout = capsys.readouterr().out
    assert act.clean_stdout == act.clean_expected_stdout
    act.reset()