106d58b0c68d98bfef948861e722789d7c7b1172cEnrico Granata#coding=utf8
2f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata"""
3f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico GranataTest that C++ supports wchar_t correctly.
4f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata"""
5f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata
6f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granataimport os, time
7f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granataimport unittest2
8f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granataimport lldb
9f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granatafrom lldbtest import *
10f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granataimport lldbutil
11f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata
12f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granataclass CxxWCharTTestCase(TestBase):
13f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata
14f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata    mydir = os.path.join("lang", "cpp", "wchar_t")
15f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata
16f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata    @unittest2.skipUnless(sys.platform.startswith("darwin"), "requires Darwin")
17f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata    @dsym_test
18f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata    def test_with_dsym(self):
19f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata        """Test that C++ supports wchar_t correctly."""
20f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata        self.buildDsym()
21f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata        self.wchar_t()
22f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata
23f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata    @dwarf_test
24f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata    def test_with_dwarf(self):
25f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata        """Test that C++ supports wchar_t correctly."""
26f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata        self.buildDwarf()
27f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata        self.wchar_t()
28f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata
29f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata    def setUp(self):
30f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata        # Call super's setUp().
31f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata        TestBase.setUp(self)
32f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata        # Find the line number to break for main.cpp.
33f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata        self.source = 'main.cpp'
34f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata        self.line = line_number(self.source, '// Set break point at this line.')
35f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata
36f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata    def wchar_t(self):
37f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata        """Test that C++ supports wchar_t correctly."""
38f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata        exe = os.path.join(os.getcwd(), "a.out")
39f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata
40f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata        # Create a target by the debugger.
41f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata        target = self.dbg.CreateTarget(exe)
42f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata        self.assertTrue(target, VALID_TARGET)
43f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata
44f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata        # Break on the struct declration statement in main.cpp.
45f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata        lldbutil.run_break_set_by_file_and_line (self, "main.cpp", self.line)
46f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata
47f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata        # Now launch the process, and do not stop at entry point.
48f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata        process = target.LaunchSimple(None, None, os.getcwd())
49f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata
50f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata        if not process:
51f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata            self.fail("SBTarget.Launch() failed")
52f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata
53f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata        # Check that we correctly report templates on wchar_t
54f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata        self.expect("frame variable foo_y",
55f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata            substrs = ['(Foo<wchar_t>) foo_y = '])
56f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata
57f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata        # Check that we correctly report templates on int
58f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata        self.expect("frame variable foo_x",
59f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata            substrs = ['(Foo<int>) foo_x = '])
60f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata
61f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata        # Check that we correctly report wchar_t
62f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata        self.expect("frame variable foo_y.object",
63f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata            substrs = ['(wchar_t) foo_y.object = '])
64f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata
65f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata        # Check that we correctly report int
66f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata        self.expect("frame variable foo_x.object",
67f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata            substrs = ['(int) foo_x.object = '])
68f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata
69f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata        # Check that we can run expressions that return wchar_t
70cd8cd61e7280b9c050a85ba29fa4d3b20737df3aEnrico Granata        self.expect("expression L'a'",substrs = ['(wchar_t) $',"61 L'a'"])
71f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata
7206d58b0c68d98bfef948861e722789d7c7b1172cEnrico Granata        # Mazel Tov if this works!
7306d58b0c68d98bfef948861e722789d7c7b1172cEnrico Granata        self.expect("frame variable mazeltov",
743f4844e5ce44bb9d596357c9e7442d6a83037ed7Enrico Granata            substrs = ['(const wchar_t *) mazeltov = ','L"מזל טוב"'])
7506d58b0c68d98bfef948861e722789d7c7b1172cEnrico Granata
76f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata
77f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granataif __name__ == '__main__':
78f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata    import atexit
79f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata    lldb.SBDebugger.Initialize()
80f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata    atexit.register(lambda: lldb.SBDebugger.Terminate())
81f34feefb6fe62d79a8a261a716194e34f1c4b8bcEnrico Granata    unittest2.main()
82