def test_open(self):
url = 'test-url'
node = Node('test-queue')
consumer = ConsumerThread(node, url)
consumer.reader = Mock()
# test
consumer.open()
# validation
consumer.reader.open.assert_called_once_with()
def test_open(self):
url = 'test-url'
node = Node('test-queue')
consumer = ConsumerThread(node, url)
consumer.reader = Mock()
# test
consumer.open()
# validation
consumer.reader.open.assert_called_once_with()
def test_open_exception(self, sleep):
url = 'test-url'
node = Node('test-queue')
consumer = ConsumerThread(node, url)
consumer.reader = Mock()
consumer.reader.open.side_effect = [ValueError, None]
# test
consumer.open()
# validation
sleep.assert_called_once_with(30)
self.assertEqual(consumer.reader.open.call_count, 2)
def test_open_exception(self, sleep):
url = 'test-url'
node = Node('test-queue')
consumer = ConsumerThread(node, url)
consumer.reader = Mock()
consumer.reader.open.side_effect = [ValueError, None]
# test
consumer.open()
# validation
sleep.assert_called_once_with(30)
self.assertEqual(consumer.reader.open.call_count, 2)
def test_open_not_found_raised(self, sleep):
url = 'test-url'
node = Node('test-queue')
consumer = ConsumerThread(node, url)
consumer.reader = Mock()
consumer.reader.open.side_effect = [NotFound, None]
consumer.no_route = Mock()
# test
consumer.open()
# validation
sleep.assert_called_once_with(10)
consumer.no_route.assert_called_once_with()
self.assertEqual(consumer.reader.open.call_count, 2)
def test_repair(self):
url = 'test-url'
node = Node('test-queue')
consumer = ConsumerThread(node, url)
consumer.open = Mock()
consumer.close = Mock()
# Test
consumer.repair()
# Validation
consumer.close.assert_called_once_with()
consumer.open.assert_called_once_with()
def test_read_exception(self, sleep):
url = 'test-url'
node = Node('test-queue')
consumer = ConsumerThread(node, url)
consumer.reader = Mock()
consumer.reader.next.side_effect = IndexError
consumer.open = Mock()
consumer.close = Mock()
# test
consumer.read()
# validation
consumer.close.assert_called_once_with()
consumer.open.assert_called_once_with()
sleep.assert_called_once_with(60)
def test_read_exception(self, sleep):
url = 'test-url'
node = Node('test-queue')
consumer = ConsumerThread(node, url)
consumer.reader = Mock()
consumer.reader.next.side_effect = IndexError
consumer.open = Mock()
consumer.close = Mock()
# test
consumer.read()
# validation
consumer.close.assert_called_once_with()
consumer.open.assert_called_once_with()
sleep.assert_called_once_with(60)
def test_read_not_found(self, sleep):
url = 'test-url'
node = Node('test-queue')
consumer = ConsumerThread(node, url)
consumer.reader = Mock()
consumer.reader.next.side_effect = NotFound
consumer.open = Mock()
consumer.close = Mock()
# test
consumer.read()
# validation
consumer.close.assert_called_once_with()
consumer.open.assert_called_once_with()
sleep.assert_called_once_with(10)
def test_run(self, reader):
url = 'test-url'
node = Node('test-queue')
consumer = ConsumerThread(node, url)
consumer.open = Mock()
consumer.close = Mock()
consumer.read = Mock(side_effect=StopIteration)
# test
try:
consumer.run()
except StopIteration:
pass
# validation
reader.assert_called_once_with(node, url)
consumer.open.assert_called_once_with()
consumer.read.assert_called_once_with()
consumer.close.assert_called_once_with()
def test_run(self, reader):
url = 'test-url'
node = Node('test-queue')
consumer = ConsumerThread(node, url)
consumer.open = Mock()
consumer.close = Mock()
consumer.read = Mock(side_effect=StopIteration)
# test
try:
consumer.run()
except StopIteration:
pass
# validation
reader.assert_called_once_with(node, url)
consumer.open.assert_called_once_with()
consumer.read.assert_called_once_with()
consumer.close.assert_called_once_with()
