mirror of
https://github.com/FirebirdSQL/firebird-qa.git
synced 2025-01-23 14:03:06 +01:00
94 lines
3.9 KiB
Python
94 lines
3.9 KiB
Python
#coding:utf-8
|
|
|
|
"""
|
|
ID: trigger.database.trigger_on_disconnect_04
|
|
TITLE: ON DISCONNECT trigger must fire if attachment is closed by IDLE timeout.
|
|
DESCRIPTION:
|
|
Improvement was introduced 06-apr-2023, commit:
|
|
https://github.com/FirebirdSQL/firebird/commit/d64868cdcd91450e3c58aa9bfa9c1090912ac358
|
|
Test creates table 'detach_audit' which is fulfilled from ON DISCONNECT trigger if current user not SYSDBA.
|
|
We create non-privileged user ('tmp_worker') and make connection to database.
|
|
Also, trigger ON CONNECT is created which establishes session idle timeout to <SESSION_IDLE_TIMEOUT> value.
|
|
When user makes connection, it will be alive only for <SESSION_IDLE_TIMEOUT> seconds, and will be closed
|
|
after this time expired. This *must* cause firing trigger with adding record to 'detach_audit' table.
|
|
|
|
Finally, we check that table 'detach_audit' has ONE record (containing name of non-priv user).
|
|
NOTES:
|
|
[28.02.2023] pzotov
|
|
1. Confirmed missed trigger firing on 5.0.0.459 (05-apr-2023): table 'detach_audit' remained empty.
|
|
2. It seems that con_worker.is_closed() does not return True when connection is closed because of session
|
|
timeout expiration. This cause to make additional connection ('con_admin') with check content of
|
|
mon$attachments for verifying that attachment 'con_worker' was really closed (cursor must return 0 rows).
|
|
Checked on 5.0.0.961 - all OK.
|
|
"""
|
|
|
|
import pytest
|
|
from firebird.qa import *
|
|
import time
|
|
|
|
tmp_worker = user_factory('db', name='tmp_worker', password='123')
|
|
db = db_factory()
|
|
act = python_act('db')
|
|
|
|
expected_stdout_chk = """
|
|
ID 1
|
|
WHO TMP_WORKER
|
|
Records affected: 1
|
|
"""
|
|
|
|
SESSION_IDLE_TIMEOUT = 3
|
|
WORKER_IN_IDLE_STATE = SESSION_IDLE_TIMEOUT + 2
|
|
|
|
@pytest.mark.version('>=5.0')
|
|
def test_1(act: Action, tmp_worker: User):
|
|
|
|
init_sql = f"""
|
|
recreate table detach_audit(
|
|
id int generated always as identity
|
|
,dts timestamp default 'now'
|
|
,who varchar(31) default current_user
|
|
);
|
|
commit;
|
|
grant select, insert on detach_audit to public;
|
|
|
|
set term ^;
|
|
create trigger trg_connect on connect as
|
|
begin
|
|
if ( current_user != '{act.db.user}' ) then
|
|
set session idle timeout {SESSION_IDLE_TIMEOUT} second;
|
|
end
|
|
^
|
|
create trigger trg_disconnect on disconnect as
|
|
begin
|
|
if ( current_user != '{act.db.user}' ) then
|
|
execute statement 'insert into detach_audit default values';
|
|
end
|
|
^
|
|
set term ;^
|
|
commit;
|
|
"""
|
|
act.expected_stdout = ''
|
|
act.isql(switches=['-q'], input = init_sql, combine_output = True)
|
|
assert act.clean_stdout == act.clean_expected_stdout
|
|
act.reset()
|
|
|
|
worker_is_closed = False
|
|
with act.db.connect(no_db_triggers = True) as con_admin, act.db.connect(user = tmp_worker.name, password = tmp_worker.password) as con_worker:
|
|
con_worker_attachment_id = con_worker.info.id
|
|
time.sleep(WORKER_IN_IDLE_STATE)
|
|
|
|
if 1==0:
|
|
# NB! PROBLEM EXISTS HERE!
|
|
# con_worker.is_closed() returns FALSE, despite fact that connection was closed because of timeout expiration.
|
|
worker_is_closed = con_worker.is_closed()
|
|
else:
|
|
cur_admin = con_admin.cursor()
|
|
cur_admin.execute(f'select count(*) from mon$attachments where mon$attachment_id = {con_worker_attachment_id}')
|
|
worker_is_closed = True if cur_admin.fetchone()[0] == 0 else False
|
|
|
|
assert worker_is_closed, f'### FAIL ### Worker attachment was not closed for {WORKER_IN_IDLE_STATE} second.'
|
|
|
|
act.expected_stdout = expected_stdout_chk
|
|
act.isql(switches=['-q'], input = 'set count on;set list on;select id,who from detach_audit;', combine_output = True)
|
|
assert act.clean_stdout == act.clean_expected_stdout
|