psycopg/psycopg2
1
1
Fork 0
mirror of https://github.com/psycopg/psycopg2.git synced 2024-11-14 13:06:34 +03:00
Code Issues Packages Projects Releases Wiki Activity

Adds notifies processing in pq_commit

Browse Source
This commit is contained in:
Roman Konoval 2024-09-10 12:11:39 +02:00 committed by Daniele Varrazzo
parent eaeeb76944
commit 362cb00978
2 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
psycopg/pqpath.c
View File

@ -412,6 +412,7 @@ pq_commit(connectionObject *conn)
} }
Py_BLOCK_THREADS; Py_BLOCK_THREADS;
conn_notifies_process(conn);
conn_notice_process(conn); conn_notice_process(conn);
Py_UNBLOCK_THREADS; Py_UNBLOCK_THREADS;

12
tests/test_notify.py
View File

@ -126,6 +126,18 @@ conn.close()
self.assertEqual(pid, self.conn.notifies[0][0]) self.assertEqual(pid, self.conn.notifies[0][0])
self.assertEqual('foo', self.conn.notifies[0][1]) self.assertEqual('foo', self.conn.notifies[0][1])
@slow
def test_notifies_received_on_commit(self):
self.listen("foo")
self.conn.commit()
self.conn.cursor().execute("select 1;")
pid = int(self.notify("foo").communicate()[0])
self.assertEqual(0, len(self.conn.notifies))
self.conn.commit()
self.assertEqual(1, len(self.conn.notifies))
self.assertEqual(pid, self.conn.notifies[0][0])
self.assertEqual("foo", self.conn.notifies[0][1])
@slow @slow
def test_notify_object(self): def test_notify_object(self):
self.autocommit(self.conn) self.autocommit(self.conn)