teak-llvm/lldb/packages/Python/lldbsuite/test/functionalities/command_script/TestCommandScript.py
Zachary Turner c432c8f856 Move lldb/test to lldb/packages/Python/lldbsuite/test.
This is the conclusion of an effort to get LLDB's Python code
structured into a bona-fide Python package.  This has a number
of benefits, but most notably the ability to more easily share
Python code between different but related pieces of LLDB's Python
infrastructure (for example, `scripts` can now share code with
`test`).

llvm-svn: 251532
2015-10-28 17:43:26 +00:00

143 lines
5.3 KiB
Python

"""
Test lldb Python commands.
"""
from __future__ import print_function
import use_lldb_suite
import os, time
import lldb
from 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)