def test_close(self):
reader = Mock(channel=Mock())
tag = 1234
# test
r = Receiver(reader)
r.tag = tag
r.close()
# validation
reader.channel.basic_cancel.assert_called_once_with(tag)
def test_fetch_empty(self):
channel = Mock()
reader = Mock(channel=channel)
# test
r = Receiver(reader)
r.inbox = Mock()
r.inbox.empty.return_value = False
r.inbox.get.side_effect = Empty
r._wait = Mock(side_effect=r.inbox.put)
self.assertRaises(r.fetch)
def test_close_exception(self):
reader = Mock()
reader.channel.basic_cancel.side_effect = ValueError
tag = 1234
# test
r = Receiver(reader)
r.tag = tag
r.close()
# validation
reader.channel.basic_cancel.assert_called_once_with(tag)
def test_open(self):
node = Mock(address='test')
reader = Mock(node=node, channel=Mock())
# test
r = Receiver(reader)
r = r.open()
# validation
reader.channel.basic_consume.assert_called_once_with(
node.address, callback=r.inbox.put)
self.assertEqual(r.tag, reader.channel.basic_consume.return_value)
def test_fetch(self):
timeout = 10
channel = Mock()
reader = Mock(channel=channel)
received = 33
# test
r = Receiver(reader)
r.inbox = Mock()
r.inbox.empty.return_value = True
r.inbox.get.return_value = received
r._wait = Mock(side_effect=r.inbox.put)
message = r.fetch(timeout)
# validation
fd = channel.connection.sock.fileno.return_value
r._wait.assert_called_once_with(fd, channel, timeout)
self.assertEqual(message, received)
def test_channel(self):
reader = Mock(channel=Mock())
r = Receiver(reader)
channel = r.channel()
self.assertEqual(channel, reader.channel)
def test_init(self):
reader = Mock()
r = Receiver(reader)
self.assertEqual(r.reader, reader)
self.assertEqual(r.tag, None)
self.assertTrue(isinstance(r.inbox, Inbox))
