summaryrefslogtreecommitdiff
path: root/tests/test_notify.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_notify.py')
-rwxr-xr-xtests/test_notify.py22
1 files changed, 22 insertions, 0 deletions
diff --git a/tests/test_notify.py b/tests/test_notify.py
index f838389..fc6224d 100755
--- a/tests/test_notify.py
+++ b/tests/test_notify.py
@@ -155,6 +155,27 @@ conn.close()
self.assertEqual('foo', notify.channel)
self.assertEqual('Hello, world!', notify.payload)
+ def test_notify_deque(self):
+ from collections import deque
+ self.autocommit(self.conn)
+ self.conn.notifies = deque()
+ self.listen('foo')
+ self.notify('foo').communicate()
+ time.sleep(0.5)
+ self.conn.poll()
+ notify = self.conn.notifies.popleft()
+ self.assert_(isinstance(notify, psycopg2.extensions.Notify))
+ self.assertEqual(len(self.conn.notifies), 0)
+
+ def test_notify_noappend(self):
+ self.autocommit(self.conn)
+ self.conn.notifies = None
+ self.listen('foo')
+ self.notify('foo').communicate()
+ time.sleep(0.5)
+ self.conn.poll()
+ self.assertEqual(self.conn.notifies, None)
+
def test_notify_init(self):
n = psycopg2.extensions.Notify(10, 'foo')
self.assertEqual(10, n.pid)
@@ -192,6 +213,7 @@ conn.close()
self.assertNotEqual(hash(Notify(10, 'foo', 'bar')),
hash(Notify(10, 'foo')))
+
def test_suite():
return unittest.TestLoader().loadTestsFromName(__name__)