diff options
| -rw-r--r-- | lldb/test/command_source/TestCommandSource.py | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/lldb/test/command_source/TestCommandSource.py b/lldb/test/command_source/TestCommandSource.py index b7ae947c5ce..0f34eb464a5 100644 --- a/lldb/test/command_source/TestCommandSource.py +++ b/lldb/test/command_source/TestCommandSource.py @@ -20,9 +20,23 @@ class CommandSourceTestCase(TestBase): # 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 |
