Split FakeRepos class in two so it can be reused for commti-queue unit tests.

tests/fake_repos.py

 #!/usr/bin/python
 # Copyright (c) 2010 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.
 
 """Generate fake repositories for testing."""
 
 import atexit
 import errno
 import logging
@@ skipping 177 matching lines @@
   check_call(['git', 'add', '-A', '-f'], cwd=repo)
   check_call(['git', 'commit', '-q', '--message', 'foo'], cwd=repo)
   rev = Popen(['git', 'show-ref', '--head', 'HEAD'],
               cwd=repo).communicate()[0].split(' ', 1)[0]
   logging.debug('At revision %s' % rev)
   return rev
 
 
 _FAKE_LOADED = False
 
-class FakeRepos(object):
+class FakeReposBase(object):
   """Generate both svn and git repositories to test gclient functionality.
 
   Many DEPS functionalities need to be tested: Var, File, From, deps_os, hooks,
   use_relative_paths.
 
-  And types of dependencies: Relative urls, Full urls, both svn and git."""
+  And types of dependencies: Relative urls, Full urls, both svn and git.
+
+  populateSvn() and populateGit() need to be implemented by the subclass.
+  """
 
   # Should leak the repositories.
   SHOULD_LEAK = False
   # Override if unhappy.
   TRIAL_DIR = None
   # Hostname
   HOST = '127.0.0.1'
+  NB_GIT_REPOS = 1
 
   def __init__(self, trial_dir=None, leak=None, host=None):
     global _FAKE_LOADED
     if _FAKE_LOADED:
       raise Exception('You can only start one FakeRepos at a time.')
     _FAKE_LOADED = True
     # Quick hack.
     if '-v' in sys.argv:
       logging.basicConfig(level=logging.DEBUG)
     elif leak is not None:
@@ skipping 86 matching lines @@
         'user2 = bar\n')
 
     # Start the daemon.
     cmd = ['svnserve', '-d', '--foreground', '-r', self.repos_dir]
     if self.HOST == '127.0.0.1':
       cmd.append('--listen-host=127.0.0.1')
     self.svnserve = Popen(cmd, cwd=root)
     self.populateSvn()
     return True
 
+  def setUpGIT(self):
+    """Creates git repositories and start the servers."""
+    if self.gitdaemon:
+      return True
+    self.setUp()
+    if sys.platform == 'win32':
+      return False
+    for repo in ['repo_%d' % r for r in range(1, self.NB_GIT_REPOS + 1)]:
+      check_call(['git', 'init', '-q', join(self.git_root, repo)])
+      self.git_hashes[repo] = [None]
+    self.populateGit()
+    # Start the daemon.
+    cmd = ['git', 'daemon', '--export-all', '--base-path=' + self.repos_dir]
+    if self.HOST == '127.0.0.1':
+      cmd.append('--listen=127.0.0.1')
+    logging.debug(cmd)
+    self.gitdaemon = Popen(cmd, cwd=self.repos_dir)
+    return True
+
+  def _commit_svn(self, tree):
+    self._genTree(self.svn_root, tree)
+    commit_svn(self.svn_root)
+    if self.svn_revs and self.svn_revs[-1]:
+      new_tree = self.svn_revs[-1].copy()
+      new_tree.update(tree)
+    else:
+      new_tree = tree.copy()
+    self.svn_revs.append(new_tree)
+
+  def _commit_git(self, repo, tree):
+    repo_root = join(self.git_root, repo)
+    self._genTree(repo_root, tree)
+    commit_hash = commit_git(repo_root)
+    if self.git_hashes[repo][-1]:
+      new_tree = self.git_hashes[repo][-1][1].copy()
+      new_tree.update(tree)
+    else:
+      new_tree = tree.copy()
+    self.git_hashes[repo].append((commit_hash, new_tree))
+
+
+class FakeRepos(FakeReposBase):
+  """Implements populateSvn() and populateGit()."""
+  NB_GIT_REPOS = 4
+
   def populateSvn(self):
     """Creates a few revisions of changes including DEPS files."""
     # Repos
     check_call(['svn', 'checkout', 'svn://127.0.0.1/svn', self.svn_root, '-q',
                 '--non-interactive', '--no-auth-cache',
                 '--username', 'user1', '--password', 'foo'])
     assert os.path.isdir(join(self.svn_root, '.svn'))
     def file_system(rev, DEPS):
       fs = {
         'origin': 'svn@%(rev)d\n',
@@ skipping 85 matching lines @@
 
 hooks = [
   {
     'pattern': '.*',
     'action': ['echo', 'foo'],
   },
 ]
 """ % { 'host': self.HOST }
     self._commit_svn(fs)
 
