e3d37db45e
As it currently stands, Pegleg clones site repositories into the /tmp directory. Even if the site repository already exists in the /tmp directory it is still cloned there which results in wasted disk space. This commit allows users to pass in a `clone_path` (-p) option to Pegleg CLI commands that specify where to clone a site repository. If the clone path matches the path of an existing repository, then a error message is logged stating so. If the repository already exists in the clone path, the user can either specify to use that local repo by passing it as the site repository or they proceed by passing in a different clone path. This commit also updates the logic that deletes the copy of the repo that is created in the temporary folder to also delete the parent folder that contains the copied repo. This scenario happens when using a local repository as the site repository. Addionally, this commit adds a cleanup fixture that removes files and directories created in the temporary folder by the unit tests. Change-Id: I1b2943493b8f201f337ea60006c009973dd941b3
58 lines
1.8 KiB
Python
58 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="class", 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)
|