xref: /llvm-project/lldb/test/API/python_api/sbmodule/TestSBModule.py (revision 9c2468821ec51defd09c246fea4a47886fff8c01)
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
10
11class SBModuleAPICase(TestBase):
12    def setUp(self):
13        TestBase.setUp(self)
14        self.background_pid = None
15
16    def tearDown(self):
17        TestBase.tearDown(self)
18        if self.background_pid:
19            os.kill(self.background_pid, signal.SIGKILL)
20
21    @skipUnlessDarwin
22    @skipIfRemote
23    def test_module_is_file_backed(self):
24        """Test the SBModule::IsFileBacked() method"""
25        self.build()
26        target, _, _, _ = lldbutil.run_to_source_breakpoint(
27            self, "// break here", lldb.SBFileSpec("main.c")
28        )
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(
34            main_module.IsFileBacked(), "The module should be backed by a file on disk"
35        )
36
37        self.dbg.DeleteTarget(target)
38        self.assertEqual(self.dbg.GetNumTargets(), 0)
39
40        exe = self.getBuildArtifact("a.out")
41        background_process = subprocess.Popen([exe])
42        self.assertTrue(background_process, "process is not valid")
43        self.background_pid = background_process.pid
44        os.unlink(exe)
45
46        target = self.dbg.CreateTarget("")
47        self.assertEqual(self.dbg.GetNumTargets(), 1)
48        error = lldb.SBError()
49        process = target.AttachToProcessWithID(
50            self.dbg.GetListener(), self.background_pid, error
51        )
52        self.assertTrue(error.Success() and process, PROCESS_IS_VALID)
53        main_module = target.FindModule(lldb.SBFileSpec("a.out"))
54        self.assertIsNotNone(main_module)
55        self.assertFalse(
56            main_module.IsFileBacked(),
57            "The module should not be backed by a file on disk.",
58        )
59
60        error = process.Destroy()
61        self.assertSuccess(
62            error, "couldn't destroy process %s" % background_process.pid
63        )
64