tools/bisect_test.py

   1 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 # Use of this source code is governed by a BSD-style license that can be
   3 # found in the LICENSE file.
   4
   5 import unittest
   6
   7 bisect_builds = __import__('bisect-builds')
   8
   9
  10 class BisectTest(unittest.TestCase):
  11
  12   patched = []
  13   max_rev = 10000
  14
  15   def monkey_patch(self, obj, name, new):
  16     self.patched.append((obj, name, getattr(obj, name)))
  17     setattr(obj, name, new)
  18
  19   def clear_patching(self):
  20     for obj, name, old in self.patched:
  21       setattr(obj, name, old)
  22     self.patched = []
  23
  24   def setUp(self):
  25     self.monkey_patch(bisect_builds.DownloadJob, 'Start', lambda *args: None)
  26     self.monkey_patch(bisect_builds.DownloadJob, 'Stop', lambda *args: None)
  27     self.monkey_patch(bisect_builds.DownloadJob, 'WaitFor', lambda *args: None)
  28     self.monkey_patch(bisect_builds, 'RunRevision', lambda *args: (0, "", ""))
  29     self.monkey_patch(bisect_builds.PathContext, 'ParseDirectoryIndex',
  30                       lambda *args: range(self.max_rev))
  31
  32   def tearDown(self):
  33     self.clear_patching()
  34
  35   def bisect(self, good_rev, bad_rev, evaluate):
  36     return bisect_builds.Bisect(good_rev=good_rev,
  37                                 bad_rev=bad_rev,
  38                                 evaluate=evaluate,
  39                                 num_runs=1,
  40                                 official_builds=False,
  41                                 platform='linux',
  42                                 profile=None,
  43                                 try_args=())
  44
  45   def testBisectConsistentAnswer(self):
  46     self.assertEqual(self.bisect(1000, 100, lambda *args: 'g'), (100, 101))
  47     self.assertEqual(self.bisect(100, 1000, lambda *args: 'b'), (100, 101))
  48     self.assertEqual(self.bisect(2000, 200, lambda *args: 'b'), (1999, 2000))
  49     self.assertEqual(self.bisect(200, 2000, lambda *args: 'g'), (1999, 2000))
  50
  51
  52 if __name__ == '__main__':
  53   unittest.main()