1 files changed, 67 insertions, 0 deletions
diff --git a/packages/Python/lldbsuite/test/benchmarks/stepping/TestSteppingSpeed.py b/packages/Python/lldbsuite/test/benchmarks/stepping/TestSteppingSpeed.py
new file mode 100644
index 0000000000000..14738a978e865
--- /dev/null
+++ b/packages/Python/lldbsuite/test/benchmarks/stepping/TestSteppingSpeed.py
@@ -0,0 +1,67 @@
+"""Test lldb's stepping speed."""
+
+from __future__ import print_function
+
+import os, sys
+import lldb
+from lldbsuite.test import configuration
+from lldbsuite.test import lldbtest_config
+from lldbsuite.test.lldbbench import *
+
+class SteppingSpeedBench(BenchBase):
+
+    mydir = TestBase.compute_mydir(__file__)
+
+    def setUp(self):
+        BenchBase.setUp(self)
+        self.exe = lldbtest_config.lldbExec
+        self.break_spec = '-n main'
+        self.count = 50
+
+        #print("self.exe=%s" % self.exe)
+        #print("self.break_spec=%s" % self.break_spec)
+
+    @benchmarks_test
+    @no_debug_info_test
+    @expectedFailureWindows("llvm.org/pr22274: need a pexpect replacement for windows")
+    def test_run_lldb_steppings(self):
+        """Test lldb steppings on a large executable."""
+        print()
+        self.run_lldb_steppings(self.exe, self.break_spec, self.count)
+        print("lldb stepping benchmark:", self.stopwatch)
+
+    def run_lldb_steppings(self, exe, break_spec, count):
+        import pexpect
+        # Set self.child_prompt, which is "(lldb) ".
+        self.child_prompt = '(lldb) '
+        prompt = self.child_prompt
+
+        # So that the child gets torn down after the test.
+        self.child = pexpect.spawn('%s %s %s' % (lldbtest_config.lldbExec, self.lldbOption, exe))
+        child = self.child
+
+        # Turn on logging for what the child sends back.
+        if self.TraceOn():
+            child.logfile_read = sys.stdout
+
+        child.expect_exact(prompt)
+        child.sendline('breakpoint set %s' % break_spec)
+        child.expect_exact(prompt)
+        child.sendline('run')
+        child.expect_exact(prompt)
+
+        # Reset the stopwatch now.
+        self.stopwatch.reset()
+        for i in range(count):
+            with self.stopwatch:
+                # Disassemble the function.
+                child.sendline('next') # Aka 'thread step-over'.
+                child.expect_exact(prompt)
+
+        child.sendline('quit')
+        try:
+            self.child.expect(pexpect.EOF)
+        except:
+            pass
+
+        self.child = None