test/helper.py

   1 from __future__ import absolute_import, division, unicode_literals
   2 import os
   3 import shutil
   4 import stat
   5 import unittest
   6 import tempfile
   7
   8 import mock
   9
  10 from cola import core
  11 from cola import git
  12 from cola import gitcfg
  13 from cola import gitcmds
  14 from cola.models import main
  15
  16
  17 def tmp_path(*paths):
  18     """Returns a path relative to the test/tmp directory"""
  19     dirname = core.decode(os.path.dirname(__file__))
  20     return os.path.join(dirname, 'tmp', *paths)
  21
  22
  23 def fixture(*paths):
  24     dirname = core.decode(os.path.dirname(__file__))
  25     return os.path.join(dirname, 'fixtures', *paths)
  26
  27
  28 def run_unittest(suite):
  29     return unittest.TextTestRunner(verbosity=2).run(suite)
  30
  31
  32 # shutil.rmtree() can't remove read-only files on Windows.  This onerror
  33 # handler, adapted from <http://stackoverflow.com/a/1889686/357338>, works
  34 # around this by changing such files to be writable and then re-trying.
  35 def remove_readonly(func, path, _exc_info):
  36     if func == os.remove and not os.access(path, os.W_OK):
  37         os.chmod(path, stat.S_IWRITE)
  38         func(path)
  39     else:
  40         raise AssertionError('Should not happen')
  41
  42
  43 class TmpPathTestCase(unittest.TestCase):
  44
  45     def setUp(self):
  46         self._testdir = tempfile.mkdtemp('_cola_test')
  47         os.chdir(self._testdir)
  48
  49     def tearDown(self):
  50         """Remove the test directory and return to the tmp root."""
  51         path = self._testdir
  52         os.chdir(tmp_path())
  53         shutil.rmtree(path, onerror=remove_readonly)
  54
  55     @staticmethod
  56     def touch(*paths):
  57         for path in paths:
  58             open(path, 'a').close()
  59
  60     @staticmethod
  61     def write_file(path, content):
  62         with open(path, 'w') as f:
  63             f.write(content)
  64
  65     @staticmethod
  66     def append_file(path, content):
  67         with open(path, 'a') as f:
  68             f.write(content)
  69
  70     def test_path(self, *paths):
  71         return os.path.join(self._testdir, *paths)
  72
  73
  74 class GitRepositoryTestCase(TmpPathTestCase):
  75     """Tests that operate on temporary git repositories."""
  76
  77     def setUp(self):
  78         TmpPathTestCase.setUp(self)
  79         self.initialize_repo()
  80         self.context = context = mock.Mock()
  81         context.git = git.create()
  82         context.git.set_worktree(core.getcwd())
  83         context.cfg = gitcfg.create(context)
  84         context.model = self.model = main.create(self.context)
  85         self.git = context.git
  86         self.cfg = context.cfg
  87         self.cfg.reset()
  88         gitcmds.reset()
  89
  90     def run_git(self, *args):
  91         status, out, _ = core.run_command(['git'] + list(args))
  92         self.assertEqual(status, 0)
  93         return out
  94
  95     def initialize_repo(self):
  96         self.run_git('init')
  97         self.run_git('config', '--local', 'user.name', 'Your Name')
  98         self.run_git('config', '--local', 'user.email', 'you@example.com')
  99         self.touch('A', 'B')
 100         self.run_git('add', 'A', 'B')
 101
 102     def commit_files(self):
 103         self.run_git('commit', '-m', 'initial commit')