1 files changed, 142 insertions, 0 deletions
diff --git a/packages/Python/lldbsuite/test/functionalities/command_script/TestCommandScript.py b/packages/Python/lldbsuite/test/functionalities/command_script/TestCommandScript.py
new file mode 100644
index 000000000000..542ee7a1e5f3
--- /dev/null
+++ b/packages/Python/lldbsuite/test/functionalities/command_script/TestCommandScript.py
@@ -0,0 +1,142 @@
+"""
+Test lldb Python commands.
+"""
+
+from __future__ import print_function
+
+
+
+import os, time
+import lldb
+from lldbsuite.test.lldbtest import *
+
+class CmdPythonTestCase(TestBase):
+
+    mydir = TestBase.compute_mydir(__file__)
+
+    def test (self):
+        self.build ()
+        self.pycmd_tests ()
+
+    def pycmd_tests (self):
+        self.runCmd("command source py_import")
+
+        # Verify command that specifies eCommandRequiresTarget returns failure
+        # without a target.
+        self.expect('targetname',
+            substrs = ['a.out'], matching=False, error=True)
+
+        exe = os.path.join (os.getcwd(), "a.out")
+        self.expect("file " + exe,
+                    patterns = [ "Current executable set to .*a.out" ])
+
+        self.expect('targetname',
+            substrs = ['a.out'], matching=True, error=False)
+
+        # This is the function to remove the custom commands in order to have a
+        # clean slate for the next test case.
+        def cleanup():
+            self.runCmd('command script delete welcome', check=False)
+            self.runCmd('command script delete targetname', check=False)
+            self.runCmd('command script delete longwait', check=False)
+            self.runCmd('command script delete mysto', check=False)
+            self.runCmd('command script delete tell_sync', check=False)
+            self.runCmd('command script delete tell_async', check=False)
+            self.runCmd('command script delete tell_curr', check=False)
+            self.runCmd('command script delete bug11569', check=False)
+            self.runCmd('command script delete takes_exe_ctx', check=False)
+
+        # Execute the cleanup function during test case tear down.
+        self.addTearDownHook(cleanup)
+
+        # Interact with debugger in synchronous mode
+        self.setAsync(False)
+
+        # We don't want to display the stdout if not in TraceOn() mode.
+        if not self.TraceOn():
+            self.HideStdout()
+
+        self.expect('welcome Enrico',
+            substrs = ['Hello Enrico, welcome to LLDB']);
+                
+        self.expect("help welcome",
+                    substrs = ['Just a docstring for welcome_impl',
+                               'A command that says hello to LLDB users'])
+
+        self.expect("help",
+                    substrs = ['For more information run',
+                               'welcome'])
+
+        self.expect("help -a",
+                    substrs = ['For more information run',
+                               'welcome'])
+
+        self.expect("help -u", matching=False,
+                    substrs = ['For more information run'])
+
+        self.runCmd("command script delete welcome");
+
+        self.expect('welcome Enrico', matching=False, error=True,
+                substrs = ['Hello Enrico, welcome to LLDB']);
+
+        self.expect('targetname fail', error=True,
+                    substrs = ['a test for error in command'])
+
+        self.expect('command script list',
+            substrs = ['targetname',
+                       'For more information run'])
+
+        self.expect("help targetname",
+                    substrs = ['This', 'command', 'takes', '\'raw\'', 'input',
+                               'quote', 'stuff'])
+
+        self.expect("longwait",
+                    substrs = ['Done; if you saw the delays I am doing OK'])
+
+        self.runCmd("b main")
+        self.runCmd("run")
+        self.runCmd("mysto 3")
+        self.expect("frame variable array",
+                    substrs = ['[0] = 79630','[1] = 388785018','[2] = 0'])
+        self.runCmd("mysto 3")
+        self.expect("frame variable array",
+                    substrs = ['[0] = 79630','[4] = 388785018','[5] = 0'])
+
+# we cannot use the stepover command to check for async execution mode since LLDB
+# seems to get confused when events start to queue up
+        self.expect("tell_sync",
+                    substrs = ['running sync'])
+        self.expect("tell_async",
+                    substrs = ['running async'])
+        self.expect("tell_curr",
+                    substrs = ['I am running sync'])
+                    
+# check that the execution context is passed in to commands that ask for it
+        self.expect("takes_exe_ctx", substrs = ["a.out"])
+
+        # Test that a python command can redefine itself
+        self.expect('command script add -f foobar welcome -h "just some help"')
+        
+        self.runCmd("command script clear")
+
+        # Test that re-defining an existing command works
+        self.runCmd('command script add my_command --class welcome.WelcomeCommand')
+        self.expect('my_command Blah', substrs = ['Hello Blah, welcome to LLDB'])
+
+        self.runCmd('command script add my_command --class welcome.TargetnameCommand')
+        self.expect('my_command', substrs = ['a.out'])
+
+        self.runCmd("command script clear")
+                
+        self.expect('command script list', matching=False,
+                    substrs = ['targetname',
+                               'longwait'])
+
+        self.expect('command script add -f foobar frame', error=True,
+                    substrs = ['cannot add command'])
+
+        # http://llvm.org/bugs/show_bug.cgi?id=11569
+        # LLDBSwigPythonCallCommand crashes when a command script returns an object 
+        self.runCmd('command script add -f bug11569 bug11569')
+        # This should not crash.
+        self.runCmd('bug11569', check=False)