Add trial_dir to manage automatic trial directory creation and deletion.

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 datetime
 import errno
 import logging
 import os
 import pprint
 import re
 import socket
 import stat
 import subprocess
 import sys
 import tempfile
 import time
-import unittest
 
-sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
-
+# trial_dir must be first for non-system libraries.
+from tests import trial_dir
 import scm
 
 ## Utility functions
 
 
 def kill_pid(pid):
   """Kills a process by its process id."""
   try:
     # Unable to import 'module'
     # pylint: disable=F0401
@@ skipping 188 matching lines @@
 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.
 
   populateSvn() and populateGit() need to be implemented by the subclass.
   """
-
-  # When SHOULD_LEAK is set to True, temporary directories created while the
-  # tests are running aren't deleted at the end of the tests. Expect failures
-  # when running more than one test due to inter-test side-effects. Helps with
-  # debugging.
-  SHOULD_LEAK = False
-  # Override if unhappy.
-  TRIAL_DIR = None
   # Hostname
   NB_GIT_REPOS = 1
   USERS = [
       ('user1@example.com', 'foo'),
       ('user2@example.com', 'bar'),
   ]
 
-  def __init__(self, trial_dir=None, leak=None, host=None):
+  def __init__(self, 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:
-      self.SHOULD_LEAK = leak
+
+    self.trial = trial_dir.TrialDir('repos')
     self.host = host or '127.0.0.1'
-    if trial_dir:
-      self.TRIAL_DIR = trial_dir
-
     # Format is [ None, tree, tree, ...]
     # i.e. revisions are 1-based.
     self.svn_revs = [None]
     # Format is { repo: [ None, (hash, tree), (hash, tree), ... ], ... }
     # so reference looks like self.git_hashes[repo][rev][0] for hash and
     # self.git_hashes[repo][rev][1] for it's tree snapshot.
     # For consistency with self.svn_revs, it is 1-based too.
     self.git_hashes = {}
     self.svnserve = None
     self.gitdaemon = None
-    self.common_init = False
-    self.repos_dir = None
     self.git_pid_file = None
     self.git_root = None
     self.svn_checkout = None
     self.svn_repo = None
     self.git_dirty = False
     self.svn_dirty = False
     self.svn_base = 'svn://%s/svn/' % self.host
     self.git_base = 'git://%s/git/' % self.host
     self.svn_port = 3690
     self.git_port = 9418
 
-  def trial_dir(self):
-    if not self.TRIAL_DIR:
-      self.TRIAL_DIR = os.path.join(
-          os.path.dirname(os.path.abspath(__file__)), '_trial')
-    return self.TRIAL_DIR
+  @property
+  def root_dir(self):
+    return self.trial.root_dir
 
   def set_up(self):
-    """All late initialization comes here.
-
-    Note that it deletes all trial_dir() and not only repos_dir.
-    """
+    """All late initialization comes here."""
     self.cleanup_dirt()
-    if not self.common_init:
-      self.common_init = True
-      self.repos_dir = os.path.join(self.trial_dir(), 'repos')
-      self.git_root = join(self.repos_dir, 'git')
-      self.svn_checkout = join(self.repos_dir, 'svn_checkout')
-      self.svn_repo = join(self.repos_dir, 'svn')
-      add_kill()
-      rmtree(self.trial_dir())
-      os.makedirs(self.repos_dir)
-      atexit.register(self.tear_down)
+    if not self.root_dir:
+      try:
+        add_kill()
+        # self.root_dir is not set before this call.
+        self.trial.set_up()
+        self.git_root = join(self.root_dir, 'git')
+        self.svn_checkout = join(self.root_dir, 'svn_checkout')
+        self.svn_repo = join(self.root_dir, 'svn')
+      finally:
+        # Registers cleanup.
+        atexit.register(self.tear_down)
 
   def cleanup_dirt(self):
     """For each dirty repository, destroy it."""
     if self.svn_dirty:
       if not self.tear_down_svn():
         logging.error('Using both leaking checkout and svn dirty checkout')
     if self.git_dirty:
       if not self.tear_down_git():
         logging.error('Using both leaking checkout and git dirty checkout')
 
   def tear_down(self):
     """Kills the servers and delete the directories."""
     self.tear_down_svn()
     self.tear_down_git()
-    if not self.SHOULD_LEAK:
-      logging.debug('Removing %s' % self.trial_dir())
-      rmtree(self.trial_dir())
+    # This deletes the directories.
+    self.trial.tear_down()
+    self.trial = None
 
   def tear_down_svn(self):
     if self.svnserve:
       logging.debug('Killing svnserve pid %s' % self.svnserve.pid)
       self.svnserve.kill()
       self.wait_for_port_to_free(self.svn_port)
       self.svnserve = None
-      if not self.SHOULD_LEAK:
+      if not self.trial.SHOULD_LEAK:
         logging.debug('Removing %s' % self.svn_repo)
         rmtree(self.svn_repo)
         logging.debug('Removing %s' % self.svn_checkout)
         rmtree(self.svn_checkout)
       else:
         return False
     return True
 
   def tear_down_git(self):
     if self.gitdaemon:
       logging.debug('Killing git-daemon pid %s' % self.gitdaemon.pid)
       self.gitdaemon.kill()
       self.gitdaemon = None
       if self.git_pid_file:
         pid = int(self.git_pid_file.read())
         self.git_pid_file.close()
         logging.debug('Killing git daemon pid %s' % pid)
         kill_pid(pid)
         self.git_pid_file = None
       self.wait_for_port_to_free(self.git_port)
-      if not self.SHOULD_LEAK:
+      if not self.trial.SHOULD_LEAK:
         logging.debug('Removing %s' % self.git_root)
         rmtree(self.git_root)
       else:
         return False
     return True
 
   @staticmethod
   def _genTree(root, tree_dict):
     """For a dictionary of file contents, generate a filesystem."""
     if not os.path.isdir(root):
@@ skipping 22 matching lines @@
     write(join(self.svn_repo, 'conf', 'svnserve.conf'),
         '[general]\n'
         'anon-access = read\n'
         'auth-access = write\n'
         'password-db = passwd\n')
     text = '[users]\n'
     text += ''.join('%s = %s\n' % (usr, pwd) for usr, pwd in self.USERS)
     write(join(self.svn_repo, 'conf', 'passwd'), text)
 
     # Start the daemon.
-    cmd = ['svnserve', '-d', '--foreground', '-r', self.repos_dir]
+    cmd = ['svnserve', '-d', '--foreground', '-r', self.root_dir]
     if self.host == '127.0.0.1':
       cmd.append('--listen-host=' + self.host)
     self.check_port_is_free(self.svn_port)
     self.svnserve = Popen(cmd, cwd=self.svn_repo)
     self.wait_for_port_to_bind(self.svn_port, self.svnserve)
     self.populateSvn()
     self.svn_dirty = False
     return True
 
   def set_up_git(self):
     """Creates git repositories and start the servers."""
     self.set_up()
     if self.gitdaemon:
       return True
     if sys.platform == 'win32':
       return False
     assert self.git_pid_file == None
     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]
     # Unlike svn, populate git before starting the server.
     self.populateGit()
     # Start the daemon.
     self.git_pid_file = tempfile.NamedTemporaryFile()
     cmd = ['git', 'daemon',
         '--export-all',
         '--reuseaddr',
-        '--base-path=' + self.repos_dir,
+        '--base-path=' + self.root_dir,
         '--pid-file=' + self.git_pid_file.name]
     if self.host == '127.0.0.1':
       cmd.append('--listen=' + self.host)
     self.check_port_is_free(self.git_port)
-    self.gitdaemon = Popen(cmd, cwd=self.repos_dir)
+    self.gitdaemon = Popen(cmd, cwd=self.root_dir)
     self.wait_for_port_to_bind(self.git_port, self.gitdaemon)
     self.git_dirty = False
     return True
 
   def _commit_svn(self, tree):
     self._genTree(self.svn_checkout, tree)
     commit_svn(self.svn_checkout, self.USERS[0][0], self.USERS[0][1])
     if self.svn_revs and self.svn_revs[-1]:
       new_tree = self.svn_revs[-1].copy()
       new_tree.update(tree)
@@ skipping 268 matching lines @@
         'git_base': self.git_base,
         # 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',
     })
 
 
-class FakeReposTestBase(unittest.TestCase):
+class FakeReposTestBase(trial_dir.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)
+  def setUp(self):
+    super(FakeReposTestBase, self).setUp()
     if not FakeReposTestBase.FAKE_REPOS:
       # Lazy create the global instance.
       FakeReposTestBase.FAKE_REPOS = self.FAKE_REPOS_CLASS()
-
-  def setUp(self):
-    unittest.TestCase.setUp(self)
-    self.FAKE_REPOS.set_up()
-
-    # 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)
-    os.makedirs(self.root_dir)
-
-  def tearDown(self):
-    if not self.FAKE_REPOS.SHOULD_LEAK:
-      rmtree(self.root_dir)
+    # No need to call self.FAKE_REPOS.setUp(), it will be called by the child
+    # class.
+    # Do not define tearDown(), since super's version does the right thing and
+    # FAKE_REPOS is kept across tests.
 
   @property
   def svn_base(self):
     """Shortcut."""
     return self.FAKE_REPOS.svn_base
 
   @property
   def git_base(self):
     """Shortcut."""
     return self.FAKE_REPOS.git_base
@@ skipping 59 matching lines @@
     """Sort-hand: Returns the hash for a git 'revision'."""
     return self.FAKE_REPOS.git_hashes[repo][int(rev)][0]
 
   def gittree(self, repo, rev):
     """Sort-hand: returns the directory tree for a git 'revision'."""
     return self.FAKE_REPOS.git_hashes[repo][int(rev)][1]
 
 
 def main(argv):
   fake = FakeRepos()
-  print 'Using %s' % fake.trial_dir()
+  print 'Using %s' % fake.root_dir
   try:
     fake.set_up_svn()
     fake.set_up_git()
     print('Fake setup, press enter to quit or Ctrl-C to keep the checkouts.')
     sys.stdin.readline()
   except KeyboardInterrupt:
-    fake.SHOULD_LEAK = True
+    trial_dir.TrialDir.SHOULD_LEAK.leak = True
   return 0
 
 
-if '-l' in sys.argv:
-  # See SHOULD_LEAK definition in FakeReposBase for its purpose.
-  FakeReposBase.SHOULD_LEAK = True
-  print 'Leaking!'
-  sys.argv.remove('-l')
-
-
 if __name__ == '__main__':
   sys.exit(main(sys.argv))