File: TestDataFormatterCallbackMatching.py

package info (click to toggle)
llvm-toolchain-19 1%3A19.1.7-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, trixie
  • size: 1,998,520 kB
  • sloc: cpp: 6,951,680; ansic: 1,486,157; asm: 913,598; python: 232,024; f90: 80,126; objc: 75,281; lisp: 37,276; pascal: 16,990; sh: 10,009; ml: 5,058; perl: 4,724; awk: 3,523; makefile: 3,167; javascript: 2,504; xml: 892; fortran: 664; cs: 573
file content (99 lines) | stat: -rw-r--r-- 3,455 bytes parent folder | download | duplicates (10)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
"""
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"])