venv/cros_venv/testcases.py

# Copyright 2017 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""TestCase classes"""

from __future__ import absolute_import
from __future__ import print_function
from __future__ import unicode_literals

import shutil
import tempfile
import os
import unittest


class TmpdirTestCase(unittest.TestCase):
    """TestCase subclass providing a tmpdir fixture."""

    def setUp(self):
        self.tmpdir = tempfile.mkdtemp()
        self._old_cwd = os.getcwd()
        os.chdir(self.tmpdir)

    def tearDown(self):
        os.chdir(self._old_cwd)
        shutil.rmtree(self.tmpdir)