depot_tools/tests/git_number_test.py

#!/usr/bin/env vpython3
# Copyright 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


class Basic(git_test_utils.GitRepoReadWriteTestBase):
  REPO_SCHEMA = """
  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.old_POOL_KIND = cls.gn.POOL_KIND
    cls.gn.POOL_KIND = 'threads'

  @classmethod
  def tearDownClass(cls):
    cls.gn.POOL_KIND = cls.old_POOL_KIND
    super(Basic, cls).tearDownClass()

  def tearDown(self):
    self.gn.clear_caches()
    super(Basic, self).tearDown()

  def _git_number(self, refs, cache=False):
    refs = [binascii.unhexlify(ref) for ref in refs]
    self.repo.run(self.gn.load_generation_numbers, refs)
    if cache:
      self.repo.run(self.gn.finalize, refs)
    return [self.gn.get_num(ref) for ref in refs]

  def testBasic(self):
    self.assertEqual([0], self._git_number([self.repo['A']]))
    self.assertEqual([2], self._git_number([self.repo['F']]))
    self.assertEqual([0], self._git_number([self.repo['X']]))
    self.assertEqual([4], self._git_number([self.repo['E']]))

  def testInProcessCache(self):
    self.assertEqual(
        None,
        self.repo.run(self.gn.get_num, binascii.unhexlify(self.repo['A'])))
    self.assertEqual([4], self._git_number([self.repo['E']]))
    self.assertEqual(
        0,
        self.repo.run(self.gn.get_num, binascii.unhexlify(self.repo['A'])))

  def testOnDiskCache(self):
    self.assertEqual(
        None,
        self.repo.run(self.gn.get_num, binascii.unhexlify(self.repo['A'])))
    self.assertEqual([4], self._git_number([self.repo['E']], cache=True))
    self.assertEqual([4], self._git_number([self.repo['E']], cache=True))
    self.gn.clear_caches()
    self.assertEqual(
        0,
        self.repo.run(self.gn.get_num, binascii.unhexlify(self.repo['A'])))
    self.gn.clear_caches()
    self.repo.run(self.gn.clear_caches, True)
    self.assertEqual(
        None,
        self.repo.run(self.gn.get_num, binascii.unhexlify(self.repo['A'])))


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