d16e111b53
It is often quite useful to try to see what the contained causes are on versions of python that do not have the native support for this built-in so to make everyones life easier add basic support for traversing the cause list and printing out associated causes when we are able to. Change-Id: Ia0a7e13757a989722291bcc06599d04014706d8c
113 lines
3.9 KiB
Python
113 lines
3.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright (C) 2013 Yahoo! Inc. All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import six
|
|
import testtools
|
|
|
|
from taskflow import exceptions as exc
|
|
from taskflow import test
|
|
|
|
|
|
class TestExceptions(test.TestCase):
|
|
def test_cause(self):
|
|
capture = None
|
|
try:
|
|
raise exc.TaskFlowException("broken", cause=IOError("dead"))
|
|
except Exception as e:
|
|
capture = e
|
|
self.assertIsNotNone(capture)
|
|
self.assertIsInstance(capture, exc.TaskFlowException)
|
|
self.assertIsNotNone(capture.cause)
|
|
self.assertIsInstance(capture.cause, IOError)
|
|
|
|
def test_cause_pformat(self):
|
|
capture = None
|
|
try:
|
|
raise exc.TaskFlowException("broken", cause=IOError("dead"))
|
|
except Exception as e:
|
|
capture = e
|
|
self.assertIsNotNone(capture)
|
|
self.assertGreater(0, len(capture.pformat()))
|
|
|
|
def test_raise_with(self):
|
|
capture = None
|
|
try:
|
|
raise IOError('broken')
|
|
except Exception:
|
|
try:
|
|
exc.raise_with_cause(exc.TaskFlowException, 'broken')
|
|
except Exception as e:
|
|
capture = e
|
|
self.assertIsNotNone(capture)
|
|
self.assertIsInstance(capture, exc.TaskFlowException)
|
|
self.assertIsNotNone(capture.cause)
|
|
self.assertIsInstance(capture.cause, IOError)
|
|
|
|
def test_pformat_str(self):
|
|
ex = None
|
|
try:
|
|
try:
|
|
try:
|
|
raise IOError("Didn't work")
|
|
except IOError:
|
|
exc.raise_with_cause(exc.TaskFlowException,
|
|
"It didn't go so well")
|
|
except exc.TaskFlowException:
|
|
exc.raise_with_cause(exc.TaskFlowException, "I Failed")
|
|
except exc.TaskFlowException as e:
|
|
ex = e
|
|
|
|
self.assertIsNotNone(ex)
|
|
self.assertIsInstance(ex, exc.TaskFlowException)
|
|
self.assertIsInstance(ex.cause, exc.TaskFlowException)
|
|
self.assertIsInstance(ex.cause.cause, IOError)
|
|
|
|
p_msg = ex.pformat()
|
|
p_str_msg = str(ex)
|
|
for msg in ["I Failed", "It didn't go so well", "Didn't work"]:
|
|
self.assertIn(msg, p_msg)
|
|
self.assertIn(msg, p_str_msg)
|
|
|
|
def test_pformat_root_class(self):
|
|
ex = exc.TaskFlowException("Broken")
|
|
self.assertIn("TaskFlowException",
|
|
ex.pformat(show_root_class=True))
|
|
self.assertNotIn("TaskFlowException",
|
|
ex.pformat(show_root_class=False))
|
|
self.assertIn("Broken",
|
|
ex.pformat(show_root_class=True))
|
|
|
|
def test_invalid_pformat_indent(self):
|
|
ex = exc.TaskFlowException("Broken")
|
|
self.assertRaises(ValueError, ex.pformat, indent=-100)
|
|
|
|
@testtools.skipIf(not six.PY3, 'py3.x is not available')
|
|
def test_raise_with_cause(self):
|
|
capture = None
|
|
try:
|
|
raise IOError('broken')
|
|
except Exception:
|
|
try:
|
|
exc.raise_with_cause(exc.TaskFlowException, 'broken')
|
|
except Exception as e:
|
|
capture = e
|
|
self.assertIsNotNone(capture)
|
|
self.assertIsInstance(capture, exc.TaskFlowException)
|
|
self.assertIsNotNone(capture.cause)
|
|
self.assertIsInstance(capture.cause, IOError)
|
|
self.assertIsNotNone(capture.__cause__)
|
|
self.assertIsInstance(capture.__cause__, IOError)
|