1"""
2Test lldb data formatter subsystem.
3"""
4
5import os, time
6import unittest2
7import lldb
8from lldbtest import *
9import datetime
10import lldbutil
11
12class DataFormatterOSTypeTestCase(TestBase):
13
14    mydir = os.path.join("functionalities", "data-formatter", "rdar-11628688")
15
16    @unittest2.skipUnless(sys.platform.startswith("darwin"), "requires Darwin")
17    @dsym_test
18    def test_ostype_with_dsym_and_run_command(self):
19        """Test the formatters we use for OSType."""
20        self.buildDsym()
21        self.ostype_data_formatter_commands()
22
23    @unittest2.skipUnless(sys.platform.startswith("darwin"), "requires Darwin")
24    @dwarf_test
25    def test_ostype_with_dwarf_and_run_command(self):
26        """Test the formatters we use for OSType."""
27        self.buildDwarf()
28        self.ostype_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 ostype_data_formatter_commands(self):
37        """Test the formatters we use for OSType."""
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 OSType
60        self.expect('frame variable',
61                    substrs = ["(OSType) a = 1952805748 'test'","(OSType) b = 1650815860 'best'"])
62
63
64if __name__ == '__main__':
65    import atexit
66    lldb.SBDebugger.Initialize()
67    atexit.register(lambda: lldb.SBDebugger.Terminate())
68    unittest2.main()
69