mirror of
https://chromium.googlesource.com/chromium/tools/depot_tools.git
synced 2026-01-11 10:41:31 +00:00
f38dc929a8
Reformat this dir by itself to help merging with conflicts with other CLs. Reformatted using: parallel ./yapf -i -- testing_support/*.py ~/chromiumos/chromite/contrib/reflow_overlong_comments testing_support/*.py The files that still had strings that were too long were manually reformatted. testing_support/coverage_utils.py testing_support/fake_repos.py testing_support/git_test_utils.py testing_support/presubmit_canned_checks_test_mocks.py Change-Id: I4726a4bbd279a70bcf65d0987fcff0ff9a231386 Reviewed-on: https://chromium-review.googlesource.com/c/chromium/tools/depot_tools/+/4842593 Reviewed-by: Josip Sokcevic <sokcevic@chromium.org> Commit-Queue: Josip Sokcevic <sokcevic@chromium.org> Auto-Submit: Mike Frysinger <vapier@chromium.org>
99 lines
2.9 KiB
Python
99 lines
2.9 KiB
Python
# Copyright (c) 2011 The Chromium Authors. All rights reserved.
|
|
# Use of this source code is governed by a BSD-style license that can be
|
|
# found in the LICENSE file.
|
|
|
|
from __future__ import print_function
|
|
|
|
import atexit
|
|
import logging
|
|
import os
|
|
import sys
|
|
import tempfile
|
|
import unittest
|
|
|
|
import gclient_utils
|
|
|
|
|
|
class TrialDir(object):
|
|
"""Manages a temporary directory.
|
|
|
|
On first object creation, TrialDir.TRIAL_ROOT will be set to a new temporary
|
|
directory created in /tmp or the equivalent. It will be deleted on process
|
|
exit unless TrialDir.SHOULD_LEAK is set to True.
|
|
"""
|
|
# When SHOULD_LEAK is set to True, temporary directories created while the
|
|
# tests are running aren't deleted at the end of the tests. Expect failures
|
|
# when running more than one test due to inter-test side-effects. Helps with
|
|
# debugging.
|
|
SHOULD_LEAK = False
|
|
|
|
# Main root directory.
|
|
TRIAL_ROOT = None
|
|
|
|
def __init__(self, subdir, leak=False):
|
|
self.leak = self.SHOULD_LEAK or leak
|
|
self.subdir = subdir
|
|
self.root_dir = None
|
|
|
|
def set_up(self):
|
|
"""All late initialization comes here."""
|
|
# You can override self.TRIAL_ROOT.
|
|
if not self.TRIAL_ROOT:
|
|
# Was not yet initialized.
|
|
TrialDir.TRIAL_ROOT = os.path.realpath(
|
|
tempfile.mkdtemp(prefix='trial'))
|
|
atexit.register(self._clean)
|
|
self.root_dir = os.path.join(TrialDir.TRIAL_ROOT, self.subdir)
|
|
gclient_utils.rmtree(self.root_dir)
|
|
os.makedirs(self.root_dir)
|
|
|
|
def tear_down(self):
|
|
"""Cleans the trial subdirectory for this instance."""
|
|
if not self.leak:
|
|
logging.debug('Removing %s' % self.root_dir)
|
|
gclient_utils.rmtree(self.root_dir)
|
|
else:
|
|
logging.error('Leaking %s' % self.root_dir)
|
|
self.root_dir = None
|
|
|
|
@staticmethod
|
|
def _clean():
|
|
"""Cleans the root trial directory."""
|
|
if not TrialDir.SHOULD_LEAK:
|
|
logging.debug('Removing %s' % TrialDir.TRIAL_ROOT)
|
|
gclient_utils.rmtree(TrialDir.TRIAL_ROOT)
|
|
else:
|
|
logging.error('Leaking %s' % TrialDir.TRIAL_ROOT)
|
|
|
|
|
|
class TrialDirMixIn(object):
|
|
def setUp(self):
|
|
# Create a specific directory just for the test.
|
|
self.trial = TrialDir(self.id())
|
|
self.trial.set_up()
|
|
|
|
def tearDown(self):
|
|
self.trial.tear_down()
|
|
|
|
@property
|
|
def root_dir(self):
|
|
return self.trial.root_dir
|
|
|
|
|
|
class TestCase(unittest.TestCase, TrialDirMixIn):
|
|
"""Base unittest class that cleans off a trial directory in tearDown()."""
|
|
def setUp(self):
|
|
unittest.TestCase.setUp(self)
|
|
TrialDirMixIn.setUp(self)
|
|
|
|
def tearDown(self):
|
|
TrialDirMixIn.tearDown(self)
|
|
unittest.TestCase.tearDown(self)
|
|
|
|
|
|
if '-l' in sys.argv:
|
|
# See SHOULD_LEAK definition in TrialDir for its purpose.
|
|
TrialDir.SHOULD_LEAK = True
|
|
print('Leaking!')
|
|
sys.argv.remove('-l')
|