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
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
|
import os
import re
from importlib import import_module
import django
import pytest
from django import __version__ as DJANGO_VERSION
from django.core import management
from tests import is_gis_installed
def pytest_configure(config):
from django.conf import settings
config.addinivalue_line(
"markers", "contrib(name): mark required contrib package"
)
contrib_apps = [
'dj_rest_auth',
'dj_rest_auth.registration',
'allauth',
'allauth.account',
'oauth2_provider',
'django_filters',
'knox',
# this is not strictly required and when added django-polymorphic
# currently breaks the whole Django/DRF upstream testing.
# 'polymorphic',
# 'rest_framework_jwt',
]
# only load GIS if library is installed. This is required for the GIS test to work
if is_gis_installed():
contrib_apps.append('rest_framework_gis')
base_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
settings.configure(
DEBUG_PROPAGATE_EXCEPTIONS=True,
DATABASES={'default': {
'ENGINE': 'django.db.backends.sqlite3',
'NAME': ':memory:'
}},
SITE_ID=1,
SECRET_KEY='not very secret in tests',
USE_I18N=True,
USE_L10N=True,
LANGUAGES=[
('de-de', 'German'),
('en-us', 'English'),
],
LOCALE_PATHS=[
base_dir + '/locale/'
],
STATIC_URL='/static/',
ROOT_URLCONF='tests.urls',
TEMPLATES=[
{
'BACKEND': 'django.template.backends.django.DjangoTemplates',
'DIRS': [],
'APP_DIRS': True,
'OPTIONS': {
'context_processors': [
'django.template.context_processors.debug',
],
},
},
],
MIDDLEWARE=(
'django.contrib.sessions.middleware.SessionMiddleware',
'django.middleware.common.CommonMiddleware',
'django.contrib.auth.middleware.AuthenticationMiddleware',
'django.middleware.locale.LocaleMiddleware',
),
INSTALLED_APPS=(
'django.contrib.auth',
'django.contrib.contenttypes',
'django.contrib.sessions',
'django.contrib.sites',
'django.contrib.messages',
'django.contrib.staticfiles',
'rest_framework',
'rest_framework.authtoken',
*[app for app in contrib_apps if module_available(app)],
'drf_spectacular',
'tests',
),
PASSWORD_HASHERS=(
'django.contrib.auth.hashers.SHA1PasswordHasher',
'django.contrib.auth.hashers.PBKDF2PasswordHasher',
'django.contrib.auth.hashers.PBKDF2SHA1PasswordHasher',
'django.contrib.auth.hashers.BCryptPasswordHasher',
'django.contrib.auth.hashers.MD5PasswordHasher',
'django.contrib.auth.hashers.CryptPasswordHasher',
),
REST_FRAMEWORK={
'DEFAULT_SCHEMA_CLASS': 'drf_spectacular.openapi.AutoSchema',
'PAGE_SIZE': 10,
},
DEFAULT_AUTO_FIELD='django.db.models.AutoField',
SILENCED_SYSTEM_CHECKS=[
'rest_framework.W001', 'fields.E210', 'security.W001', 'security.W002',
'security.W003', 'security.W009', 'security.W012'
],
)
django.setup()
# For whatever reason this works locally without an issue.
# on TravisCI content_type table is missing in the sqlite db as
# if no migration ran, but then why does it work locally?!
management.call_command('migrate')
def pytest_addoption(parser):
parser.addoption(
"--skip-missing-contrib",
action="store_true",
default=False,
help="skip tests depending on missing contrib packages"
)
parser.addoption(
"--allow-contrib-fail",
action="store_true",
default=False,
help="run contrib tests but allow them to fail"
)
parser.addoption(
"--allow-skip-extra-system-req",
action="store_true",
default=False,
help=""
)
def pytest_collection_modifyitems(config, items):
skip_missing_contrib = pytest.mark.skip(reason="skip tests for missing contrib package")
allow_contrib_fail = pytest.mark.xfail(reason="contrib test were allowed to fail")
allow_sys_requirement_fail = pytest.mark.xfail(reason="may fail due to missing system library")
for item in items:
for marker in item.own_markers:
if marker.name == 'contrib' and config.getoption("--skip-missing-contrib"):
if not all([module_available(module_str) for module_str in marker.args]): # pragma: no cover
item.add_marker(skip_missing_contrib)
if marker.name == 'contrib' and config.getoption("--allow-contrib-fail"):
item.add_marker(allow_contrib_fail)
if marker.name == 'system_requirement_fulfilled' and config.getoption("--allow-skip-extra-system-req"):
if not marker.args[0]:
item.add_marker(allow_sys_requirement_fail)
@pytest.fixture()
def no_warnings(capsys):
""" make sure test emits no warnings """
yield capsys
captured = capsys.readouterr()
assert not captured.out
assert not captured.err
@pytest.fixture()
def warnings(capsys):
""" make sure test emits no warnings """
yield capsys
captured = capsys.readouterr()
assert captured.err
@pytest.fixture()
def clear_generator_settings():
from drf_spectacular.drainage import GENERATOR_STATS
yield
GENERATOR_STATS._trace_lineno = False
GENERATOR_STATS._red = GENERATOR_STATS._blue = ''
GENERATOR_STATS._yellow = GENERATOR_STATS._clear = ''
@pytest.fixture()
def clear_caches():
from drf_spectacular.plumbing import _load_enum_name_overrides, get_openapi_type_mapping
_load_enum_name_overrides.cache_clear()
get_openapi_type_mapping.cache_clear()
yield
_load_enum_name_overrides.cache_clear()
get_openapi_type_mapping.cache_clear()
def module_available(module_str):
try:
import_module(module_str)
except ImportError:
return False
else:
return True
@pytest.fixture()
def django_transforms():
def integer_field_sqlite(s):
return re.sub(
r' *maximum: 9223372036854775807\n *minimum: (-9223372036854775808|0)\n *format: int64\n',
'',
s,
flags=re.M
)
if DJANGO_VERSION >= '5':
return [integer_field_sqlite]
else:
return []
|
