1 files changed, 64 insertions, 0 deletions
diff --git a/lldb/packages/Python/lldbsuite/test/benchmarks/stepping/TestRunHooksThenSteppings.py b/lldb/packages/Python/lldbsuite/test/benchmarks/stepping/TestRunHooksThenSteppings.py
new file mode 100644
index 00000000000..cb85dd19948
--- /dev/null
+++ b/lldb/packages/Python/lldbsuite/test/benchmarks/stepping/TestRunHooksThenSteppings.py
@@ -0,0 +1,64 @@
+"""Test lldb's stepping speed."""
+
+from __future__ import print_function
+
+import use_lldb_suite
+
+import os, sys
+import lldb
+from lldbbench import *
+
+class RunHooksThenSteppingsBench(BenchBase):
+
+    mydir = TestBase.compute_mydir(__file__)
+
+    def setUp(self):
+        BenchBase.setUp(self)
+        self.count = lldb.bmIterationCount
+        if self.count <= 0:
+            self.count = 50
+
+    @benchmarks_test
+    @no_debug_info_test
+    @expectedFailureWindows("llvm.org/pr22274: need a pexpect replacement for windows")
+    def test_lldb_runhooks_then_steppings(self):
+        """Test lldb steppings on a large executable."""
+        print()
+        self.run_lldb_runhooks_then_steppings(self.count)
+        print("lldb stepping benchmark:", self.stopwatch)
+
+    def run_lldb_runhooks_then_steppings(self, count):
+        import pexpect
+        # Set self.child_prompt, which is "(lldb) ".
+        self.child_prompt = '(lldb) '
+        prompt = self.child_prompt
+
+        self.child = pexpect.spawn('%s %s' % (lldbtest_config.lldbExec, self.lldbOption))
+        self.child.expect_exact(prompt)
+        # So that the child gets torn down after the test.
+        child = self.child
+
+        # Turn on logging for what the child sends back.
+        if self.TraceOn():
+            child.logfile_read = sys.stdout
+
+        #lldb.runHooks = ['process attach -n Mail']
+
+        # Perform the run hooks to bring lldb debugger to the desired state.
+        self.runHooks(child=child, child_prompt=prompt)
+
+        # Reset the stopwatch now.
+        self.stopwatch.reset()
+        for i in range(count):
+            with self.stopwatch:
+                # Step through 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