Refactor infra git libs and testing.

infra/libs/git2/test/repo_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 hashlib
+import shutil
+import sys
+
+from infra.libs import git2
+from infra.libs.git2.test import test_util
+
+
+class TestRepo(test_util.TestBasis):
+  def testEmptyRepo(self):
+    r = git2.Repo('doesnt_exist')
+    r.repos_dir = self.repos_dir
+
+    with self.assertRaises(git2.CalledProcessError):
+      self.capture_stdio(r.reify)
+
+    with self.assertRaises(AssertionError):
+      r.run('show-ref')
+
+  def testDefaultRepo(self):
+    r = self.mkRepo()
+    r.reify()  # covers 'already initialized' portion of reify()
+    self.assertEqual(r.run('rev-parse', 'branch_F').strip(), self.repo['F'])
+    _, err = self.capture_stdio(r.run, 'rev-parse', 'rtaitnariostnr',
+                                ok_ret={128})
+    self.assertIn('fatal', err)
+
+  def testDefaultRepoGit(self):
+    shutil.move(self.repo.repo_path, self.repo.repo_path + '.git')
+    self.repo.repo_path += '.git'
+    r = self.mkRepo()
+    self.assertFalse(r._repo_path.endswith('.git'))  # pylint: disable=W0212
+    self.assertIn('refs/heads/branch_F', r.run('show-ref'))
+
+  def testRefglob(self):
+    r = self.mkRepo()
+    self.assertEqual(
+        {rf.ref for rf in r.refglob('*branch_*')},
+        {'refs/heads/branch_'+l for l in 'FOKSZ'})
+    self.assertIs(next(r.refglob('*branch_O')).repo, r)
+    self.assertEqual(list(r.refglob('*atritaosrtientsaroitna*')), [])
+
+  def testDryRun(self):
+    r = self.mkRepo()
+    r.dry_run = True
+    self.assertIsNone(r.run('push', 'origin', 'HEAD'))
+
+  def testRunIndata(self):
+    r = self.mkRepo()
+    with self.assertRaises(AssertionError):
+      r.run('bogus', indata='spam', stdin=sys.stdin)
+    self.assertEqual(
+        r.run('rev-list', '--stdin', indata='branch_O~').splitlines()[0],
+        self.repo['N'])
+
+  def testGetCommit(self):
+    # pylint: disable=W0212
+    r = self.mkRepo()
+    c = r.get_commit(self.repo['L'])
+    self.assertEqual(c.hsh, self.repo['L'])
+    self.assertEqual([self.repo['L']], r._commit_cache.keys())
+
+    c2 = r.get_commit(self.repo['L'])
+    self.assertIs(c, c2)
+
+  def testGetCommitEviction(self):
+    # pylint: disable=W0212
+    r = self.mkRepo()
+    r.MAX_CACHE_SIZE = 2
+    L = r.get_commit(self.repo['L'])
+    self.assertIs(L, r.get_commit(self.repo['L']))
+    self.assertEqual(len(r._commit_cache), 1)
+
+    O = r.get_commit(self.repo['O'])
+    self.assertIs(L, r.get_commit(self.repo['L']))
+    self.assertIs(O, r.get_commit(self.repo['O']))
+    self.assertEqual(len(r._commit_cache), 2)
+
+    N = r.get_commit(self.repo['N'])
+    self.assertIs(N, r.get_commit(self.repo['N']))
+    self.assertIs(O, r.get_commit(self.repo['O']))
+    self.assertEqual(len(r._commit_cache), 2)
+
+    self.assertIsNot(L, r.get_commit(self.repo['L']))
+
+  def testIntern(self):
+    r = self.mkRepo()
+    hsh = r.intern('catfood')
+    self.assertEqual(hsh, hashlib.sha1('blob 7\0catfood').hexdigest())
+    self.assertEqual('catfood', r.run('cat-file', 'blob', hsh))
+
+  def testGetRef(self):
+    r = self.mkRepo()
+    self.assertEqual(r['refs/heads/branch_Z'].commit.hsh,
+                     'c9813df312aae7cadb91b0d4eb89ad1a4d1b22c9')