annotate tests/common.py @ 227:484b56ac4aec default tip

hglib: cat accepts a template argument

import os, sys, tempfile, shutil
import unittest

import hglib
from hglib import client

def resultappender(list):
    def decorator(f):
        def decorated(*args, **kwargs):
            list.append(args[0])
            return f(*args, **kwargs)
        return decorated
    return decorator

class basetest(unittest.TestCase):
    def setUp(self):
        os.environ['LANG'] = os.environ['LC_ALL'] = os.environ['LANGUAGE'] = 'C'
        os.environ["EMAIL"] = "Foo Bar <[e-mail]>"
        os.environ['CDPATH'] = ''
        os.environ['COLUMNS'] = '80'
        os.environ['GREP_OPTIONS'] = ''
        os.environ['http_proxy'] = ''

        os.environ["HGEDITOR"] = sys.executable + ' -c "import sys; sys.exit(0)"'
        os.environ["HGMERGE"] = "internal:merge"
        os.environ["HGUSER"]   = "test"
        os.environ["HGENCODING"] = "ascii"
        os.environ["HGENCODINGMODE"] = "strict"
        tmpdir = tempfile.mkdtemp('', 'python-hglib.')
        os.environ["HGTMP"] = os.path.realpath(tmpdir)
        os.environ["HGRCPATH"] = os.pathsep

        self._testtmp = os.environ["TESTTMP"] = os.environ["HOME"] = \
            os.path.join(os.environ["HGTMP"], self.__class__.__name__)

        self.clients = []
        self._oldopen = hglib.client.hgclient.open
        # hglib.open = resultappender(self.clients)(hglib.open)
        c = hglib.client.hgclient
        c.open = resultappender(self.clients)(c.open)

        os.mkdir(self._testtmp)
        os.chdir(self._testtmp)
        # until we can run norepo commands in the cmdserver
        os.system('hg init')
        self.client = hglib.open()

    def tearDown(self):
        # on Windows we cannot rmtree before closing all instances
        # because of used files
        hglib.client.hgclient.open = self._oldopen
        for client in self.clients:
            if client.server is not None:
                client.close()
        os.chdir('..')

        try:
            shutil.rmtree(self._testtmp, ignore_errors=True)
        except AttributeError:
            pass # if our setUp was overriden

    def append(self, path, *args):
        f = open(path, 'ab')
        for a in args:
            f.write(a.encode('latin-1'))
        f.close()