def test_close(self):
url = 'test-url'
node = Node('test-queue')
consumer = ConsumerThread(node, url)
consumer._reader = Mock()
# test
consumer._close()
# validation
consumer._reader.close.assert_called_once_with()
def test_close(self):
url = 'test-url'
node = Node('test-queue')
consumer = ConsumerThread(node, url)
consumer._reader = Mock()
# test
consumer._close()
# validation
consumer._reader.close.assert_called_once_with()
def test_close_exception(self):
url = 'test-url'
node = Node('test-queue')
consumer = ConsumerThread(node, url)
consumer._reader = Mock()
consumer._reader.close.side_effect = ValueError
# test
consumer._close()
# validation
consumer._reader.close.assert_called_once_with()
def test_close_exception(self):
url = 'test-url'
node = Node('test-queue')
consumer = ConsumerThread(node, url)
consumer._reader = Mock()
consumer._reader.close.side_effect = ValueError
# test
consumer._close()
# validation
consumer._reader.close.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_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()
