class TestReceiveMessage(unittest.TestCase):
def setUp(self):
self.sender = Attribute()
def test_send(self):
self.sender.send()
def test_receive(self):
message = self.mock_receiver()
self.assertEquals(message, 'Hi')
def mock_receiver(self):
connection = pika.BlockingConnection(pika.ConnectionParameters(
host='localhost'))
channel = connection.channel()
method_frame, header_frame, body = channel.basic_get('task_queue')
if method_frame == None:
raise Exception('Queue was empty')
channel.basic_ack(delivery_tag=method_frame.delivery_tag)
channel.close()
connection.close()
return body
