1"""Test calling functions in class methods."""
2
3from __future__ import print_function
4
5
6import lldb
7from lldbsuite.test.decorators import *
8from lldbsuite.test.lldbtest import *
9from lldbsuite.test import lldbutil
10
11
12class TestObjCClassMethod(TestBase):
13
14    mydir = TestBase.compute_mydir(__file__)
15
16    def setUp(self):
17        # Call super's setUp().
18        TestBase.setUp(self)
19        # Find the line numbers to break inside main().
20        self.main_source = "class.m"
21        self.break_line = line_number(
22            self.main_source, '// Set breakpoint here.')
23
24    @add_test_categories(['pyapi'])
25    def test_with_python_api(self):
26        """Test calling functions in class methods."""
27        self.build()
28        exe = self.getBuildArtifact("a.out")
29
30        target = self.dbg.CreateTarget(exe)
31        self.assertTrue(target, VALID_TARGET)
32
33        bpt = target.BreakpointCreateByLocation(
34            self.main_source, self.break_line)
35        self.assertTrue(bpt, VALID_BREAKPOINT)
36
37        # Now launch the process, and do not stop at entry point.
38        process = target.LaunchSimple(
39            None, None, self.get_process_working_directory())
40
41        self.assertTrue(process, PROCESS_IS_VALID)
42
43        # The stop reason of the thread should be breakpoint.
44        thread_list = lldbutil.get_threads_stopped_at_breakpoint(process, bpt)
45
46        # Make sure we stopped at the first breakpoint.
47        self.assertTrue(
48            len(thread_list) != 0,
49            "No thread stopped at our breakpoint.")
50        self.assertTrue(len(thread_list) == 1,
51                        "More than one thread stopped at our breakpoint.")
52
53        # Now make sure we can call a function in the class method we've
54        # stopped in.
55        frame = thread_list[0].GetFrameAtIndex(0)
56        self.assertTrue(frame, "Got a valid frame 0 frame.")
57
58        cmd_value = frame.EvaluateExpression(
59            "(int)[Foo doSomethingWithString:@\"Hello\"]")
60        if self.TraceOn():
61            if cmd_value.IsValid():
62                print("cmd_value is valid")
63                print("cmd_value has the value %d" % cmd_value.GetValueAsUnsigned())
64        self.assertTrue(cmd_value.IsValid())
65        self.assertTrue(cmd_value.GetValueAsUnsigned() == 5)
66