xref: /llvm-project/lldb/test/API/python_api/sbmodule/TestSBModule.py (revision 326516448c839d8f9cc515b20a34d0f3a6ee2374)
1"""Test the SBDModule APIs."""
2
3import lldb
4from lldbsuite.test.decorators import *
5from lldbsuite.test.lldbtest import *
6from lldbsuite.test import lldbutil
7
8import os, signal, subprocess
9
10class SBModuleAPICase(TestBase):
11
12    mydir = TestBase.compute_mydir(__file__)
13
14    def setUp(self):
15        TestBase.setUp(self)
16        self.background_pid = None
17
18    def tearDown(self):
19        TestBase.tearDown(self)
20        if self.background_pid:
21            os.kill(self.background_pid, signal.SIGKILL)
22
23
24    def test_module_is_file_backed(self):
25        """Test the SBModule::IsFileBacked() method"""
26        self.build()
27        target, _, _, _  = lldbutil.run_to_source_breakpoint(self, "// break here",
28                                                    lldb.SBFileSpec("main.c"))
29
30        self.assertGreater(target.GetNumModules(), 0)
31        main_module = target.GetModuleAtIndex(0)
32        self.assertEqual(main_module.GetFileSpec().GetFilename(), "a.out")
33        self.assertTrue(main_module.IsFileBacked(),
34                         "The module should be backed by a file on disk")
35
36        self.dbg.DeleteTarget(target)
37        self.assertEqual(self.dbg.GetNumTargets(), 0)
38
39        exe = self.getBuildArtifact("a.out")
40        background_process = subprocess.Popen([exe])
41        self.assertTrue(background_process, "process is not valid")
42        self.background_pid = background_process.pid
43        os.unlink(exe)
44
45        target = self.dbg.CreateTarget('')
46        self.assertEqual(self.dbg.GetNumTargets(), 1)
47        error = lldb.SBError()
48        process = target.AttachToProcessWithID(self.dbg.GetListener(),
49                                               self.background_pid, error)
50        self.assertTrue(error.Success(),  PROCESS_IS_VALID)
51        main_module = target.GetModuleAtIndex(0)
52        self.assertEqual(main_module.GetFileSpec().GetFilename(), "a.out")
53        self.assertFalse(main_module.IsFileBacked(),
54                         "The module should not be backed by a file on disk.")
55
56        error = process.Detach()
57        self.assertTrue(error.Success(), "couldn't destroy process %s" % background_process.pid)
58
59