1"""
2Test lldb data formatter subsystem.
3"""
4
5import os, time
6import unittest2
7import lldb
8from lldbtest import *
9import datetime
10import lldbutil
11
12class DataFormatterBoolRefPtr(TestBase):
13
14    mydir = os.path.join("functionalities", "data-formatter", "rdar-11773899")
15
16    @unittest2.skipUnless(sys.platform.startswith("darwin"), "requires Darwin")
17    @dsym_test
18    def test_boolrefptr_with_dsym_and_run_command(self):
19        """Test the formatters we use for BOOL& and BOOL* in Objective-C."""
20        self.buildDsym()
21        self.boolrefptr_data_formatter_commands()
22
23    @unittest2.skipUnless(sys.platform.startswith("darwin"), "requires Darwin")
24    @dwarf_test
25    def test_boolrefptr_with_dwarf_and_run_command(self):
26        """Test the formatters we use for BOOL& and BOOL* in Objective-C."""
27        self.buildDwarf()
28        self.boolrefptr_data_formatter_commands()
29
30    def setUp(self):
31        # Call super's setUp().
32        TestBase.setUp(self)
33        # Find the line number to break at.
34        self.line = line_number('main.mm', '// Set break point at this line.')
35
36    def boolrefptr_data_formatter_commands(self):
37        """Test the formatters we use for BOOL& and BOOL* in Objective-C."""
38        self.runCmd("file a.out", CURRENT_EXECUTABLE_SET)
39
40        lldbutil.run_break_set_by_file_and_line (self, "main.mm", self.line, num_expected_locations=1, loc_exact=True)
41
42        self.runCmd("run", RUN_SUCCEEDED)
43
44        # The stop reason of the thread should be breakpoint.
45        self.expect("thread list", STOPPED_DUE_TO_BREAKPOINT,
46            substrs = ['stopped',
47                       'stop reason = breakpoint'])
48
49        # This is the function to remove the custom formats in order to have a
50        # clean slate for the next test case.
51        def cleanup():
52            self.runCmd('type format clear', check=False)
53            self.runCmd('type summary clear', check=False)
54            self.runCmd('type synth clear', check=False)
55
56        # Execute the cleanup function during test case tear down.
57        self.addTearDownHook(cleanup)
58
59        # Now check that we use the right summary for BOOL&
60        self.expect('frame variable yes_ref',
61                    substrs = ['YES'])
62        self.expect('frame variable no_ref',
63                    substrs = ['NO'])
64
65
66        # Now check that we use the right summary for BOOL*
67        self.expect('frame variable yes_ptr',
68                    substrs = ['YES'])
69        self.expect('frame variable no_ptr',
70                    substrs = ['NO'])
71
72
73        # Now check that we use the right summary for BOOL
74        self.expect('frame variable yes',
75                    substrs = ['YES'])
76        self.expect('frame variable no',
77                    substrs = ['NO'])
78
79
80if __name__ == '__main__':
81    import atexit
82    lldb.SBDebugger.Initialize()
83    atexit.register(lambda: lldb.SBDebugger.Terminate())
84    unittest2.main()
85