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)