-  def setUpGIT(self):
-    """Creates git repositories and start the servers."""
-    if self.gitdaemon:
-      return True
-    self.setUp()
-    if sys.platform == 'win32':
-      return False
-    for repo in ['repo_%d' % r for r in range(1, 5)]:
-      check_call(['git', 'init', '-q', join(self.git_root, repo)])
-      self.git_hashes[repo] = [None]
-
+  def populateGit(self):
     # Testing:
     # - dependency disapear
     # - dependency renamed
     # - versioned and unversioned reference
     # - relative and full reference
     # - deps_os
     # - var
     # - hooks
     # - From
     # TODO(maruel):
@@ skipping 79 matching lines @@
 """ % {
         'host': self.HOST,
         # See self.__init__() for the format. Grab's the hash of the first
         # commit in repo_2. Only keep the first 7 character because of:
         # TODO(maruel): http://crosbug.com/3591 We need to strip the hash.. duh.
         'hash': self.git_hashes['repo_2'][1][0][:7]
       },
       'origin': 'git/repo_1@2\n',
     })
 
-    # Start the daemon.
-    cmd = ['git', 'daemon', '--export-all', '--base-path=' + self.repos_dir]
-    if self.HOST == '127.0.0.1':
-      cmd.append('--listen=127.0.0.1')
-    logging.debug(cmd)
-    self.gitdaemon = Popen(cmd, cwd=self.repos_dir)
-    return True
-
-  def _commit_svn(self, tree):
-    self._genTree(self.svn_root, tree)
-    commit_svn(self.svn_root)
-    if self.svn_revs and self.svn_revs[-1]:
-      new_tree = self.svn_revs[-1].copy()
-      new_tree.update(tree)
-    else:
-      new_tree = tree.copy()
-    self.svn_revs.append(new_tree)
-
-  def _commit_git(self, repo, tree):
-    repo_root = join(self.git_root, repo)
-    self._genTree(repo_root, tree)
-    commit_hash = commit_git(repo_root)
-    if self.git_hashes[repo][-1]:
-      new_tree = self.git_hashes[repo][-1][1].copy()
-      new_tree.update(tree)
-    else:
-      new_tree = tree.copy()
-    self.git_hashes[repo].append((commit_hash, new_tree))
-
 
 class FakeReposTestBase(unittest.TestCase):
   """This is vaguely inspired by twisted."""
 
   # Replace this in your subclass.
   CLASS_ROOT_DIR = None
 
   # static FakeRepos instance. Lazy loaded.
   FAKE_REPOS = None
+  # Override if necessary.
+  FAKE_REPOS_CLASS = FakeRepos
 
   def __init__(self, *args, **kwargs):
     unittest.TestCase.__init__(self, *args, **kwargs)
     if not FakeReposTestBase.FAKE_REPOS:
-      FakeReposTestBase.FAKE_REPOS = FakeRepos()
+      FakeReposTestBase.FAKE_REPOS = self.FAKE_REPOS_CLASS()
 
   def setUp(self):
     unittest.TestCase.setUp(self)
     self.FAKE_REPOS.setUp()
 
     # Remove left overs and start fresh.
     if not self.CLASS_ROOT_DIR:
       self.CLASS_ROOT_DIR = join(self.FAKE_REPOS.trial_dir(), 'smoke')
     self.root_dir = join(self.CLASS_ROOT_DIR, self.id())
     rmtree(self.root_dir)
@@ skipping 84 matching lines @@
 
 
 # Kind of hack.
 if '-l' in sys.argv:
   FakeRepos.SHOULD_LEAK = True
   sys.argv.remove('-l')
 
 
 if __name__ == '__main__':
   sys.exit(main(sys.argv))