summaryrefslogtreecommitdiff
path: root/tests/goldtest.py
blob: baaa8f01d6cc24ddafeddd21e969e8885b1999e4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
# Licensed under the Apache License: http://www.apache.org/licenses/LICENSE-2.0
# For details: https://bitbucket.org/ned/coveragepy/src/default/NOTICE.txt

"""A test base class for tests based on gold file comparison."""

import os
import sys

from unittest_mixins import change_dir    # pylint: disable=unused-import

from tests.coveragetest import CoverageTest
from tests.test_farm import clean
# Import helpers, eventually test_farm.py will go away.
from tests.test_farm import (       # pylint: disable=unused-import
    compare, contains, doesnt_contain, contains_any,
)


class CoverageGoldTest(CoverageTest):
    """A test based on gold files."""

    run_in_temp_dir = False

    def setUp(self):
        super(CoverageGoldTest, self).setUp()
        self.chdir(self.root_dir)
        # Modules should be importable from the current directory.
        sys.path.insert(0, '')

    def output_dir(self, the_dir):
        """Declare where the output directory is.

        The output directory is deleted at the end of the test, unless the
        COVERAGE_KEEP_OUTPUT environment variable is set.

        """
        # To make sure tests are isolated, we always clean the directory at the
        # beginning of the test.
        clean(the_dir)

        if not os.environ.get("COVERAGE_KEEP_OUTPUT"):      # pragma: part covered
            self.addCleanup(clean, the_dir)