# Licensed under the Apache License: http://www.apache.org/licenses/LICENSE-2.0
# For details: https://github.com/nedbat/coveragepy/blob/master/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)
