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_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_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_wait_with_queued(self, epoll): fd = 0 channel = Mock(method_queue=[Mock()]) timeout = 10 epoll.return_value.poll.return_value = [fd] # test Receiver._wait(fd, channel, timeout) # validation self.assertFalse(epoll.called) channel.wait.assert_called_once_with()
def test_wait(self, epoll): fd = 0 channel = Mock(method_queue=[Mock()]) timeout = 10 epoll.return_value.poll.return_value = [fd] # test Receiver._wait(fd, channel, timeout) # validation self.assertFalse(epoll.called) channel.wait.assert_called_once_with()
def test_wait_nothing(self, epoll): fd = 0 channel = Mock(method_queue=[]) timeout = 10 epoll.return_value.poll.return_value = [] # test Receiver._wait(fd, channel, timeout) # validation epoll.assert_called_once_with() epoll.return_value.register.assert_called_with(fd, select.EPOLLIN) epoll.return_value.poll.assert_called_with(timeout) self.assertFalse(channel.wait.called)
def test_wait(self, basic, epoll): fd = 0 channel = Mock(method_queue=[]) timeout = 10 epoll.return_value.poll.return_value = [fd] basic.Deliver = (10, 70) # test Receiver._wait(fd, channel, timeout) # validation epoll.assert_called_once_with() epoll.return_value.register.assert_called_with(fd, select.EPOLLIN) epoll.return_value.poll.assert_called_with(timeout) channel.wait.assert_called_once_with(basic.Deliver)
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))