Test case for the previous commit.

llvm-svn: 205381

lldb/test/functionalities/breakpoint/breakpoint_command/TestBreakpointCommandsFromPython.py

@@ -0,0 +1,88 @@
+"""
+Test that you can set breakpoint commands successfully with the Python API's:
+"""
+
+import os
+import re
+import unittest2
+import lldb, lldbutil
+import sys
+from lldbtest import *
+
+class PythonBreakpointCommandSettingTestCase(TestBase):
+
+    mydir = TestBase.compute_mydir(__file__)
+    my_var = 10
+
+    @unittest2.skipUnless(sys.platform.startswith("darwin"), "requires Darwin")
+    @python_api_test
+    @dsym_test
+    def test_step_out_with_dsym_python(self):
+        """Test stepping out using avoid-no-debug with dsyms."""
+        self.buildDsym()
+        self.do_set_python_command_from_python()
+
+    @python_api_test
+    @dwarf_test
+    def test_step_out_with_dwarf_python(self):
+        """Test stepping out using avoid-no-debug with dsyms."""
+        self.buildDwarf()
+        self.do_set_python_command_from_python ()
+
+    def setUp (self):
+        TestBase.setUp(self)
+        self.main_source = "main.c"
+        self.main_source_spec = lldb.SBFileSpec(self.main_source)
+
+
+    def do_set_python_command_from_python (self):
+        exe = os.path.join(os.getcwd(), "a.out")
+        error = lldb.SBError()
+
+        self.target = self.dbg.CreateTarget(exe)
+        self.assertTrue(self.target, VALID_TARGET)
+
+        body_bkpt = self.target.BreakpointCreateBySourceRegex("Set break point at this line.", self.main_source_spec)
+        self.assertTrue(body_bkpt, VALID_BREAKPOINT)
+
+        func_bkpt = self.target.BreakpointCreateBySourceRegex("Set break point at this line.", self.main_source_spec)
+        self.assertTrue(func_bkpt, VALID_BREAKPOINT)
+
+        PythonBreakpointCommandSettingTestCase.my_var = 10
+        error = lldb.SBError()
+        error = body_bkpt.SetScriptCallbackBody("\
+import TestBreakpointCommandsFromPython\n\
+TestBreakpointCommandsFromPython.PythonBreakpointCommandSettingTestCase.my_var = 20\n\
+print 'Hit breakpoint'")
+        self.assertTrue (error.Success(), "Failed to set the script callback body: %s."%(error.GetCString()))
+
+        self.dbg.HandleCommand("command script import --allow-reload ./bktptcmd.py")
+        func_bkpt.SetScriptCallbackFunction("bktptcmd.function")
+
+        # We will use the function that touches a text file, so remove it first:
+        self.RemoveTempFile("output2.txt")
+
+        # Now launch the process, and do not stop at entry point.
+        self.process = self.target.LaunchSimple (None, None, self.get_process_working_directory())
+
+        self.assertTrue(self.process, PROCESS_IS_VALID)
+
+        # Now finish, and make sure the return value is correct.
+        threads = lldbutil.get_threads_stopped_at_breakpoint (self.process, body_bkpt)
+        self.assertTrue(len(threads) == 1, "Stopped at inner breakpoint.")
+        self.thread = threads[0]
+    
+        self.assertTrue(PythonBreakpointCommandSettingTestCase.my_var == 20)
+
+        # Check for the function version as well, which produced this file:
+        # Remember to clean up after ourselves...
+        self.assertTrue(os.path.isfile("output2.txt"),
+                        "'output2.txt' exists due to breakpoint command for breakpoint function.")
+        self.RemoveTempFile("output2.txt")
+
+        
+if __name__ == '__main__':
+    import atexit
+    lldb.SBDebugger.Initialize()
+    atexit.register(lambda: lldb.SBDebugger.Terminate())
+    unittest2.main()