Converted some more test cases to use runCmd()/expect().

llvm-svn: 111652

1 files changed, 2 insertions, 7 deletions

diff --git a/lldb/test/command_source/TestCommandSource.py b/lldb/test/command_source/TestCommandSource.py
index c9f151e698c..1a78a7f4285 100644
--- a/lldb/test/command_source/TestCommandSource.py
+++ b/lldb/test/command_source/TestCommandSource.py
@@ -15,18 +15,13 @@ class TestCommandSource(TestBase):
 
     def test_command_source(self):
         """Test that lldb command "command source" works correctly."""
-        res = self.res
 
         # Sourcing .lldb in the current working directory, which in turn imports
         # the "my" package that defines the date() function.
-        self.ci.HandleCommand("command source .lldb", res)
-        self.assertTrue(res.Succeeded(), CMD_MSG('command source .lldb'))
+        self.runCmd("command source .lldb")
 
         # Python should evaluate "my.date()" successfully.
-        self.ci.HandleCommand("script my.date()", res)
-        if (not res.Succeeded()):
-            print res.GetError()
-        self.assertTrue(res.Succeeded(), CMD_MSG('script my.date()'))
+        self.runCmd("script my.date()")
 
 
 if __name__ == '__main__':