Add git-number script to calculate generation numbers for commits.

tests/git_number_test.py

+#!/usr/bin/env python
+# Copyright (c) 2013 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.
+
+"""Unit tests for git_number.py"""
+
+import binascii
+import os
+import sys
+
+DEPOT_TOOLS_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+sys.path.insert(0, DEPOT_TOOLS_ROOT)
+
+from testing_support import git_test_utils
+from testing_support import coverage_utils
+

[M-A Ruel, 2013/11/08 19:32:23]

2 lines

[iannucci, 2013/11/11 22:59:24]

Done.

+class Basic(git_test_utils.GitRepoReadWriteTestBase):
+  REPO = """
+  A B C D E
+    B   F E
+  X Y     E
+  """
+
+  @classmethod
+  def setUpClass(cls):
+    super(Basic, cls).setUpClass()
+    import git_number
+    cls.gn = git_number
+    cls.gn.POOL_KIND = 'threads'
+
+  @classmethod
+  def tearDownClass(cls):
+    cls.gn.POOL_KIND = 'procs'

[M-A Ruel, 2013/11/08 19:32:23]

I'd prefer you to cache the old value before line 30 and put it back here.

[iannucci, 2013/11/11 22:59:24]

Done.

+    super(Basic, cls).tearDownClass()
+
+  def tearDown(self):
+    self.gn.clear_caches()
+    super(Basic, self).tearDown()
+
+  def _git_number(self, refs, reset=False, cache=False):
+    return self.repo.run(
+      self.gn.git_number,
+      reset, cache, refs
+    )
+
+  def testBasic(self):
+    self.assertEqual(self._git_number([self.repo['A']]), [0])
+    self.assertEqual(self._git_number([self.repo['F']]), [2])
+    self.assertEqual(self._git_number([self.repo['X']]), [0])
+    self.assertEqual(self._git_number([self.repo['E']]), [4])
+
+  def testInProcessCache(self):
+    self.assertEqual(
+      self.repo.run(self.gn.get_num, binascii.unhexlify(self.repo['A'])), None
+    )
+    self.assertEqual(self._git_number([self.repo['E']]), [4])
+    self.assertEqual(
+      self.repo.run(self.gn.get_num, binascii.unhexlify(self.repo['A'])), 0
+    )
+
+  def testOnDiskCache(self):
+    self.assertEqual(
+      self.repo.run(self.gn.get_num, binascii.unhexlify(self.repo['A'])), None
+    )
+    self.assertEqual(self._git_number([self.repo['E']], cache=True), [4])
+    self.assertEqual(self._git_number([self.repo['E']], cache=True), [4])
+    self.gn.clear_caches()
+    self.assertEqual(
+      self.repo.run(self.gn.get_num, binascii.unhexlify(self.repo['A'])), 0
+    )
+    self.gn.clear_caches()
+    self._git_number([], reset=True)
+    self.assertEqual(
+      self.repo.run(self.gn.get_num, binascii.unhexlify(self.repo['A'])), None

[M-A Ruel, 2013/11/08 19:32:23]

I personally prefer putting the expected value first but I don't mind much.

+    )
+
+
+if __name__ == '__main__':
+  sys.exit(coverage_utils.covered_main(
+    os.path.join(DEPOT_TOOLS_ROOT, 'git_number.py')
+  ))