tests: add test.support.venv.VirtualEnvironmentMixin (#129461)

This commit is contained in:
Filipe Laíns 🇵🇸 2025-01-30 03:32:24 +00:00 committed by GitHub
parent a549f43938
commit a4722449ca
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 14 additions and 9 deletions

View file

@ -6,6 +6,7 @@ import shlex
import sys
import sysconfig
import tempfile
import unittest
import venv
@ -68,3 +69,14 @@ class VirtualEnvironment:
raise
else:
return result
class VirtualEnvironmentMixin:
def venv(self, name=None, **venv_create_args):
venv_name = self.id()
if name:
venv_name += f'-{name}'
return VirtualEnvironment.from_tmpdir(
prefix=f'{venv_name}-venv-',
**venv_create_args,
)

View file

@ -20,7 +20,7 @@ from test.support import (
from test.support.import_helper import import_module
from test.support.os_helper import (TESTFN, unlink, skip_unless_symlink,
change_cwd)
from test.support.venv import VirtualEnvironment
from test.support.venv import VirtualEnvironmentMixin
import sysconfig
from sysconfig import (get_paths, get_platform, get_config_vars,
@ -37,7 +37,7 @@ import _sysconfig
HAS_USER_BASE = sysconfig._HAS_USER_BASE
class TestSysConfig(unittest.TestCase):
class TestSysConfig(unittest.TestCase, VirtualEnvironmentMixin):
def setUp(self):
super(TestSysConfig, self).setUp()
@ -111,13 +111,6 @@ class TestSysConfig(unittest.TestCase):
elif os.path.isdir(path):
shutil.rmtree(path)
def venv(self, **venv_create_args):
return VirtualEnvironment.from_tmpdir(
prefix=f'{self.id()}-venv-',
**venv_create_args,
)
def test_get_path_names(self):
self.assertEqual(get_path_names(), sysconfig._SCHEME_KEYS)