1. Jason Pellerin
  2. nose2


nose2 / nose2 / tests / _common.py

"""Common functionality."""
import unittest2.loader
import os.path
import tempfile
import shutil
import sys
import subprocess

HERE = os.path.dirname(__file__)
ROOT = os.path.join(HERE, '..', '..')
SUPPORT = os.path.join(ROOT, 'support')

class TestCase(unittest2.TestCase):
    """TestCase extension.

    If the class variable _RUN_IN_TEMP is True (default: False), tests will be
    performed in a temporary directory, which is deleted afterwards.
    _RUN_IN_TEMP = False

    def setUp(self):
        super(TestCase, self).setUp()

        if self._RUN_IN_TEMP:
            self._orig_dir = os.getcwd()
            work_dir = self._work_dir = tempfile.mkdtemp()
            # Make sure it's possible to import modules from current directory
            sys.path.insert(0, work_dir)

    def tearDown(self):
        super(TestCase, self).tearDown()

        if self._RUN_IN_TEMP:
            shutil.rmtree(self._work_dir, ignore_errors=True)

class FunctionalTestCase(unittest2.TestCase):

    tags = ['functional']

    def assertTestRunOutputMatches(self, proc, stdout=None, stderr=None):
        cmd_stdout, cmd_stderr = proc.communicate()
        if stdout:
            self.assertRegexpMatches(cmd_stdout, stdout)
        if stderr:
            self.assertRegexpMatches(cmd_stderr, stderr)

    def runIn(self, testdir, *args):
        cmd = ['nose2'] + list(args)
        proc = subprocess.Popen(cmd,
                                cwd=os.path.join(SUPPORT, testdir),
        return proc

class _FakeEventBase(object):
    """Baseclass for fake Events."""
    def __init__(self):
        self._fake_messages = []

    def message(self, msg, verbosity=(1, 2)):
        self._fake_messages.append((msg, verbosity))

class FakeHandleFileEvent(_FakeEventBase):
    """Fake HandleFileEvent."""
    def __init__(self, name):
        super(FakeHandleFileEvent, self).__init__()

        self.loader = unittest2.loader.TestLoader()
        self.loader._top_level_dir = os.getcwd()
        self.name = name
        self.path = os.path.split(name)[1]
        self.extraTests = []

class FakeStartTestEvent(_FakeEventBase):
    """Fake StartTestEvent."""
    def __init__(self, test):
        super(FakeStartTestEvent, self).__init__()
        self.test = test
        self.result = test.defaultTestResult()
        import time
        self.startTime = time.time()

class FakeLoadFromNameEvent(_FakeEventBase):
    """Fake LoadFromNameEvent."""
    def __init__(self, name):
        super(FakeLoadFromNameEvent, self).__init__()
        self.name = name

class FakeLoadFromNamesEvent(_FakeEventBase):
    """Fake LoadFromNamesEvent."""
    def __init__(self, names):
        super(FakeLoadFromNamesEvent, self).__init__()
        self.names = names

class FakeStartTestRunEvent(_FakeEventBase):
    """Fake StartTestRunEvent"""
    def __init__(self, runner=None, suite=None, result=None, startTime=None,
        super(FakeStartTestRunEvent, self).__init__()
        self.suite = suite
        self.runner = runner
        self.result = result
        self.startTime = startTime
        self.executeTests = executeTests

class Stub(object):
    """Stub object for use in tests"""