class TestNotify(TransactionTestCase):
def setUp(self):
self.q = Queue()
def test_notify(self):
"""Postgresql NOTIFY on channel with default connection"""
self.q.notify(1)
class TestListenForJobs(TransactionTestCase):
def setUp(self):
self.q = Queue()
# pre-call this so we don't need to use multi-process
# otherwise this is called within the classmethod
PQ.listen('default', ['default'])
# Fire off a notification of a fake job enqueued
self.q.notify(1)
def test_listen_for_jobs(self):
"""Test the first part of the _listen_for_jobs method which polls
for notifications"""
queue_name = PQ._listen_for_jobs(['default'], 'default', 1)
self.assertEqual('default', queue_name)
