1"""
2Test that LLDB can emit JIT objects when the appropriate setting is enabled
3"""
4
5
6import os
7import lldb
8from lldbsuite.test.decorators import *
9from lldbsuite.test.lldbtest import *
10from lldbsuite.test import lldbutil
11
12class SaveJITObjectsTestCase(TestBase):
13
14    def enumerateJITFiles(self):
15        return [f for f in os.listdir(self.getBuildDir()) if f.startswith("jit")]
16
17    def countJITFiles(self):
18        return len(self.enumerateJITFiles())
19
20    def cleanJITFiles(self):
21        for j in self.enumerateJITFiles():
22            os.remove(j)
23        return
24
25    @expectedFailureAll(oslist=["windows"])
26    def test_save_jit_objects(self):
27        self.build()
28        os.chdir(self.getBuildDir())
29        src_file = "main.c"
30        src_file_spec = lldb.SBFileSpec(src_file)
31
32        (target, process, thread, bkpt) = lldbutil.run_to_source_breakpoint(
33            self, "break", src_file_spec)
34
35        frame = thread.frames[0]
36
37        self.cleanJITFiles()
38        frame.EvaluateExpression("(void*)malloc(0x1)")
39        self.assertEquals(self.countJITFiles(), 0,
40                          "No files emitted with save-jit-objects-dir empty")
41
42        self.runCmd("settings set target.save-jit-objects-dir {0}".format(self.getBuildDir()))
43        frame.EvaluateExpression("(void*)malloc(0x1)")
44        jit_files_count = self.countJITFiles()
45        self.cleanJITFiles()
46        self.assertNotEqual(jit_files_count, 0,
47                            "At least one file emitted with save-jit-objects-dir set to the build dir")
48
49        process.Kill()
50        os.chdir(self.getSourceDir())
51