4570d3eba0
This has led to many test suite failures because of copy and paste where new test cases were based off of other test cases and the "mydir" variable wasn't updated. Now you can call your superclasses "compute_mydir()" function with "__file__" as the sole argument and the relative path will be computed for you. llvm-svn: 196985
47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
"""
|
|
Test that lldb command "command source" works correctly.
|
|
|
|
See also http://llvm.org/viewvc/llvm-project?view=rev&revision=109673.
|
|
"""
|
|
|
|
import os, sys
|
|
import unittest2
|
|
import lldb
|
|
from lldbtest import *
|
|
|
|
class CommandSourceTestCase(TestBase):
|
|
|
|
mydir = TestBase.compute_mydir(__file__)
|
|
|
|
def test_command_source(self):
|
|
"""Test that lldb command "command source" works correctly."""
|
|
|
|
# Sourcing .lldb in the current working directory, which in turn imports
|
|
# the "my" package that defines the date() function.
|
|
self.runCmd("command source .lldb")
|
|
|
|
# Let's temporarily redirect the stdout to our StringIO session object
|
|
# in order to capture the script evaluation output.
|
|
old_stdout = sys.stdout
|
|
session = StringIO.StringIO()
|
|
sys.stdout = session
|
|
|
|
# Python should evaluate "my.date()" successfully.
|
|
# Pass 'check=False' so that sys.stdout gets restored unconditionally.
|
|
self.runCmd("script my.date()", check=False)
|
|
|
|
# Now restore stdout to the way we were. :-)
|
|
sys.stdout = old_stdout
|
|
|
|
import datetime
|
|
self.expect(session.getvalue(), "script my.date() runs successfully",
|
|
exe=False,
|
|
substrs = [str(datetime.date.today())])
|
|
|
|
|
|
if __name__ == '__main__':
|
|
import atexit
|
|
lldb.SBDebugger.Initialize()
|
|
atexit.register(lambda: lldb.SBDebugger.Terminate())
|
|
unittest2.main()
|