1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
|
import collections
import functools
import importlib
import inspect
import os
import six
import sys
import pytest
from library.python.pytest.plugins.fixtures import metrics, links # noqa
from library.python.pytest import module_utils
import yatest.common
ConftestInfo = collections.namedtuple('ConftestInfo', ['module_name', 'proper_path'])
LoadedConftestInfo = collections.namedtuple('LoadedConftestInfo', ['conftest_dir', 'module'])
def _patch_inspect_getfile():
orig_getfile = inspect.getfile
def getfile(object):
if inspect.ismodule(object) and getattr(object, '__orig_file__', None):
res = object.__orig_file__
else:
res = orig_getfile(object)
return res
inspect.getfile = getfile
def _consider_conftest(pluginmanager, module, registration_name):
if six.PY3:
pluginmanager.consider_conftest(module, registration_name=registration_name)
else:
pluginmanager.consider_conftest(module)
def _import_conftest(pluginmanager, module_name, proper_path):
"""Essentially PytestPluginManager._importconftest, adapted to resource-based conftest modules."""
module = importlib.import_module(module_name)
if six.PY3:
pluginmanager._check_non_top_pytest_plugins(module, proper_path)
_consider_conftest(pluginmanager, module, proper_path)
conftest_info = LoadedConftestInfo(conftest_dir=os.path.dirname(proper_path), module=module)
pluginmanager._ya_loaded_conftest_infos.append(conftest_info)
def _getconftestmodules_local(self, path):
"""
_getconftestmodules returns a cached list of conftest modules that relate to the given test path.
This list is then used by pytest to temporarily mute other non-initial conftests when processing those tests.
We patch pytest's implementation to avoid fs access when discovering conftests and in is_file() checks.
"""
path = str(path)
if path.endswith('.py'):
path = os.path.dirname(path)
cache = self._ya_getconftestmodules_cache
if path not in cache:
cache[path] = [
conftest_info.module
for conftest_info in self._ya_loaded_conftest_infos
if module_utils.is_relative_to(path, conftest_info.conftest_dir)
]
return cache[path]
def _getconftestmodules_non_local(path):
return _conftest_modules
_conftest_modules = []
@pytest.hookimpl(trylast=True)
def pytest_load_initial_conftests(early_config):
pluginmanager = early_config.pluginmanager
conftests = filter(lambda name: name.endswith('.conftest'), getattr(sys, 'extra_modules', []))
if os.getenv('CONFTEST_LOAD_POLICY') == 'LOCAL':
test_dir = str(yatest.common.context.project_path)
initial_conftests = []
non_initial_conftests = []
for module_name in conftests:
proper_path = module_utils.get_proper_module_path(module_name)
assert proper_path is not None
conftest_dir = os.path.dirname(proper_path)
# There are two kinds of relevant conftests:
# - Initial: conftest is in the directory of the test suite or in a parent directory (applies to all tests)
# - Non-initial: conftest is strictly within the test directory
if module_utils.is_relative_to(test_dir, conftest_dir):
initial_conftests.append(ConftestInfo(module_name=module_name, proper_path=proper_path))
elif module_utils.is_relative_to(conftest_dir, test_dir):
non_initial_conftests.append(ConftestInfo(module_name=module_name, proper_path=proper_path))
# Ensure parent conftests are loaded before child conftests
initial_conftests.sort(key=lambda x: len(x.proper_path))
non_initial_conftests.sort(key=lambda x: len(x.proper_path))
pluginmanager._ya_loaded_conftest_infos = []
pluginmanager._ya_getconftestmodules_cache = {}
for conftest_info in initial_conftests:
_import_conftest(pluginmanager, conftest_info.module_name, conftest_info.proper_path)
pluginmanager._getconftestmodules = functools.partial(_getconftestmodules_local, pluginmanager)
def _ya_register_non_initial_conftests():
for conftest_info in non_initial_conftests:
_import_conftest(pluginmanager, conftest_info.module_name, conftest_info.proper_path)
# See collection.py, non-initial conftests are registered at the start of collection phase, like in vanilla.
# Alas, stash is not available in py2.
early_config._ya_register_non_initial_conftests = _ya_register_non_initial_conftests
else:
_patch_inspect_getfile()
def conftest_key(name):
if not name.startswith('__tests__.'):
# Make __tests__ come last
return '_.' + name
return name
for name in sorted(conftests, key=conftest_key):
module = importlib.import_module(name)
setattr(module, '__orig_file__', module.__file__)
module.__file__ = ''
_conftest_modules.append(module)
_consider_conftest(pluginmanager, module, module.__file__)
pluginmanager._getconftestmodules = _getconftestmodules_non_local
|
