Run "git stash" and "git stash apply" before and after a dry test run.

tools/auto_bisect/bisect_perf_regression_test.py

 # Copyright 2014 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import os
 import re
 import shutil
 import sys
 import unittest
 
@@ skipping 134 matching lines @@
 def _GenericDryRun(options, print_results=False):
   """Performs a dry run of the bisector.
 
   Args:
     options: Dictionary containing the options for the bisect instance.
     print_results: Boolean telling whether to call FormatAndPrintResults.
 
   Returns:
     The results dictionary as returned by the bisect Run method.
   """
+  _AbortIfThereAreStagedChanges()
   # Disable rmtree to avoid deleting local trees.
   old_rmtree = shutil.rmtree
+  shutil.rmtree = lambda path, on_error: None
+  # git reset HEAD may be run during the dry run, which removes staged changes.
   try:
-    shutil.rmtree = lambda path, onerror: None
     bisect_instance = _GetBisectPerformanceMetricsInstance(options)
     results = bisect_instance.Run(
         bisect_instance.opts.command, bisect_instance.opts.bad_revision,
         bisect_instance.opts.good_revision, bisect_instance.opts.metric)
 
     if print_results:
       bisect_instance.printer.FormatAndPrintResults(results)
 
     return results
   finally:
     shutil.rmtree = old_rmtree
 
 
+def _AbortIfThereAreStagedChanges():
+  """Exits the test prematurely if there are staged changes."""
+  # The output of "git status --short" will be an empty string if there are
+  # no staged changes in the current branch. Untracked files are ignored
+  # because when running the presubmit on the trybot there are sometimes
+  # untracked changes to the run-perf-test.cfg and bisect.cfg files.
+  status_output = bisect_utils.CheckRunGit(
+      ['status', '--short', '--untracked-files=no'])
+  if status_output:
+    print 'There are un-committed changes in the current branch.'
+    print 'Aborting the tests to avoid destroying local changes. Changes:'
+    print status_output
+    sys.exit(1)
+
+
 class BisectPerfRegressionTest(unittest.TestCase):
   """Test case for other functions and classes in bisect-perf-regression.py."""
 
   def setUp(self):
     self.cwd = os.getcwd()
     os.chdir(os.path.abspath(os.path.join(os.path.dirname(__file__),
                                           os.path.pardir, os.path.pardir)))
 
   def tearDown(self):
     os.chdir(self.cwd)
@@ skipping 460 matching lines @@
     ]
     self._SetupRunGitMock(try_cmd)
     bisect_perf_regression._StartBuilderTryJob(
         fetch_build.PERF_BUILDER, git_revision, bot_name, bisect_job_name,
         patch)
 
 
 if __name__ == '__main__':
   unittest.main()