def test_listen_for_jobs_select(self):
"""Test the 2nd part of the _listen_for_jobs method which
blocks and waits for postgresql to notify it"""
queue_name = PQ._listen_for_jobs(['default', 'farq'], 'default', 5)
self.assertEqual('farq', queue_name)
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)
