mirror of
https://github.com/pypa/pip
synced 2023-12-13 21:30:23 +01:00
63 lines
1.5 KiB
Python
63 lines
1.5 KiB
Python
"""Helpers for tests that check configuration
|
|
"""
|
|
|
|
import contextlib
|
|
import functools
|
|
import os
|
|
import tempfile
|
|
import textwrap
|
|
|
|
import pip.configuration
|
|
from pip.utils import ensure_dir
|
|
|
|
# This is so that tests don't need to import pip.configuration
|
|
kinds = pip.configuration.kinds
|
|
|
|
|
|
class ConfigurationMixin(object):
|
|
|
|
def setup(self):
|
|
self.configuration = pip.configuration.Configuration(isolated=False)
|
|
self._files_to_clear = []
|
|
|
|
self._old_environ = os.environ.copy()
|
|
|
|
def teardown(self):
|
|
for fname in self._files_to_clear:
|
|
fname.stop()
|
|
|
|
os.environ = self._old_environ
|
|
|
|
def patch_configuration(self, variant, di):
|
|
old = self.configuration._load_config_files
|
|
|
|
@functools.wraps(old)
|
|
def overidden():
|
|
# Manual Overload
|
|
self.configuration._config[variant].update(di)
|
|
self.configuration._parsers[variant].append((None, None))
|
|
return old()
|
|
|
|
self.configuration._load_config_files = overidden
|
|
|
|
@contextlib.contextmanager
|
|
def tmpfile(self, contents):
|
|
# Create a temporary file
|
|
fd, path = tempfile.mkstemp(
|
|
prefix="pip_", suffix="_config.ini", text=True
|
|
)
|
|
os.close(fd)
|
|
|
|
contents = textwrap.dedent(contents).lstrip()
|
|
ensure_dir(os.path.dirname(path))
|
|
with open(path, "w") as f:
|
|
f.write(contents)
|
|
|
|
yield path
|
|
|
|
os.remove(path)
|
|
|
|
@staticmethod
|
|
def get_file_contents(path):
|
|
with open(path) as f:
|
|
return f.read()
|