Files
clang-p2996/lldb/test/API/functionalities/data-formatter/callback-matching/TestDataFormatterCallbackMatching.py
Jonas Devlieghere 2238dcc393 [NFC][Py Reformat] Reformat python files in lldb
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
2023-05-25 12:54:09 -07:00

100 lines
3.4 KiB
Python

"""
Test lldb data formatter callback-based matching.
"""
import lldb
from lldbsuite.test.decorators import *
from lldbsuite.test.lldbtest import *
from lldbsuite.test import lldbutil
class PythonSynthDataFormatterTestCase(TestBase):
def setUp(self):
# Call super's setUp().
TestBase.setUp(self)
# Find the line number to break at.
self.line = line_number("main.cpp", "// Set break point at this line.")
def test_callback_matchers_api_registration(self):
"""Test data formatter commands."""
self.build()
_, process, thread, _ = lldbutil.run_to_line_breakpoint(
self, lldb.SBFileSpec("main.cpp"), self.line
)
# Print derived without a formatter.
self.expect("frame variable derived", substrs=["x = 2222", "y = 3333"])
# now set up a summary function that uses a python callback to match
# classes that derive from `Base`.
self.runCmd(
"command script import --allow-reload ./formatters_with_callback.py"
)
self.runCmd(
"script formatters_with_callback.register_formatters(lldb.debugger)"
)
# Now `derived` should use our callback summary + synthetic children.
self.expect(
"frame variable derived",
substrs=["hello from callback summary", "synthetic_child = 9999"],
)
# But not other classes.
self.expect(
"frame variable base",
matching=False,
substrs=["hello from callback summary"],
)
self.expect("frame variable base", substrs=["x = 1111"])
self.expect(
"frame variable nd", matching=False, substrs=["hello from callback summary"]
)
self.expect("frame variable nd", substrs=["z = 4444"])
def test_callback_matchers_cli_registration(self):
"""Test data formatter commands."""
self.build()
_, process, thread, _ = lldbutil.run_to_line_breakpoint(
self, lldb.SBFileSpec("main.cpp"), self.line
)
# Print derived without a formatter.
self.expect("frame variable derived", substrs=["x = 2222", "y = 3333"])
# now set up a summary function that uses a python callback to match
# classes that derive from `Base`.
self.runCmd(
"command script import --allow-reload ./formatters_with_callback.py"
)
self.runCmd(
"type summary add -e -s 'hello from callback summary' "
"--recognizer-function formatters_with_callback.derives_from_base"
)
self.runCmd(
"type synth add -l formatters_with_callback.SynthProvider "
"--recognizer-function formatters_with_callback.derives_from_base"
)
# Now `derived` should use our callback summary + synthetic children.
self.expect(
"frame variable derived",
substrs=["hello from callback summary", "synthetic_child = 9999"],
)
# But not other classes.
self.expect(
"frame variable base",
matching=False,
substrs=["hello from callback summary"],
)
self.expect("frame variable base", substrs=["x = 1111"])
self.expect(
"frame variable nd", matching=False, substrs=["hello from callback summary"]
)
self.expect("frame variable nd", substrs=["z = 4444"])