2238dcc393
This is an ongoing series of commits that are reformatting our Python code. Reformatting is done with `black` (23.1.0). If you end up having problems merging this commit because you have made changes to a python file, the best way to handle that is to run `git checkout --ours <yourfile>` and then reformat it with black. RFC: https://discourse.llvm.org/t/rfc-document-and-standardize-python-code-style Differential revision: https://reviews.llvm.org/D151460
100 lines
3.6 KiB
Python
100 lines
3.6 KiB
Python
"""
|
|
Test calling a function that throws an ObjC exception, make sure that it doesn't propagate the exception.
|
|
"""
|
|
|
|
|
|
import lldb
|
|
from lldbsuite.test.decorators import *
|
|
from lldbsuite.test.lldbtest import *
|
|
from lldbsuite.test import lldbutil
|
|
|
|
|
|
class ExprCommandWithThrowTestCase(TestBase):
|
|
def setUp(self):
|
|
# Call super's setUp().
|
|
TestBase.setUp(self)
|
|
|
|
self.main_source = "call-throws.m"
|
|
self.main_source_spec = lldb.SBFileSpec(self.main_source)
|
|
|
|
@add_test_categories(["objc"])
|
|
def test(self):
|
|
"""Test calling a function that throws and ObjC exception."""
|
|
self.build()
|
|
self.call_function()
|
|
|
|
def check_after_call(self):
|
|
# Check that we are back where we were before:
|
|
frame = self.thread.GetFrameAtIndex(0)
|
|
self.assertEqual(
|
|
self.orig_frame_pc, frame.GetPC(), "Restored the zeroth frame correctly"
|
|
)
|
|
|
|
def call_function(self):
|
|
"""Test calling function that throws."""
|
|
(target, process, self.thread, bkpt) = lldbutil.run_to_source_breakpoint(
|
|
self, "I am about to throw.", self.main_source_spec
|
|
)
|
|
|
|
options = lldb.SBExpressionOptions()
|
|
options.SetUnwindOnError(True)
|
|
|
|
frame = self.thread.GetFrameAtIndex(0)
|
|
# Store away the PC to check that the functions unwind to the right
|
|
# place after calls
|
|
self.orig_frame_pc = frame.GetPC()
|
|
|
|
value = frame.EvaluateExpression("[my_class callMeIThrow]", options)
|
|
self.assertTrue(value.IsValid())
|
|
self.assertEquals(value.GetError().Success(), False)
|
|
|
|
self.check_after_call()
|
|
|
|
# Okay, now try with a breakpoint in the called code in the case where
|
|
# we are ignoring breakpoint hits.
|
|
handler_bkpt = target.BreakpointCreateBySourceRegex(
|
|
"I felt like it", self.main_source_spec
|
|
)
|
|
self.assertTrue(handler_bkpt.GetNumLocations() > 0)
|
|
options.SetIgnoreBreakpoints(True)
|
|
options.SetUnwindOnError(True)
|
|
|
|
value = frame.EvaluateExpression("[my_class callMeIThrow]", options)
|
|
|
|
self.assertTrue(value.IsValid() and value.GetError().Success() == False)
|
|
self.check_after_call()
|
|
|
|
# Now set the ObjC language breakpoint and make sure that doesn't
|
|
# interfere with the call:
|
|
exception_bkpt = target.BreakpointCreateForException(
|
|
lldb.eLanguageTypeObjC, False, True
|
|
)
|
|
self.assertTrue(exception_bkpt.GetNumLocations() > 0)
|
|
|
|
options.SetIgnoreBreakpoints(True)
|
|
options.SetUnwindOnError(True)
|
|
|
|
value = frame.EvaluateExpression("[my_class callMeIThrow]", options)
|
|
|
|
self.assertTrue(value.IsValid() and value.GetError().Success() == False)
|
|
self.check_after_call()
|
|
|
|
# Now turn off exception trapping, and call a function that catches the exceptions,
|
|
# and make sure the function actually completes, and we get the right
|
|
# value:
|
|
options.SetTrapExceptions(False)
|
|
value = frame.EvaluateExpression("[my_class iCatchMyself]", options)
|
|
self.assertTrue(value.IsValid())
|
|
self.assertSuccess(value.GetError())
|
|
self.assertEquals(value.GetValueAsUnsigned(), 57)
|
|
self.check_after_call()
|
|
options.SetTrapExceptions(True)
|
|
|
|
# Now set this unwind on error to false, and make sure that we stop
|
|
# where the exception was thrown
|
|
options.SetUnwindOnError(False)
|
|
value = frame.EvaluateExpression("[my_class callMeIThrow]", options)
|
|
|
|
self.assertTrue(value.IsValid() and value.GetError().Success() == False)
|
|
self.check_after_call()
|