b50619b06d
This patch set refactors uses of tempfile.mkdtemp() used throughout many tests in Pegleg which leaves lingering temporary directories around. Recently a fixture was introduced in [0] which automatically cleans up after itself. This patch set applies the fixture everywhere possible to minimize the testing footprint. [0] https://review.openstack.org/#/c/609818/20/tests/unit/fixtures.py Change-Id: Id4c1195c4f248b974a5396a429d651943a84ee83
59 lines
1.8 KiB
Python
59 lines
1.8 KiB
Python
# Copyright 2018 AT&T Intellectual Property. All other rights reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
import copy
|
|
|
|
import os
|
|
import pytest
|
|
import shutil
|
|
import tempfile
|
|
|
|
from pegleg import config
|
|
"""Fixtures that are applied to all unit tests."""
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def restore_config():
|
|
"""Used for ensuring the original global context is reset in memory
|
|
following each test execution.
|
|
"""
|
|
original_global_context = copy.deepcopy(config.GLOBAL_CONTEXT)
|
|
try:
|
|
yield
|
|
finally:
|
|
config.GLOBAL_CONTEXT = original_global_context
|
|
|
|
|
|
@pytest.fixture(scope="module", autouse=True)
|
|
def clean_temporary_git_repos():
|
|
"""Iterates through all temporarily created directories and deletes each
|
|
one that was created for testing.
|
|
|
|
"""
|
|
|
|
def temporary_git_repos():
|
|
root_tempdir = tempfile.gettempdir()
|
|
tempdirs = os.listdir(root_tempdir)
|
|
for tempdir in tempdirs:
|
|
path = os.path.join(root_tempdir, tempdir)
|
|
if os.path.isdir(path) and os.access(path, os.R_OK):
|
|
if any(p.startswith('airship') for p in os.listdir(path)):
|
|
yield path
|
|
|
|
try:
|
|
yield
|
|
finally:
|
|
for tempdir in temporary_git_repos():
|
|
shutil.rmtree(tempdir, ignore_errors=True)
|