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
128 lines
4.5 KiB
Python
128 lines
4.5 KiB
Python
"""
|
|
Test number of threads.
|
|
"""
|
|
|
|
|
|
import lldb
|
|
from lldbsuite.test.decorators import *
|
|
from lldbsuite.test.lldbtest import *
|
|
import lldbsuite.test.lldbutil as lldbutil
|
|
|
|
|
|
class ThreadExitTestCase(TestBase):
|
|
def setUp(self):
|
|
# Call super's setUp().
|
|
TestBase.setUp(self)
|
|
# Find the line numbers for our breakpoints.
|
|
self.break_1 = line_number("main.cpp", "// Set first breakpoint here")
|
|
self.break_2 = line_number("main.cpp", "// Set second breakpoint here")
|
|
self.break_3 = line_number("main.cpp", "// Set third breakpoint here")
|
|
self.break_4 = line_number("main.cpp", "// Set fourth breakpoint here")
|
|
|
|
@skipIfWindows # This is flakey on Windows: llvm.org/pr38373
|
|
def test(self):
|
|
"""Test thread exit handling."""
|
|
self.build()
|
|
exe = self.getBuildArtifact("a.out")
|
|
self.runCmd("file " + exe, CURRENT_EXECUTABLE_SET)
|
|
|
|
# This should create a breakpoint with 1 location.
|
|
bp1_id = lldbutil.run_break_set_by_file_and_line(
|
|
self, "main.cpp", self.break_1, num_expected_locations=1
|
|
)
|
|
bp2_id = lldbutil.run_break_set_by_file_and_line(
|
|
self, "main.cpp", self.break_2, num_expected_locations=1
|
|
)
|
|
bp3_id = lldbutil.run_break_set_by_file_and_line(
|
|
self, "main.cpp", self.break_3, num_expected_locations=1
|
|
)
|
|
bp4_id = lldbutil.run_break_set_by_file_and_line(
|
|
self, "main.cpp", self.break_4, num_expected_locations=1
|
|
)
|
|
|
|
# The breakpoint list should show 1 locations.
|
|
self.expect(
|
|
"breakpoint list -f",
|
|
"Breakpoint location shown correctly",
|
|
substrs=[
|
|
"1: file = 'main.cpp', line = %d, exact_match = 0, locations = 1"
|
|
% self.break_1,
|
|
"2: file = 'main.cpp', line = %d, exact_match = 0, locations = 1"
|
|
% self.break_2,
|
|
"3: file = 'main.cpp', line = %d, exact_match = 0, locations = 1"
|
|
% self.break_3,
|
|
"4: file = 'main.cpp', line = %d, exact_match = 0, locations = 1"
|
|
% self.break_4,
|
|
],
|
|
)
|
|
|
|
# Run the program.
|
|
self.runCmd("run", RUN_SUCCEEDED)
|
|
# Get the target process
|
|
target = self.dbg.GetSelectedTarget()
|
|
process = target.GetProcess()
|
|
|
|
stopped_thread = lldbutil.get_one_thread_stopped_at_breakpoint_id(
|
|
process, bp1_id
|
|
)
|
|
self.assertIsNotNone(stopped_thread, "Process is not stopped at breakpoint 1")
|
|
|
|
# Get the number of threads
|
|
num_threads = process.GetNumThreads()
|
|
self.assertGreaterEqual(
|
|
num_threads,
|
|
2,
|
|
"Number of expected threads and actual threads do not match at breakpoint 1.",
|
|
)
|
|
|
|
# Run to the second breakpoint
|
|
self.runCmd("continue")
|
|
stopped_thread = lldbutil.get_one_thread_stopped_at_breakpoint_id(
|
|
process, bp2_id
|
|
)
|
|
self.assertIsNotNone(stopped_thread, "Process is not stopped at breakpoint 2")
|
|
|
|
# Update the number of threads
|
|
new_num_threads = process.GetNumThreads()
|
|
self.assertEqual(
|
|
new_num_threads,
|
|
num_threads + 1,
|
|
"Number of expected threads did not increase by 1 at bp 2.",
|
|
)
|
|
|
|
# Run to the third breakpoint
|
|
self.runCmd("continue")
|
|
stopped_thread = lldbutil.get_one_thread_stopped_at_breakpoint_id(
|
|
process, bp3_id
|
|
)
|
|
self.assertIsNotNone(stopped_thread, "Process is not stopped at breakpoint 3")
|
|
|
|
# Update the number of threads
|
|
new_num_threads = process.GetNumThreads()
|
|
self.assertEqual(
|
|
new_num_threads,
|
|
num_threads,
|
|
"Number of expected threads is not equal to original number of threads at bp 3.",
|
|
)
|
|
|
|
# Run to the fourth breakpoint
|
|
self.runCmd("continue")
|
|
stopped_thread = lldbutil.get_one_thread_stopped_at_breakpoint_id(
|
|
process, bp4_id
|
|
)
|
|
self.assertIsNotNone(stopped_thread, "Process is not stopped at breakpoint 4")
|
|
|
|
# Update the number of threads
|
|
new_num_threads = process.GetNumThreads()
|
|
self.assertEqual(
|
|
new_num_threads,
|
|
num_threads - 1,
|
|
"Number of expected threads did not decrease by 1 at bp 4.",
|
|
)
|
|
|
|
# Run to completion
|
|
self.runCmd("continue")
|
|
|
|
# At this point, the inferior process should have exited.
|
|
self.assertState(process.GetState(), lldb.eStateExited, PROCESS_EXITED)
|