Hardened the test_command_source() test case by actually capturing the output
from running the "script my.date()" lldb command and comparing it against our
expected result.
llvm-svn: 124499
diff --git a/lldb/test/command_source/TestCommandSource.py b/lldb/test/command_source/TestCommandSource.py
index b7ae947..0f34eb46 100644
--- a/lldb/test/command_source/TestCommandSource.py
+++ b/lldb/test/command_source/TestCommandSource.py
@@ -20,9 +20,23 @@
# 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.
self.runCmd("script my.date()")
+ import datetime
+ self.expect(session.getvalue(), "script my.date() runs successfully",
+ exe=False,
+ substrs = [str(datetime.date.today())])
+
+ # Now restore stdout to the way we were. :-)
+ sys.stdout = old_stdout
+
if __name__ == '__main__':
import atexit