Use correct interpreter for ansible-test injector.
This commit is contained in:
parent
c25db4ee8d
commit
eec21a3d12
2 changed files with 34 additions and 12 deletions
|
@ -346,7 +346,7 @@ def command_network_integration(args):
|
|||
instances = [] # type: list [lib.thread.WrappedThread]
|
||||
|
||||
if args.platform:
|
||||
get_coverage_path(args) # initialize before starting threads
|
||||
get_coverage_path(args, args.python_version, args.python_executable) # initialize before starting threads
|
||||
|
||||
configs = dict((config['platform_version'], config) for config in args.metadata.instance_config)
|
||||
|
||||
|
@ -514,7 +514,7 @@ def command_windows_integration(args):
|
|||
httptester_id = None
|
||||
|
||||
if args.windows:
|
||||
get_coverage_path(args) # initialize before starting threads
|
||||
get_coverage_path(args, args.python_version, args.python_executable) # initialize before starting threads
|
||||
|
||||
configs = dict((config['platform_version'], config) for config in args.metadata.instance_config)
|
||||
|
||||
|
|
|
@ -39,8 +39,7 @@ except ImportError:
|
|||
from configparser import ConfigParser
|
||||
|
||||
DOCKER_COMPLETION = {}
|
||||
|
||||
coverage_path = '' # pylint: disable=locally-disabled, invalid-name
|
||||
COVERAGE_PATHS = {} # type: dict[str, str]
|
||||
|
||||
|
||||
def get_docker_completion():
|
||||
|
@ -196,10 +195,10 @@ def intercept_command(args, cmd, target_name, capture=False, env=None, data=None
|
|||
env = common_environment()
|
||||
|
||||
cmd = list(cmd)
|
||||
inject_path = get_coverage_path(args)
|
||||
config_path = os.path.join(inject_path, 'injector.json')
|
||||
version = python_version or args.python_version
|
||||
interpreter = find_python(version, path)
|
||||
inject_path = get_coverage_path(args, version, interpreter)
|
||||
config_path = os.path.join(inject_path, 'injector.json')
|
||||
coverage_file = os.path.abspath(os.path.join(inject_path, '..', 'output', '%s=%s=%s=%s=coverage' % (
|
||||
args.command, target_name, args.coverage_label or 'local-%s' % version, 'python-%s' % version)))
|
||||
|
||||
|
@ -223,17 +222,19 @@ def intercept_command(args, cmd, target_name, capture=False, env=None, data=None
|
|||
return run_command(args, cmd, capture=capture, env=env, data=data, cwd=cwd)
|
||||
|
||||
|
||||
def get_coverage_path(args):
|
||||
def get_coverage_path(args, version, interpreter):
|
||||
"""
|
||||
:type args: TestConfig
|
||||
:type version: str
|
||||
:type interpreter: str
|
||||
:rtype: str
|
||||
"""
|
||||
global coverage_path # pylint: disable=locally-disabled, global-statement, invalid-name
|
||||
coverage_path = COVERAGE_PATHS.get(version)
|
||||
|
||||
if coverage_path:
|
||||
return os.path.join(coverage_path, 'coverage')
|
||||
|
||||
prefix = 'ansible-test-coverage-'
|
||||
prefix = 'ansible-test-coverage-python-%s-' % version
|
||||
tmp_dir = '/tmp'
|
||||
|
||||
if args.explain:
|
||||
|
@ -247,6 +248,15 @@ def get_coverage_path(args):
|
|||
shutil.copytree(src, os.path.join(coverage_path, 'coverage'))
|
||||
shutil.copy('.coveragerc', os.path.join(coverage_path, 'coverage', '.coveragerc'))
|
||||
|
||||
injector = os.path.join(coverage_path, 'coverage', 'injector.py')
|
||||
|
||||
with open(injector, 'r+') as injector_fd:
|
||||
code = injector_fd.read()
|
||||
code = re.sub(r'^#!.*', '#!%s' % interpreter, code, count=1)
|
||||
injector_fd.seek(0)
|
||||
injector_fd.write(code)
|
||||
injector_fd.truncate()
|
||||
|
||||
for root, dir_names, file_names in os.walk(coverage_path):
|
||||
for name in dir_names + file_names:
|
||||
os.chmod(os.path.join(root, name), stat.S_IRWXU | stat.S_IRGRP | stat.S_IXGRP | stat.S_IROTH | stat.S_IXOTH)
|
||||
|
@ -255,13 +265,25 @@ def get_coverage_path(args):
|
|||
os.mkdir(os.path.join(coverage_path, directory))
|
||||
os.chmod(os.path.join(coverage_path, directory), stat.S_IRWXU | stat.S_IRWXG | stat.S_IRWXO)
|
||||
|
||||
atexit.register(cleanup_coverage_dir)
|
||||
if not COVERAGE_PATHS:
|
||||
atexit.register(cleanup_coverage_dirs)
|
||||
|
||||
COVERAGE_PATHS[version] = coverage_path
|
||||
|
||||
return os.path.join(coverage_path, 'coverage')
|
||||
|
||||
|
||||
def cleanup_coverage_dir():
|
||||
"""Copy over coverage data from temporary directory and purge temporary directory."""
|
||||
def cleanup_coverage_dirs():
|
||||
"""Clean up all coverage directories."""
|
||||
for version, path in COVERAGE_PATHS.items():
|
||||
display.info('Cleaning up coverage directory for Python %s: %s' % (version, path), verbosity=2)
|
||||
cleanup_coverage_dir(path)
|
||||
|
||||
|
||||
def cleanup_coverage_dir(coverage_path):
|
||||
"""Copy over coverage data from temporary directory and purge temporary directory.
|
||||
:type coverage_path: str
|
||||
"""
|
||||
output_dir = os.path.join(coverage_path, 'output')
|
||||
|
||||
for filename in os.listdir(output_dir):
|
||||
|
|
Loading…
Reference in a new issue