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