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_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_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()