mirror of https://github.com/ansible/ansible.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
"""Enable unit testing of Ansible collections. PYTEST_DONT_REWRITE"""
|
|
from __future__ import (absolute_import, division, print_function)
|
|
__metaclass__ = type
|
|
|
|
import os
|
|
import sys
|
|
|
|
# set by ansible-test to a single directory, rather than a list of directories as supported by Ansible itself
|
|
ANSIBLE_COLLECTIONS_PATH = os.path.join(os.environ['ANSIBLE_COLLECTIONS_PATHS'], 'ansible_collections')
|
|
|
|
|
|
def collection_pypkgpath(self):
|
|
"""Configure the Python package path so that pytest can find our collections."""
|
|
for parent in self.parts(reverse=True):
|
|
if str(parent) == ANSIBLE_COLLECTIONS_PATH:
|
|
return parent
|
|
|
|
raise Exception('File "%s" not found in collection path "%s".' % (self.strpath, ANSIBLE_COLLECTIONS_PATH))
|
|
|
|
|
|
def pytest_configure():
|
|
"""Configure this pytest plugin."""
|
|
try:
|
|
if pytest_configure.executed:
|
|
return
|
|
except AttributeError:
|
|
pytest_configure.executed = True
|
|
|
|
from ansible.utils.collection_loader import AnsibleCollectionLoader
|
|
|
|
# allow unit tests to import code from collections
|
|
sys.meta_path.insert(0, AnsibleCollectionLoader())
|
|
|
|
# noinspection PyProtectedMember
|
|
import py._path.local
|
|
|
|
# force collections unit tests to be loaded with the ansible_collections namespace
|
|
# original idea from https://stackoverflow.com/questions/50174130/how-do-i-pytest-a-project-using-pep-420-namespace-packages/50175552#50175552
|
|
# noinspection PyProtectedMember
|
|
py._path.local.LocalPath.pypkgpath = collection_pypkgpath # pylint: disable=protected-access
|
|
|
|
|
|
pytest_configure()
|