##// END OF EJS Templates
tests: allow specifing repo_type in user_util create_repo
marcink -
r1569:037619e0 default
parent child Browse files
Show More
@@ -1,1808 +1,1809 b''
1 # -*- coding: utf-8 -*-
1 # -*- coding: utf-8 -*-
2
2
3 # Copyright (C) 2010-2017 RhodeCode GmbH
3 # Copyright (C) 2010-2017 RhodeCode GmbH
4 #
4 #
5 # This program is free software: you can redistribute it and/or modify
5 # This program is free software: you can redistribute it and/or modify
6 # it under the terms of the GNU Affero General Public License, version 3
6 # it under the terms of the GNU Affero General Public License, version 3
7 # (only), as published by the Free Software Foundation.
7 # (only), as published by the Free Software Foundation.
8 #
8 #
9 # This program is distributed in the hope that it will be useful,
9 # This program is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 # GNU General Public License for more details.
12 # GNU General Public License for more details.
13 #
13 #
14 # You should have received a copy of the GNU Affero General Public License
14 # You should have received a copy of the GNU Affero General Public License
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
16 #
16 #
17 # This program is dual-licensed. If you wish to learn more about the
17 # This program is dual-licensed. If you wish to learn more about the
18 # RhodeCode Enterprise Edition, including its added features, Support services,
18 # RhodeCode Enterprise Edition, including its added features, Support services,
19 # and proprietary license terms, please see https://rhodecode.com/licenses/
19 # and proprietary license terms, please see https://rhodecode.com/licenses/
20
20
21 import collections
21 import collections
22 import datetime
22 import datetime
23 import hashlib
23 import hashlib
24 import os
24 import os
25 import re
25 import re
26 import pprint
26 import pprint
27 import shutil
27 import shutil
28 import socket
28 import socket
29 import subprocess32
29 import subprocess32
30 import time
30 import time
31 import uuid
31 import uuid
32 import dateutil.tz
32 import dateutil.tz
33
33
34 import mock
34 import mock
35 import pyramid.testing
35 import pyramid.testing
36 import pytest
36 import pytest
37 import colander
37 import colander
38 import requests
38 import requests
39
39
40 import rhodecode
40 import rhodecode
41 from rhodecode.lib.utils2 import AttributeDict
41 from rhodecode.lib.utils2 import AttributeDict
42 from rhodecode.model.changeset_status import ChangesetStatusModel
42 from rhodecode.model.changeset_status import ChangesetStatusModel
43 from rhodecode.model.comment import CommentsModel
43 from rhodecode.model.comment import CommentsModel
44 from rhodecode.model.db import (
44 from rhodecode.model.db import (
45 PullRequest, Repository, RhodeCodeSetting, ChangesetStatus, RepoGroup,
45 PullRequest, Repository, RhodeCodeSetting, ChangesetStatus, RepoGroup,
46 UserGroup, RepoRhodeCodeUi, RepoRhodeCodeSetting, RhodeCodeUi)
46 UserGroup, RepoRhodeCodeUi, RepoRhodeCodeSetting, RhodeCodeUi)
47 from rhodecode.model.meta import Session
47 from rhodecode.model.meta import Session
48 from rhodecode.model.pull_request import PullRequestModel
48 from rhodecode.model.pull_request import PullRequestModel
49 from rhodecode.model.repo import RepoModel
49 from rhodecode.model.repo import RepoModel
50 from rhodecode.model.repo_group import RepoGroupModel
50 from rhodecode.model.repo_group import RepoGroupModel
51 from rhodecode.model.user import UserModel
51 from rhodecode.model.user import UserModel
52 from rhodecode.model.settings import VcsSettingsModel
52 from rhodecode.model.settings import VcsSettingsModel
53 from rhodecode.model.user_group import UserGroupModel
53 from rhodecode.model.user_group import UserGroupModel
54 from rhodecode.model.integration import IntegrationModel
54 from rhodecode.model.integration import IntegrationModel
55 from rhodecode.integrations import integration_type_registry
55 from rhodecode.integrations import integration_type_registry
56 from rhodecode.integrations.types.base import IntegrationTypeBase
56 from rhodecode.integrations.types.base import IntegrationTypeBase
57 from rhodecode.lib.utils import repo2db_mapper
57 from rhodecode.lib.utils import repo2db_mapper
58 from rhodecode.lib.vcs import create_vcsserver_proxy
58 from rhodecode.lib.vcs import create_vcsserver_proxy
59 from rhodecode.lib.vcs.backends import get_backend
59 from rhodecode.lib.vcs.backends import get_backend
60 from rhodecode.lib.vcs.nodes import FileNode
60 from rhodecode.lib.vcs.nodes import FileNode
61 from rhodecode.tests import (
61 from rhodecode.tests import (
62 login_user_session, get_new_dir, utils, TESTS_TMP_PATH,
62 login_user_session, get_new_dir, utils, TESTS_TMP_PATH,
63 TEST_USER_ADMIN_LOGIN, TEST_USER_REGULAR_LOGIN, TEST_USER_REGULAR2_LOGIN,
63 TEST_USER_ADMIN_LOGIN, TEST_USER_REGULAR_LOGIN, TEST_USER_REGULAR2_LOGIN,
64 TEST_USER_REGULAR_PASS)
64 TEST_USER_REGULAR_PASS)
65 from rhodecode.tests.utils import CustomTestApp
65 from rhodecode.tests.utils import CustomTestApp
66 from rhodecode.tests.fixture import Fixture
66 from rhodecode.tests.fixture import Fixture
67
67
68
68
69 def _split_comma(value):
69 def _split_comma(value):
70 return value.split(',')
70 return value.split(',')
71
71
72
72
73 def pytest_addoption(parser):
73 def pytest_addoption(parser):
74 parser.addoption(
74 parser.addoption(
75 '--keep-tmp-path', action='store_true',
75 '--keep-tmp-path', action='store_true',
76 help="Keep the test temporary directories")
76 help="Keep the test temporary directories")
77 parser.addoption(
77 parser.addoption(
78 '--backends', action='store', type=_split_comma,
78 '--backends', action='store', type=_split_comma,
79 default=['git', 'hg', 'svn'],
79 default=['git', 'hg', 'svn'],
80 help="Select which backends to test for backend specific tests.")
80 help="Select which backends to test for backend specific tests.")
81 parser.addoption(
81 parser.addoption(
82 '--dbs', action='store', type=_split_comma,
82 '--dbs', action='store', type=_split_comma,
83 default=['sqlite'],
83 default=['sqlite'],
84 help="Select which database to test for database specific tests. "
84 help="Select which database to test for database specific tests. "
85 "Possible options are sqlite,postgres,mysql")
85 "Possible options are sqlite,postgres,mysql")
86 parser.addoption(
86 parser.addoption(
87 '--appenlight', '--ae', action='store_true',
87 '--appenlight', '--ae', action='store_true',
88 help="Track statistics in appenlight.")
88 help="Track statistics in appenlight.")
89 parser.addoption(
89 parser.addoption(
90 '--appenlight-api-key', '--ae-key',
90 '--appenlight-api-key', '--ae-key',
91 help="API key for Appenlight.")
91 help="API key for Appenlight.")
92 parser.addoption(
92 parser.addoption(
93 '--appenlight-url', '--ae-url',
93 '--appenlight-url', '--ae-url',
94 default="https://ae.rhodecode.com",
94 default="https://ae.rhodecode.com",
95 help="Appenlight service URL, defaults to https://ae.rhodecode.com")
95 help="Appenlight service URL, defaults to https://ae.rhodecode.com")
96 parser.addoption(
96 parser.addoption(
97 '--sqlite-connection-string', action='store',
97 '--sqlite-connection-string', action='store',
98 default='', help="Connection string for the dbs tests with SQLite")
98 default='', help="Connection string for the dbs tests with SQLite")
99 parser.addoption(
99 parser.addoption(
100 '--postgres-connection-string', action='store',
100 '--postgres-connection-string', action='store',
101 default='', help="Connection string for the dbs tests with Postgres")
101 default='', help="Connection string for the dbs tests with Postgres")
102 parser.addoption(
102 parser.addoption(
103 '--mysql-connection-string', action='store',
103 '--mysql-connection-string', action='store',
104 default='', help="Connection string for the dbs tests with MySQL")
104 default='', help="Connection string for the dbs tests with MySQL")
105 parser.addoption(
105 parser.addoption(
106 '--repeat', type=int, default=100,
106 '--repeat', type=int, default=100,
107 help="Number of repetitions in performance tests.")
107 help="Number of repetitions in performance tests.")
108
108
109
109
110 def pytest_configure(config):
110 def pytest_configure(config):
111 # Appy the kombu patch early on, needed for test discovery on Python 2.7.11
111 # Appy the kombu patch early on, needed for test discovery on Python 2.7.11
112 from rhodecode.config import patches
112 from rhodecode.config import patches
113 patches.kombu_1_5_1_python_2_7_11()
113 patches.kombu_1_5_1_python_2_7_11()
114
114
115
115
116 def pytest_collection_modifyitems(session, config, items):
116 def pytest_collection_modifyitems(session, config, items):
117 # nottest marked, compare nose, used for transition from nose to pytest
117 # nottest marked, compare nose, used for transition from nose to pytest
118 remaining = [
118 remaining = [
119 i for i in items if getattr(i.obj, '__test__', True)]
119 i for i in items if getattr(i.obj, '__test__', True)]
120 items[:] = remaining
120 items[:] = remaining
121
121
122
122
123 def pytest_generate_tests(metafunc):
123 def pytest_generate_tests(metafunc):
124 # Support test generation based on --backend parameter
124 # Support test generation based on --backend parameter
125 if 'backend_alias' in metafunc.fixturenames:
125 if 'backend_alias' in metafunc.fixturenames:
126 backends = get_backends_from_metafunc(metafunc)
126 backends = get_backends_from_metafunc(metafunc)
127 scope = None
127 scope = None
128 if not backends:
128 if not backends:
129 pytest.skip("Not enabled for any of selected backends")
129 pytest.skip("Not enabled for any of selected backends")
130 metafunc.parametrize('backend_alias', backends, scope=scope)
130 metafunc.parametrize('backend_alias', backends, scope=scope)
131 elif hasattr(metafunc.function, 'backends'):
131 elif hasattr(metafunc.function, 'backends'):
132 backends = get_backends_from_metafunc(metafunc)
132 backends = get_backends_from_metafunc(metafunc)
133 if not backends:
133 if not backends:
134 pytest.skip("Not enabled for any of selected backends")
134 pytest.skip("Not enabled for any of selected backends")
135
135
136
136
137 def get_backends_from_metafunc(metafunc):
137 def get_backends_from_metafunc(metafunc):
138 requested_backends = set(metafunc.config.getoption('--backends'))
138 requested_backends = set(metafunc.config.getoption('--backends'))
139 if hasattr(metafunc.function, 'backends'):
139 if hasattr(metafunc.function, 'backends'):
140 # Supported backends by this test function, created from
140 # Supported backends by this test function, created from
141 # pytest.mark.backends
141 # pytest.mark.backends
142 backends = metafunc.function.backends.args
142 backends = metafunc.function.backends.args
143 elif hasattr(metafunc.cls, 'backend_alias'):
143 elif hasattr(metafunc.cls, 'backend_alias'):
144 # Support class attribute "backend_alias", this is mainly
144 # Support class attribute "backend_alias", this is mainly
145 # for legacy reasons for tests not yet using pytest.mark.backends
145 # for legacy reasons for tests not yet using pytest.mark.backends
146 backends = [metafunc.cls.backend_alias]
146 backends = [metafunc.cls.backend_alias]
147 else:
147 else:
148 backends = metafunc.config.getoption('--backends')
148 backends = metafunc.config.getoption('--backends')
149 return requested_backends.intersection(backends)
149 return requested_backends.intersection(backends)
150
150
151
151
152 @pytest.fixture(scope='session', autouse=True)
152 @pytest.fixture(scope='session', autouse=True)
153 def activate_example_rcextensions(request):
153 def activate_example_rcextensions(request):
154 """
154 """
155 Patch in an example rcextensions module which verifies passed in kwargs.
155 Patch in an example rcextensions module which verifies passed in kwargs.
156 """
156 """
157 from rhodecode.tests.other import example_rcextensions
157 from rhodecode.tests.other import example_rcextensions
158
158
159 old_extensions = rhodecode.EXTENSIONS
159 old_extensions = rhodecode.EXTENSIONS
160 rhodecode.EXTENSIONS = example_rcextensions
160 rhodecode.EXTENSIONS = example_rcextensions
161
161
162 @request.addfinalizer
162 @request.addfinalizer
163 def cleanup():
163 def cleanup():
164 rhodecode.EXTENSIONS = old_extensions
164 rhodecode.EXTENSIONS = old_extensions
165
165
166
166
167 @pytest.fixture
167 @pytest.fixture
168 def capture_rcextensions():
168 def capture_rcextensions():
169 """
169 """
170 Returns the recorded calls to entry points in rcextensions.
170 Returns the recorded calls to entry points in rcextensions.
171 """
171 """
172 calls = rhodecode.EXTENSIONS.calls
172 calls = rhodecode.EXTENSIONS.calls
173 calls.clear()
173 calls.clear()
174 # Note: At this moment, it is still the empty dict, but that will
174 # Note: At this moment, it is still the empty dict, but that will
175 # be filled during the test run and since it is a reference this
175 # be filled during the test run and since it is a reference this
176 # is enough to make it work.
176 # is enough to make it work.
177 return calls
177 return calls
178
178
179
179
180 @pytest.fixture(scope='session')
180 @pytest.fixture(scope='session')
181 def http_environ_session():
181 def http_environ_session():
182 """
182 """
183 Allow to use "http_environ" in session scope.
183 Allow to use "http_environ" in session scope.
184 """
184 """
185 return http_environ(
185 return http_environ(
186 http_host_stub=http_host_stub())
186 http_host_stub=http_host_stub())
187
187
188
188
189 @pytest.fixture
189 @pytest.fixture
190 def http_host_stub():
190 def http_host_stub():
191 """
191 """
192 Value of HTTP_HOST in the test run.
192 Value of HTTP_HOST in the test run.
193 """
193 """
194 return 'test.example.com:80'
194 return 'test.example.com:80'
195
195
196
196
197 @pytest.fixture
197 @pytest.fixture
198 def http_environ(http_host_stub):
198 def http_environ(http_host_stub):
199 """
199 """
200 HTTP extra environ keys.
200 HTTP extra environ keys.
201
201
202 User by the test application and as well for setting up the pylons
202 User by the test application and as well for setting up the pylons
203 environment. In the case of the fixture "app" it should be possible
203 environment. In the case of the fixture "app" it should be possible
204 to override this for a specific test case.
204 to override this for a specific test case.
205 """
205 """
206 return {
206 return {
207 'SERVER_NAME': http_host_stub.split(':')[0],
207 'SERVER_NAME': http_host_stub.split(':')[0],
208 'SERVER_PORT': http_host_stub.split(':')[1],
208 'SERVER_PORT': http_host_stub.split(':')[1],
209 'HTTP_HOST': http_host_stub,
209 'HTTP_HOST': http_host_stub,
210 'HTTP_USER_AGENT': 'rc-test-agent',
210 'HTTP_USER_AGENT': 'rc-test-agent',
211 'REQUEST_METHOD': 'GET'
211 'REQUEST_METHOD': 'GET'
212 }
212 }
213
213
214
214
215 @pytest.fixture(scope='function')
215 @pytest.fixture(scope='function')
216 def app(request, pylonsapp, http_environ):
216 def app(request, pylonsapp, http_environ):
217 app = CustomTestApp(
217 app = CustomTestApp(
218 pylonsapp,
218 pylonsapp,
219 extra_environ=http_environ)
219 extra_environ=http_environ)
220 if request.cls:
220 if request.cls:
221 request.cls.app = app
221 request.cls.app = app
222 return app
222 return app
223
223
224
224
225 @pytest.fixture(scope='session')
225 @pytest.fixture(scope='session')
226 def app_settings(pylonsapp, pylons_config):
226 def app_settings(pylonsapp, pylons_config):
227 """
227 """
228 Settings dictionary used to create the app.
228 Settings dictionary used to create the app.
229
229
230 Parses the ini file and passes the result through the sanitize and apply
230 Parses the ini file and passes the result through the sanitize and apply
231 defaults mechanism in `rhodecode.config.middleware`.
231 defaults mechanism in `rhodecode.config.middleware`.
232 """
232 """
233 from paste.deploy.loadwsgi import loadcontext, APP
233 from paste.deploy.loadwsgi import loadcontext, APP
234 from rhodecode.config.middleware import (
234 from rhodecode.config.middleware import (
235 sanitize_settings_and_apply_defaults)
235 sanitize_settings_and_apply_defaults)
236 context = loadcontext(APP, 'config:' + pylons_config)
236 context = loadcontext(APP, 'config:' + pylons_config)
237 settings = sanitize_settings_and_apply_defaults(context.config())
237 settings = sanitize_settings_and_apply_defaults(context.config())
238 return settings
238 return settings
239
239
240
240
241 @pytest.fixture(scope='session')
241 @pytest.fixture(scope='session')
242 def db(app_settings):
242 def db(app_settings):
243 """
243 """
244 Initializes the database connection.
244 Initializes the database connection.
245
245
246 It uses the same settings which are used to create the ``pylonsapp`` or
246 It uses the same settings which are used to create the ``pylonsapp`` or
247 ``app`` fixtures.
247 ``app`` fixtures.
248 """
248 """
249 from rhodecode.config.utils import initialize_database
249 from rhodecode.config.utils import initialize_database
250 initialize_database(app_settings)
250 initialize_database(app_settings)
251
251
252
252
253 LoginData = collections.namedtuple('LoginData', ('csrf_token', 'user'))
253 LoginData = collections.namedtuple('LoginData', ('csrf_token', 'user'))
254
254
255
255
256 def _autologin_user(app, *args):
256 def _autologin_user(app, *args):
257 session = login_user_session(app, *args)
257 session = login_user_session(app, *args)
258 csrf_token = rhodecode.lib.auth.get_csrf_token(session)
258 csrf_token = rhodecode.lib.auth.get_csrf_token(session)
259 return LoginData(csrf_token, session['rhodecode_user'])
259 return LoginData(csrf_token, session['rhodecode_user'])
260
260
261
261
262 @pytest.fixture
262 @pytest.fixture
263 def autologin_user(app):
263 def autologin_user(app):
264 """
264 """
265 Utility fixture which makes sure that the admin user is logged in
265 Utility fixture which makes sure that the admin user is logged in
266 """
266 """
267 return _autologin_user(app)
267 return _autologin_user(app)
268
268
269
269
270 @pytest.fixture
270 @pytest.fixture
271 def autologin_regular_user(app):
271 def autologin_regular_user(app):
272 """
272 """
273 Utility fixture which makes sure that the regular user is logged in
273 Utility fixture which makes sure that the regular user is logged in
274 """
274 """
275 return _autologin_user(
275 return _autologin_user(
276 app, TEST_USER_REGULAR_LOGIN, TEST_USER_REGULAR_PASS)
276 app, TEST_USER_REGULAR_LOGIN, TEST_USER_REGULAR_PASS)
277
277
278
278
279 @pytest.fixture(scope='function')
279 @pytest.fixture(scope='function')
280 def csrf_token(request, autologin_user):
280 def csrf_token(request, autologin_user):
281 return autologin_user.csrf_token
281 return autologin_user.csrf_token
282
282
283
283
284 @pytest.fixture(scope='function')
284 @pytest.fixture(scope='function')
285 def xhr_header(request):
285 def xhr_header(request):
286 return {'HTTP_X_REQUESTED_WITH': 'XMLHttpRequest'}
286 return {'HTTP_X_REQUESTED_WITH': 'XMLHttpRequest'}
287
287
288
288
289 @pytest.fixture
289 @pytest.fixture
290 def real_crypto_backend(monkeypatch):
290 def real_crypto_backend(monkeypatch):
291 """
291 """
292 Switch the production crypto backend on for this test.
292 Switch the production crypto backend on for this test.
293
293
294 During the test run the crypto backend is replaced with a faster
294 During the test run the crypto backend is replaced with a faster
295 implementation based on the MD5 algorithm.
295 implementation based on the MD5 algorithm.
296 """
296 """
297 monkeypatch.setattr(rhodecode, 'is_test', False)
297 monkeypatch.setattr(rhodecode, 'is_test', False)
298
298
299
299
300 @pytest.fixture(scope='class')
300 @pytest.fixture(scope='class')
301 def index_location(request, pylonsapp):
301 def index_location(request, pylonsapp):
302 index_location = pylonsapp.config['app_conf']['search.location']
302 index_location = pylonsapp.config['app_conf']['search.location']
303 if request.cls:
303 if request.cls:
304 request.cls.index_location = index_location
304 request.cls.index_location = index_location
305 return index_location
305 return index_location
306
306
307
307
308 @pytest.fixture(scope='session', autouse=True)
308 @pytest.fixture(scope='session', autouse=True)
309 def tests_tmp_path(request):
309 def tests_tmp_path(request):
310 """
310 """
311 Create temporary directory to be used during the test session.
311 Create temporary directory to be used during the test session.
312 """
312 """
313 if not os.path.exists(TESTS_TMP_PATH):
313 if not os.path.exists(TESTS_TMP_PATH):
314 os.makedirs(TESTS_TMP_PATH)
314 os.makedirs(TESTS_TMP_PATH)
315
315
316 if not request.config.getoption('--keep-tmp-path'):
316 if not request.config.getoption('--keep-tmp-path'):
317 @request.addfinalizer
317 @request.addfinalizer
318 def remove_tmp_path():
318 def remove_tmp_path():
319 shutil.rmtree(TESTS_TMP_PATH)
319 shutil.rmtree(TESTS_TMP_PATH)
320
320
321 return TESTS_TMP_PATH
321 return TESTS_TMP_PATH
322
322
323
323
324 @pytest.fixture
324 @pytest.fixture
325 def test_repo_group(request):
325 def test_repo_group(request):
326 """
326 """
327 Create a temporary repository group, and destroy it after
327 Create a temporary repository group, and destroy it after
328 usage automatically
328 usage automatically
329 """
329 """
330 fixture = Fixture()
330 fixture = Fixture()
331 repogroupid = 'test_repo_group_%s' % int(time.time())
331 repogroupid = 'test_repo_group_%s' % int(time.time())
332 repo_group = fixture.create_repo_group(repogroupid)
332 repo_group = fixture.create_repo_group(repogroupid)
333
333
334 def _cleanup():
334 def _cleanup():
335 fixture.destroy_repo_group(repogroupid)
335 fixture.destroy_repo_group(repogroupid)
336
336
337 request.addfinalizer(_cleanup)
337 request.addfinalizer(_cleanup)
338 return repo_group
338 return repo_group
339
339
340
340
341 @pytest.fixture
341 @pytest.fixture
342 def test_user_group(request):
342 def test_user_group(request):
343 """
343 """
344 Create a temporary user group, and destroy it after
344 Create a temporary user group, and destroy it after
345 usage automatically
345 usage automatically
346 """
346 """
347 fixture = Fixture()
347 fixture = Fixture()
348 usergroupid = 'test_user_group_%s' % int(time.time())
348 usergroupid = 'test_user_group_%s' % int(time.time())
349 user_group = fixture.create_user_group(usergroupid)
349 user_group = fixture.create_user_group(usergroupid)
350
350
351 def _cleanup():
351 def _cleanup():
352 fixture.destroy_user_group(user_group)
352 fixture.destroy_user_group(user_group)
353
353
354 request.addfinalizer(_cleanup)
354 request.addfinalizer(_cleanup)
355 return user_group
355 return user_group
356
356
357
357
358 @pytest.fixture(scope='session')
358 @pytest.fixture(scope='session')
359 def test_repo(request):
359 def test_repo(request):
360 container = TestRepoContainer()
360 container = TestRepoContainer()
361 request.addfinalizer(container._cleanup)
361 request.addfinalizer(container._cleanup)
362 return container
362 return container
363
363
364
364
365 class TestRepoContainer(object):
365 class TestRepoContainer(object):
366 """
366 """
367 Container for test repositories which are used read only.
367 Container for test repositories which are used read only.
368
368
369 Repositories will be created on demand and re-used during the lifetime
369 Repositories will be created on demand and re-used during the lifetime
370 of this object.
370 of this object.
371
371
372 Usage to get the svn test repository "minimal"::
372 Usage to get the svn test repository "minimal"::
373
373
374 test_repo = TestContainer()
374 test_repo = TestContainer()
375 repo = test_repo('minimal', 'svn')
375 repo = test_repo('minimal', 'svn')
376
376
377 """
377 """
378
378
379 dump_extractors = {
379 dump_extractors = {
380 'git': utils.extract_git_repo_from_dump,
380 'git': utils.extract_git_repo_from_dump,
381 'hg': utils.extract_hg_repo_from_dump,
381 'hg': utils.extract_hg_repo_from_dump,
382 'svn': utils.extract_svn_repo_from_dump,
382 'svn': utils.extract_svn_repo_from_dump,
383 }
383 }
384
384
385 def __init__(self):
385 def __init__(self):
386 self._cleanup_repos = []
386 self._cleanup_repos = []
387 self._fixture = Fixture()
387 self._fixture = Fixture()
388 self._repos = {}
388 self._repos = {}
389
389
390 def __call__(self, dump_name, backend_alias):
390 def __call__(self, dump_name, backend_alias):
391 key = (dump_name, backend_alias)
391 key = (dump_name, backend_alias)
392 if key not in self._repos:
392 if key not in self._repos:
393 repo = self._create_repo(dump_name, backend_alias)
393 repo = self._create_repo(dump_name, backend_alias)
394 self._repos[key] = repo.repo_id
394 self._repos[key] = repo.repo_id
395 return Repository.get(self._repos[key])
395 return Repository.get(self._repos[key])
396
396
397 def _create_repo(self, dump_name, backend_alias):
397 def _create_repo(self, dump_name, backend_alias):
398 repo_name = '%s-%s' % (backend_alias, dump_name)
398 repo_name = '%s-%s' % (backend_alias, dump_name)
399 backend_class = get_backend(backend_alias)
399 backend_class = get_backend(backend_alias)
400 dump_extractor = self.dump_extractors[backend_alias]
400 dump_extractor = self.dump_extractors[backend_alias]
401 repo_path = dump_extractor(dump_name, repo_name)
401 repo_path = dump_extractor(dump_name, repo_name)
402 vcs_repo = backend_class(repo_path)
402 vcs_repo = backend_class(repo_path)
403 repo2db_mapper({repo_name: vcs_repo})
403 repo2db_mapper({repo_name: vcs_repo})
404 repo = RepoModel().get_by_repo_name(repo_name)
404 repo = RepoModel().get_by_repo_name(repo_name)
405 self._cleanup_repos.append(repo_name)
405 self._cleanup_repos.append(repo_name)
406 return repo
406 return repo
407
407
408 def _cleanup(self):
408 def _cleanup(self):
409 for repo_name in reversed(self._cleanup_repos):
409 for repo_name in reversed(self._cleanup_repos):
410 self._fixture.destroy_repo(repo_name)
410 self._fixture.destroy_repo(repo_name)
411
411
412
412
413 @pytest.fixture
413 @pytest.fixture
414 def backend(request, backend_alias, pylonsapp, test_repo):
414 def backend(request, backend_alias, pylonsapp, test_repo):
415 """
415 """
416 Parametrized fixture which represents a single backend implementation.
416 Parametrized fixture which represents a single backend implementation.
417
417
418 It respects the option `--backends` to focus the test run on specific
418 It respects the option `--backends` to focus the test run on specific
419 backend implementations.
419 backend implementations.
420
420
421 It also supports `pytest.mark.xfail_backends` to mark tests as failing
421 It also supports `pytest.mark.xfail_backends` to mark tests as failing
422 for specific backends. This is intended as a utility for incremental
422 for specific backends. This is intended as a utility for incremental
423 development of a new backend implementation.
423 development of a new backend implementation.
424 """
424 """
425 if backend_alias not in request.config.getoption('--backends'):
425 if backend_alias not in request.config.getoption('--backends'):
426 pytest.skip("Backend %s not selected." % (backend_alias, ))
426 pytest.skip("Backend %s not selected." % (backend_alias, ))
427
427
428 utils.check_xfail_backends(request.node, backend_alias)
428 utils.check_xfail_backends(request.node, backend_alias)
429 utils.check_skip_backends(request.node, backend_alias)
429 utils.check_skip_backends(request.node, backend_alias)
430
430
431 repo_name = 'vcs_test_%s' % (backend_alias, )
431 repo_name = 'vcs_test_%s' % (backend_alias, )
432 backend = Backend(
432 backend = Backend(
433 alias=backend_alias,
433 alias=backend_alias,
434 repo_name=repo_name,
434 repo_name=repo_name,
435 test_name=request.node.name,
435 test_name=request.node.name,
436 test_repo_container=test_repo)
436 test_repo_container=test_repo)
437 request.addfinalizer(backend.cleanup)
437 request.addfinalizer(backend.cleanup)
438 return backend
438 return backend
439
439
440
440
441 @pytest.fixture
441 @pytest.fixture
442 def backend_git(request, pylonsapp, test_repo):
442 def backend_git(request, pylonsapp, test_repo):
443 return backend(request, 'git', pylonsapp, test_repo)
443 return backend(request, 'git', pylonsapp, test_repo)
444
444
445
445
446 @pytest.fixture
446 @pytest.fixture
447 def backend_hg(request, pylonsapp, test_repo):
447 def backend_hg(request, pylonsapp, test_repo):
448 return backend(request, 'hg', pylonsapp, test_repo)
448 return backend(request, 'hg', pylonsapp, test_repo)
449
449
450
450
451 @pytest.fixture
451 @pytest.fixture
452 def backend_svn(request, pylonsapp, test_repo):
452 def backend_svn(request, pylonsapp, test_repo):
453 return backend(request, 'svn', pylonsapp, test_repo)
453 return backend(request, 'svn', pylonsapp, test_repo)
454
454
455
455
456 @pytest.fixture
456 @pytest.fixture
457 def backend_random(backend_git):
457 def backend_random(backend_git):
458 """
458 """
459 Use this to express that your tests need "a backend.
459 Use this to express that your tests need "a backend.
460
460
461 A few of our tests need a backend, so that we can run the code. This
461 A few of our tests need a backend, so that we can run the code. This
462 fixture is intended to be used for such cases. It will pick one of the
462 fixture is intended to be used for such cases. It will pick one of the
463 backends and run the tests.
463 backends and run the tests.
464
464
465 The fixture `backend` would run the test multiple times for each
465 The fixture `backend` would run the test multiple times for each
466 available backend which is a pure waste of time if the test is
466 available backend which is a pure waste of time if the test is
467 independent of the backend type.
467 independent of the backend type.
468 """
468 """
469 # TODO: johbo: Change this to pick a random backend
469 # TODO: johbo: Change this to pick a random backend
470 return backend_git
470 return backend_git
471
471
472
472
473 @pytest.fixture
473 @pytest.fixture
474 def backend_stub(backend_git):
474 def backend_stub(backend_git):
475 """
475 """
476 Use this to express that your tests need a backend stub
476 Use this to express that your tests need a backend stub
477
477
478 TODO: mikhail: Implement a real stub logic instead of returning
478 TODO: mikhail: Implement a real stub logic instead of returning
479 a git backend
479 a git backend
480 """
480 """
481 return backend_git
481 return backend_git
482
482
483
483
484 @pytest.fixture
484 @pytest.fixture
485 def repo_stub(backend_stub):
485 def repo_stub(backend_stub):
486 """
486 """
487 Use this to express that your tests need a repository stub
487 Use this to express that your tests need a repository stub
488 """
488 """
489 return backend_stub.create_repo()
489 return backend_stub.create_repo()
490
490
491
491
492 class Backend(object):
492 class Backend(object):
493 """
493 """
494 Represents the test configuration for one supported backend
494 Represents the test configuration for one supported backend
495
495
496 Provides easy access to different test repositories based on
496 Provides easy access to different test repositories based on
497 `__getitem__`. Such repositories will only be created once per test
497 `__getitem__`. Such repositories will only be created once per test
498 session.
498 session.
499 """
499 """
500
500
501 invalid_repo_name = re.compile(r'[^0-9a-zA-Z]+')
501 invalid_repo_name = re.compile(r'[^0-9a-zA-Z]+')
502 _master_repo = None
502 _master_repo = None
503 _commit_ids = {}
503 _commit_ids = {}
504
504
505 def __init__(self, alias, repo_name, test_name, test_repo_container):
505 def __init__(self, alias, repo_name, test_name, test_repo_container):
506 self.alias = alias
506 self.alias = alias
507 self.repo_name = repo_name
507 self.repo_name = repo_name
508 self._cleanup_repos = []
508 self._cleanup_repos = []
509 self._test_name = test_name
509 self._test_name = test_name
510 self._test_repo_container = test_repo_container
510 self._test_repo_container = test_repo_container
511 # TODO: johbo: Used as a delegate interim. Not yet sure if Backend or
511 # TODO: johbo: Used as a delegate interim. Not yet sure if Backend or
512 # Fixture will survive in the end.
512 # Fixture will survive in the end.
513 self._fixture = Fixture()
513 self._fixture = Fixture()
514
514
515 def __getitem__(self, key):
515 def __getitem__(self, key):
516 return self._test_repo_container(key, self.alias)
516 return self._test_repo_container(key, self.alias)
517
517
518 @property
518 @property
519 def repo(self):
519 def repo(self):
520 """
520 """
521 Returns the "current" repository. This is the vcs_test repo or the
521 Returns the "current" repository. This is the vcs_test repo or the
522 last repo which has been created with `create_repo`.
522 last repo which has been created with `create_repo`.
523 """
523 """
524 from rhodecode.model.db import Repository
524 from rhodecode.model.db import Repository
525 return Repository.get_by_repo_name(self.repo_name)
525 return Repository.get_by_repo_name(self.repo_name)
526
526
527 @property
527 @property
528 def default_branch_name(self):
528 def default_branch_name(self):
529 VcsRepository = get_backend(self.alias)
529 VcsRepository = get_backend(self.alias)
530 return VcsRepository.DEFAULT_BRANCH_NAME
530 return VcsRepository.DEFAULT_BRANCH_NAME
531
531
532 @property
532 @property
533 def default_head_id(self):
533 def default_head_id(self):
534 """
534 """
535 Returns the default head id of the underlying backend.
535 Returns the default head id of the underlying backend.
536
536
537 This will be the default branch name in case the backend does have a
537 This will be the default branch name in case the backend does have a
538 default branch. In the other cases it will point to a valid head
538 default branch. In the other cases it will point to a valid head
539 which can serve as the base to create a new commit on top of it.
539 which can serve as the base to create a new commit on top of it.
540 """
540 """
541 vcsrepo = self.repo.scm_instance()
541 vcsrepo = self.repo.scm_instance()
542 head_id = (
542 head_id = (
543 vcsrepo.DEFAULT_BRANCH_NAME or
543 vcsrepo.DEFAULT_BRANCH_NAME or
544 vcsrepo.commit_ids[-1])
544 vcsrepo.commit_ids[-1])
545 return head_id
545 return head_id
546
546
547 @property
547 @property
548 def commit_ids(self):
548 def commit_ids(self):
549 """
549 """
550 Returns the list of commits for the last created repository
550 Returns the list of commits for the last created repository
551 """
551 """
552 return self._commit_ids
552 return self._commit_ids
553
553
554 def create_master_repo(self, commits):
554 def create_master_repo(self, commits):
555 """
555 """
556 Create a repository and remember it as a template.
556 Create a repository and remember it as a template.
557
557
558 This allows to easily create derived repositories to construct
558 This allows to easily create derived repositories to construct
559 more complex scenarios for diff, compare and pull requests.
559 more complex scenarios for diff, compare and pull requests.
560
560
561 Returns a commit map which maps from commit message to raw_id.
561 Returns a commit map which maps from commit message to raw_id.
562 """
562 """
563 self._master_repo = self.create_repo(commits=commits)
563 self._master_repo = self.create_repo(commits=commits)
564 return self._commit_ids
564 return self._commit_ids
565
565
566 def create_repo(
566 def create_repo(
567 self, commits=None, number_of_commits=0, heads=None,
567 self, commits=None, number_of_commits=0, heads=None,
568 name_suffix=u'', **kwargs):
568 name_suffix=u'', **kwargs):
569 """
569 """
570 Create a repository and record it for later cleanup.
570 Create a repository and record it for later cleanup.
571
571
572 :param commits: Optional. A sequence of dict instances.
572 :param commits: Optional. A sequence of dict instances.
573 Will add a commit per entry to the new repository.
573 Will add a commit per entry to the new repository.
574 :param number_of_commits: Optional. If set to a number, this number of
574 :param number_of_commits: Optional. If set to a number, this number of
575 commits will be added to the new repository.
575 commits will be added to the new repository.
576 :param heads: Optional. Can be set to a sequence of of commit
576 :param heads: Optional. Can be set to a sequence of of commit
577 names which shall be pulled in from the master repository.
577 names which shall be pulled in from the master repository.
578
578
579 """
579 """
580 self.repo_name = self._next_repo_name() + name_suffix
580 self.repo_name = self._next_repo_name() + name_suffix
581 repo = self._fixture.create_repo(
581 repo = self._fixture.create_repo(
582 self.repo_name, repo_type=self.alias, **kwargs)
582 self.repo_name, repo_type=self.alias, **kwargs)
583 self._cleanup_repos.append(repo.repo_name)
583 self._cleanup_repos.append(repo.repo_name)
584
584
585 commits = commits or [
585 commits = commits or [
586 {'message': 'Commit %s of %s' % (x, self.repo_name)}
586 {'message': 'Commit %s of %s' % (x, self.repo_name)}
587 for x in xrange(number_of_commits)]
587 for x in xrange(number_of_commits)]
588 self._add_commits_to_repo(repo.scm_instance(), commits)
588 self._add_commits_to_repo(repo.scm_instance(), commits)
589 if heads:
589 if heads:
590 self.pull_heads(repo, heads)
590 self.pull_heads(repo, heads)
591
591
592 return repo
592 return repo
593
593
594 def pull_heads(self, repo, heads):
594 def pull_heads(self, repo, heads):
595 """
595 """
596 Make sure that repo contains all commits mentioned in `heads`
596 Make sure that repo contains all commits mentioned in `heads`
597 """
597 """
598 vcsmaster = self._master_repo.scm_instance()
598 vcsmaster = self._master_repo.scm_instance()
599 vcsrepo = repo.scm_instance()
599 vcsrepo = repo.scm_instance()
600 vcsrepo.config.clear_section('hooks')
600 vcsrepo.config.clear_section('hooks')
601 commit_ids = [self._commit_ids[h] for h in heads]
601 commit_ids = [self._commit_ids[h] for h in heads]
602 vcsrepo.pull(vcsmaster.path, commit_ids=commit_ids)
602 vcsrepo.pull(vcsmaster.path, commit_ids=commit_ids)
603
603
604 def create_fork(self):
604 def create_fork(self):
605 repo_to_fork = self.repo_name
605 repo_to_fork = self.repo_name
606 self.repo_name = self._next_repo_name()
606 self.repo_name = self._next_repo_name()
607 repo = self._fixture.create_fork(repo_to_fork, self.repo_name)
607 repo = self._fixture.create_fork(repo_to_fork, self.repo_name)
608 self._cleanup_repos.append(self.repo_name)
608 self._cleanup_repos.append(self.repo_name)
609 return repo
609 return repo
610
610
611 def new_repo_name(self, suffix=u''):
611 def new_repo_name(self, suffix=u''):
612 self.repo_name = self._next_repo_name() + suffix
612 self.repo_name = self._next_repo_name() + suffix
613 self._cleanup_repos.append(self.repo_name)
613 self._cleanup_repos.append(self.repo_name)
614 return self.repo_name
614 return self.repo_name
615
615
616 def _next_repo_name(self):
616 def _next_repo_name(self):
617 return u"%s_%s" % (
617 return u"%s_%s" % (
618 self.invalid_repo_name.sub(u'_', self._test_name),
618 self.invalid_repo_name.sub(u'_', self._test_name),
619 len(self._cleanup_repos))
619 len(self._cleanup_repos))
620
620
621 def ensure_file(self, filename, content='Test content\n'):
621 def ensure_file(self, filename, content='Test content\n'):
622 assert self._cleanup_repos, "Avoid writing into vcs_test repos"
622 assert self._cleanup_repos, "Avoid writing into vcs_test repos"
623 commits = [
623 commits = [
624 {'added': [
624 {'added': [
625 FileNode(filename, content=content),
625 FileNode(filename, content=content),
626 ]},
626 ]},
627 ]
627 ]
628 self._add_commits_to_repo(self.repo.scm_instance(), commits)
628 self._add_commits_to_repo(self.repo.scm_instance(), commits)
629
629
630 def enable_downloads(self):
630 def enable_downloads(self):
631 repo = self.repo
631 repo = self.repo
632 repo.enable_downloads = True
632 repo.enable_downloads = True
633 Session().add(repo)
633 Session().add(repo)
634 Session().commit()
634 Session().commit()
635
635
636 def cleanup(self):
636 def cleanup(self):
637 for repo_name in reversed(self._cleanup_repos):
637 for repo_name in reversed(self._cleanup_repos):
638 self._fixture.destroy_repo(repo_name)
638 self._fixture.destroy_repo(repo_name)
639
639
640 def _add_commits_to_repo(self, repo, commits):
640 def _add_commits_to_repo(self, repo, commits):
641 commit_ids = _add_commits_to_repo(repo, commits)
641 commit_ids = _add_commits_to_repo(repo, commits)
642 if not commit_ids:
642 if not commit_ids:
643 return
643 return
644 self._commit_ids = commit_ids
644 self._commit_ids = commit_ids
645
645
646 # Creating refs for Git to allow fetching them from remote repository
646 # Creating refs for Git to allow fetching them from remote repository
647 if self.alias == 'git':
647 if self.alias == 'git':
648 refs = {}
648 refs = {}
649 for message in self._commit_ids:
649 for message in self._commit_ids:
650 # TODO: mikhail: do more special chars replacements
650 # TODO: mikhail: do more special chars replacements
651 ref_name = 'refs/test-refs/{}'.format(
651 ref_name = 'refs/test-refs/{}'.format(
652 message.replace(' ', ''))
652 message.replace(' ', ''))
653 refs[ref_name] = self._commit_ids[message]
653 refs[ref_name] = self._commit_ids[message]
654 self._create_refs(repo, refs)
654 self._create_refs(repo, refs)
655
655
656 def _create_refs(self, repo, refs):
656 def _create_refs(self, repo, refs):
657 for ref_name in refs:
657 for ref_name in refs:
658 repo.set_refs(ref_name, refs[ref_name])
658 repo.set_refs(ref_name, refs[ref_name])
659
659
660
660
661 @pytest.fixture
661 @pytest.fixture
662 def vcsbackend(request, backend_alias, tests_tmp_path, pylonsapp, test_repo):
662 def vcsbackend(request, backend_alias, tests_tmp_path, pylonsapp, test_repo):
663 """
663 """
664 Parametrized fixture which represents a single vcs backend implementation.
664 Parametrized fixture which represents a single vcs backend implementation.
665
665
666 See the fixture `backend` for more details. This one implements the same
666 See the fixture `backend` for more details. This one implements the same
667 concept, but on vcs level. So it does not provide model instances etc.
667 concept, but on vcs level. So it does not provide model instances etc.
668
668
669 Parameters are generated dynamically, see :func:`pytest_generate_tests`
669 Parameters are generated dynamically, see :func:`pytest_generate_tests`
670 for how this works.
670 for how this works.
671 """
671 """
672 if backend_alias not in request.config.getoption('--backends'):
672 if backend_alias not in request.config.getoption('--backends'):
673 pytest.skip("Backend %s not selected." % (backend_alias, ))
673 pytest.skip("Backend %s not selected." % (backend_alias, ))
674
674
675 utils.check_xfail_backends(request.node, backend_alias)
675 utils.check_xfail_backends(request.node, backend_alias)
676 utils.check_skip_backends(request.node, backend_alias)
676 utils.check_skip_backends(request.node, backend_alias)
677
677
678 repo_name = 'vcs_test_%s' % (backend_alias, )
678 repo_name = 'vcs_test_%s' % (backend_alias, )
679 repo_path = os.path.join(tests_tmp_path, repo_name)
679 repo_path = os.path.join(tests_tmp_path, repo_name)
680 backend = VcsBackend(
680 backend = VcsBackend(
681 alias=backend_alias,
681 alias=backend_alias,
682 repo_path=repo_path,
682 repo_path=repo_path,
683 test_name=request.node.name,
683 test_name=request.node.name,
684 test_repo_container=test_repo)
684 test_repo_container=test_repo)
685 request.addfinalizer(backend.cleanup)
685 request.addfinalizer(backend.cleanup)
686 return backend
686 return backend
687
687
688
688
689 @pytest.fixture
689 @pytest.fixture
690 def vcsbackend_git(request, tests_tmp_path, pylonsapp, test_repo):
690 def vcsbackend_git(request, tests_tmp_path, pylonsapp, test_repo):
691 return vcsbackend(request, 'git', tests_tmp_path, pylonsapp, test_repo)
691 return vcsbackend(request, 'git', tests_tmp_path, pylonsapp, test_repo)
692
692
693
693
694 @pytest.fixture
694 @pytest.fixture
695 def vcsbackend_hg(request, tests_tmp_path, pylonsapp, test_repo):
695 def vcsbackend_hg(request, tests_tmp_path, pylonsapp, test_repo):
696 return vcsbackend(request, 'hg', tests_tmp_path, pylonsapp, test_repo)
696 return vcsbackend(request, 'hg', tests_tmp_path, pylonsapp, test_repo)
697
697
698
698
699 @pytest.fixture
699 @pytest.fixture
700 def vcsbackend_svn(request, tests_tmp_path, pylonsapp, test_repo):
700 def vcsbackend_svn(request, tests_tmp_path, pylonsapp, test_repo):
701 return vcsbackend(request, 'svn', tests_tmp_path, pylonsapp, test_repo)
701 return vcsbackend(request, 'svn', tests_tmp_path, pylonsapp, test_repo)
702
702
703
703
704 @pytest.fixture
704 @pytest.fixture
705 def vcsbackend_random(vcsbackend_git):
705 def vcsbackend_random(vcsbackend_git):
706 """
706 """
707 Use this to express that your tests need "a vcsbackend".
707 Use this to express that your tests need "a vcsbackend".
708
708
709 The fixture `vcsbackend` would run the test multiple times for each
709 The fixture `vcsbackend` would run the test multiple times for each
710 available vcs backend which is a pure waste of time if the test is
710 available vcs backend which is a pure waste of time if the test is
711 independent of the vcs backend type.
711 independent of the vcs backend type.
712 """
712 """
713 # TODO: johbo: Change this to pick a random backend
713 # TODO: johbo: Change this to pick a random backend
714 return vcsbackend_git
714 return vcsbackend_git
715
715
716
716
717 @pytest.fixture
717 @pytest.fixture
718 def vcsbackend_stub(vcsbackend_git):
718 def vcsbackend_stub(vcsbackend_git):
719 """
719 """
720 Use this to express that your test just needs a stub of a vcsbackend.
720 Use this to express that your test just needs a stub of a vcsbackend.
721
721
722 Plan is to eventually implement an in-memory stub to speed tests up.
722 Plan is to eventually implement an in-memory stub to speed tests up.
723 """
723 """
724 return vcsbackend_git
724 return vcsbackend_git
725
725
726
726
727 class VcsBackend(object):
727 class VcsBackend(object):
728 """
728 """
729 Represents the test configuration for one supported vcs backend.
729 Represents the test configuration for one supported vcs backend.
730 """
730 """
731
731
732 invalid_repo_name = re.compile(r'[^0-9a-zA-Z]+')
732 invalid_repo_name = re.compile(r'[^0-9a-zA-Z]+')
733
733
734 def __init__(self, alias, repo_path, test_name, test_repo_container):
734 def __init__(self, alias, repo_path, test_name, test_repo_container):
735 self.alias = alias
735 self.alias = alias
736 self._repo_path = repo_path
736 self._repo_path = repo_path
737 self._cleanup_repos = []
737 self._cleanup_repos = []
738 self._test_name = test_name
738 self._test_name = test_name
739 self._test_repo_container = test_repo_container
739 self._test_repo_container = test_repo_container
740
740
741 def __getitem__(self, key):
741 def __getitem__(self, key):
742 return self._test_repo_container(key, self.alias).scm_instance()
742 return self._test_repo_container(key, self.alias).scm_instance()
743
743
744 @property
744 @property
745 def repo(self):
745 def repo(self):
746 """
746 """
747 Returns the "current" repository. This is the vcs_test repo of the last
747 Returns the "current" repository. This is the vcs_test repo of the last
748 repo which has been created.
748 repo which has been created.
749 """
749 """
750 Repository = get_backend(self.alias)
750 Repository = get_backend(self.alias)
751 return Repository(self._repo_path)
751 return Repository(self._repo_path)
752
752
753 @property
753 @property
754 def backend(self):
754 def backend(self):
755 """
755 """
756 Returns the backend implementation class.
756 Returns the backend implementation class.
757 """
757 """
758 return get_backend(self.alias)
758 return get_backend(self.alias)
759
759
760 def create_repo(self, commits=None, number_of_commits=0, _clone_repo=None):
760 def create_repo(self, commits=None, number_of_commits=0, _clone_repo=None):
761 repo_name = self._next_repo_name()
761 repo_name = self._next_repo_name()
762 self._repo_path = get_new_dir(repo_name)
762 self._repo_path = get_new_dir(repo_name)
763 repo_class = get_backend(self.alias)
763 repo_class = get_backend(self.alias)
764 src_url = None
764 src_url = None
765 if _clone_repo:
765 if _clone_repo:
766 src_url = _clone_repo.path
766 src_url = _clone_repo.path
767 repo = repo_class(self._repo_path, create=True, src_url=src_url)
767 repo = repo_class(self._repo_path, create=True, src_url=src_url)
768 self._cleanup_repos.append(repo)
768 self._cleanup_repos.append(repo)
769
769
770 commits = commits or [
770 commits = commits or [
771 {'message': 'Commit %s of %s' % (x, repo_name)}
771 {'message': 'Commit %s of %s' % (x, repo_name)}
772 for x in xrange(number_of_commits)]
772 for x in xrange(number_of_commits)]
773 _add_commits_to_repo(repo, commits)
773 _add_commits_to_repo(repo, commits)
774 return repo
774 return repo
775
775
776 def clone_repo(self, repo):
776 def clone_repo(self, repo):
777 return self.create_repo(_clone_repo=repo)
777 return self.create_repo(_clone_repo=repo)
778
778
779 def cleanup(self):
779 def cleanup(self):
780 for repo in self._cleanup_repos:
780 for repo in self._cleanup_repos:
781 shutil.rmtree(repo.path)
781 shutil.rmtree(repo.path)
782
782
783 def new_repo_path(self):
783 def new_repo_path(self):
784 repo_name = self._next_repo_name()
784 repo_name = self._next_repo_name()
785 self._repo_path = get_new_dir(repo_name)
785 self._repo_path = get_new_dir(repo_name)
786 return self._repo_path
786 return self._repo_path
787
787
788 def _next_repo_name(self):
788 def _next_repo_name(self):
789 return "%s_%s" % (
789 return "%s_%s" % (
790 self.invalid_repo_name.sub('_', self._test_name),
790 self.invalid_repo_name.sub('_', self._test_name),
791 len(self._cleanup_repos))
791 len(self._cleanup_repos))
792
792
793 def add_file(self, repo, filename, content='Test content\n'):
793 def add_file(self, repo, filename, content='Test content\n'):
794 imc = repo.in_memory_commit
794 imc = repo.in_memory_commit
795 imc.add(FileNode(filename, content=content))
795 imc.add(FileNode(filename, content=content))
796 imc.commit(
796 imc.commit(
797 message=u'Automatic commit from vcsbackend fixture',
797 message=u'Automatic commit from vcsbackend fixture',
798 author=u'Automatic')
798 author=u'Automatic')
799
799
800 def ensure_file(self, filename, content='Test content\n'):
800 def ensure_file(self, filename, content='Test content\n'):
801 assert self._cleanup_repos, "Avoid writing into vcs_test repos"
801 assert self._cleanup_repos, "Avoid writing into vcs_test repos"
802 self.add_file(self.repo, filename, content)
802 self.add_file(self.repo, filename, content)
803
803
804
804
805 def _add_commits_to_repo(vcs_repo, commits):
805 def _add_commits_to_repo(vcs_repo, commits):
806 commit_ids = {}
806 commit_ids = {}
807 if not commits:
807 if not commits:
808 return commit_ids
808 return commit_ids
809
809
810 imc = vcs_repo.in_memory_commit
810 imc = vcs_repo.in_memory_commit
811 commit = None
811 commit = None
812
812
813 for idx, commit in enumerate(commits):
813 for idx, commit in enumerate(commits):
814 message = unicode(commit.get('message', 'Commit %s' % idx))
814 message = unicode(commit.get('message', 'Commit %s' % idx))
815
815
816 for node in commit.get('added', []):
816 for node in commit.get('added', []):
817 imc.add(FileNode(node.path, content=node.content))
817 imc.add(FileNode(node.path, content=node.content))
818 for node in commit.get('changed', []):
818 for node in commit.get('changed', []):
819 imc.change(FileNode(node.path, content=node.content))
819 imc.change(FileNode(node.path, content=node.content))
820 for node in commit.get('removed', []):
820 for node in commit.get('removed', []):
821 imc.remove(FileNode(node.path))
821 imc.remove(FileNode(node.path))
822
822
823 parents = [
823 parents = [
824 vcs_repo.get_commit(commit_id=commit_ids[p])
824 vcs_repo.get_commit(commit_id=commit_ids[p])
825 for p in commit.get('parents', [])]
825 for p in commit.get('parents', [])]
826
826
827 operations = ('added', 'changed', 'removed')
827 operations = ('added', 'changed', 'removed')
828 if not any((commit.get(o) for o in operations)):
828 if not any((commit.get(o) for o in operations)):
829 imc.add(FileNode('file_%s' % idx, content=message))
829 imc.add(FileNode('file_%s' % idx, content=message))
830
830
831 commit = imc.commit(
831 commit = imc.commit(
832 message=message,
832 message=message,
833 author=unicode(commit.get('author', 'Automatic')),
833 author=unicode(commit.get('author', 'Automatic')),
834 date=commit.get('date'),
834 date=commit.get('date'),
835 branch=commit.get('branch'),
835 branch=commit.get('branch'),
836 parents=parents)
836 parents=parents)
837
837
838 commit_ids[commit.message] = commit.raw_id
838 commit_ids[commit.message] = commit.raw_id
839
839
840 return commit_ids
840 return commit_ids
841
841
842
842
843 @pytest.fixture
843 @pytest.fixture
844 def reposerver(request):
844 def reposerver(request):
845 """
845 """
846 Allows to serve a backend repository
846 Allows to serve a backend repository
847 """
847 """
848
848
849 repo_server = RepoServer()
849 repo_server = RepoServer()
850 request.addfinalizer(repo_server.cleanup)
850 request.addfinalizer(repo_server.cleanup)
851 return repo_server
851 return repo_server
852
852
853
853
854 class RepoServer(object):
854 class RepoServer(object):
855 """
855 """
856 Utility to serve a local repository for the duration of a test case.
856 Utility to serve a local repository for the duration of a test case.
857
857
858 Supports only Subversion so far.
858 Supports only Subversion so far.
859 """
859 """
860
860
861 url = None
861 url = None
862
862
863 def __init__(self):
863 def __init__(self):
864 self._cleanup_servers = []
864 self._cleanup_servers = []
865
865
866 def serve(self, vcsrepo):
866 def serve(self, vcsrepo):
867 if vcsrepo.alias != 'svn':
867 if vcsrepo.alias != 'svn':
868 raise TypeError("Backend %s not supported" % vcsrepo.alias)
868 raise TypeError("Backend %s not supported" % vcsrepo.alias)
869
869
870 proc = subprocess32.Popen(
870 proc = subprocess32.Popen(
871 ['svnserve', '-d', '--foreground', '--listen-host', 'localhost',
871 ['svnserve', '-d', '--foreground', '--listen-host', 'localhost',
872 '--root', vcsrepo.path])
872 '--root', vcsrepo.path])
873 self._cleanup_servers.append(proc)
873 self._cleanup_servers.append(proc)
874 self.url = 'svn://localhost'
874 self.url = 'svn://localhost'
875
875
876 def cleanup(self):
876 def cleanup(self):
877 for proc in self._cleanup_servers:
877 for proc in self._cleanup_servers:
878 proc.terminate()
878 proc.terminate()
879
879
880
880
881 @pytest.fixture
881 @pytest.fixture
882 def pr_util(backend, request):
882 def pr_util(backend, request):
883 """
883 """
884 Utility for tests of models and for functional tests around pull requests.
884 Utility for tests of models and for functional tests around pull requests.
885
885
886 It gives an instance of :class:`PRTestUtility` which provides various
886 It gives an instance of :class:`PRTestUtility` which provides various
887 utility methods around one pull request.
887 utility methods around one pull request.
888
888
889 This fixture uses `backend` and inherits its parameterization.
889 This fixture uses `backend` and inherits its parameterization.
890 """
890 """
891
891
892 util = PRTestUtility(backend)
892 util = PRTestUtility(backend)
893
893
894 @request.addfinalizer
894 @request.addfinalizer
895 def cleanup():
895 def cleanup():
896 util.cleanup()
896 util.cleanup()
897
897
898 return util
898 return util
899
899
900
900
901 class PRTestUtility(object):
901 class PRTestUtility(object):
902
902
903 pull_request = None
903 pull_request = None
904 pull_request_id = None
904 pull_request_id = None
905 mergeable_patcher = None
905 mergeable_patcher = None
906 mergeable_mock = None
906 mergeable_mock = None
907 notification_patcher = None
907 notification_patcher = None
908
908
909 def __init__(self, backend):
909 def __init__(self, backend):
910 self.backend = backend
910 self.backend = backend
911
911
912 def create_pull_request(
912 def create_pull_request(
913 self, commits=None, target_head=None, source_head=None,
913 self, commits=None, target_head=None, source_head=None,
914 revisions=None, approved=False, author=None, mergeable=False,
914 revisions=None, approved=False, author=None, mergeable=False,
915 enable_notifications=True, name_suffix=u'', reviewers=None,
915 enable_notifications=True, name_suffix=u'', reviewers=None,
916 title=u"Test", description=u"Description"):
916 title=u"Test", description=u"Description"):
917 self.set_mergeable(mergeable)
917 self.set_mergeable(mergeable)
918 if not enable_notifications:
918 if not enable_notifications:
919 # mock notification side effect
919 # mock notification side effect
920 self.notification_patcher = mock.patch(
920 self.notification_patcher = mock.patch(
921 'rhodecode.model.notification.NotificationModel.create')
921 'rhodecode.model.notification.NotificationModel.create')
922 self.notification_patcher.start()
922 self.notification_patcher.start()
923
923
924 if not self.pull_request:
924 if not self.pull_request:
925 if not commits:
925 if not commits:
926 commits = [
926 commits = [
927 {'message': 'c1'},
927 {'message': 'c1'},
928 {'message': 'c2'},
928 {'message': 'c2'},
929 {'message': 'c3'},
929 {'message': 'c3'},
930 ]
930 ]
931 target_head = 'c1'
931 target_head = 'c1'
932 source_head = 'c2'
932 source_head = 'c2'
933 revisions = ['c2']
933 revisions = ['c2']
934
934
935 self.commit_ids = self.backend.create_master_repo(commits)
935 self.commit_ids = self.backend.create_master_repo(commits)
936 self.target_repository = self.backend.create_repo(
936 self.target_repository = self.backend.create_repo(
937 heads=[target_head], name_suffix=name_suffix)
937 heads=[target_head], name_suffix=name_suffix)
938 self.source_repository = self.backend.create_repo(
938 self.source_repository = self.backend.create_repo(
939 heads=[source_head], name_suffix=name_suffix)
939 heads=[source_head], name_suffix=name_suffix)
940 self.author = author or UserModel().get_by_username(
940 self.author = author or UserModel().get_by_username(
941 TEST_USER_ADMIN_LOGIN)
941 TEST_USER_ADMIN_LOGIN)
942
942
943 model = PullRequestModel()
943 model = PullRequestModel()
944 self.create_parameters = {
944 self.create_parameters = {
945 'created_by': self.author,
945 'created_by': self.author,
946 'source_repo': self.source_repository.repo_name,
946 'source_repo': self.source_repository.repo_name,
947 'source_ref': self._default_branch_reference(source_head),
947 'source_ref': self._default_branch_reference(source_head),
948 'target_repo': self.target_repository.repo_name,
948 'target_repo': self.target_repository.repo_name,
949 'target_ref': self._default_branch_reference(target_head),
949 'target_ref': self._default_branch_reference(target_head),
950 'revisions': [self.commit_ids[r] for r in revisions],
950 'revisions': [self.commit_ids[r] for r in revisions],
951 'reviewers': reviewers or self._get_reviewers(),
951 'reviewers': reviewers or self._get_reviewers(),
952 'title': title,
952 'title': title,
953 'description': description,
953 'description': description,
954 }
954 }
955 self.pull_request = model.create(**self.create_parameters)
955 self.pull_request = model.create(**self.create_parameters)
956 assert model.get_versions(self.pull_request) == []
956 assert model.get_versions(self.pull_request) == []
957
957
958 self.pull_request_id = self.pull_request.pull_request_id
958 self.pull_request_id = self.pull_request.pull_request_id
959
959
960 if approved:
960 if approved:
961 self.approve()
961 self.approve()
962
962
963 Session().add(self.pull_request)
963 Session().add(self.pull_request)
964 Session().commit()
964 Session().commit()
965
965
966 return self.pull_request
966 return self.pull_request
967
967
968 def approve(self):
968 def approve(self):
969 self.create_status_votes(
969 self.create_status_votes(
970 ChangesetStatus.STATUS_APPROVED,
970 ChangesetStatus.STATUS_APPROVED,
971 *self.pull_request.reviewers)
971 *self.pull_request.reviewers)
972
972
973 def close(self):
973 def close(self):
974 PullRequestModel().close_pull_request(self.pull_request, self.author)
974 PullRequestModel().close_pull_request(self.pull_request, self.author)
975
975
976 def _default_branch_reference(self, commit_message):
976 def _default_branch_reference(self, commit_message):
977 reference = '%s:%s:%s' % (
977 reference = '%s:%s:%s' % (
978 'branch',
978 'branch',
979 self.backend.default_branch_name,
979 self.backend.default_branch_name,
980 self.commit_ids[commit_message])
980 self.commit_ids[commit_message])
981 return reference
981 return reference
982
982
983 def _get_reviewers(self):
983 def _get_reviewers(self):
984 model = UserModel()
984 model = UserModel()
985 return [
985 return [
986 model.get_by_username(TEST_USER_REGULAR_LOGIN),
986 model.get_by_username(TEST_USER_REGULAR_LOGIN),
987 model.get_by_username(TEST_USER_REGULAR2_LOGIN),
987 model.get_by_username(TEST_USER_REGULAR2_LOGIN),
988 ]
988 ]
989
989
990 def update_source_repository(self, head=None):
990 def update_source_repository(self, head=None):
991 heads = [head or 'c3']
991 heads = [head or 'c3']
992 self.backend.pull_heads(self.source_repository, heads=heads)
992 self.backend.pull_heads(self.source_repository, heads=heads)
993
993
994 def add_one_commit(self, head=None):
994 def add_one_commit(self, head=None):
995 self.update_source_repository(head=head)
995 self.update_source_repository(head=head)
996 old_commit_ids = set(self.pull_request.revisions)
996 old_commit_ids = set(self.pull_request.revisions)
997 PullRequestModel().update_commits(self.pull_request)
997 PullRequestModel().update_commits(self.pull_request)
998 commit_ids = set(self.pull_request.revisions)
998 commit_ids = set(self.pull_request.revisions)
999 new_commit_ids = commit_ids - old_commit_ids
999 new_commit_ids = commit_ids - old_commit_ids
1000 assert len(new_commit_ids) == 1
1000 assert len(new_commit_ids) == 1
1001 return new_commit_ids.pop()
1001 return new_commit_ids.pop()
1002
1002
1003 def remove_one_commit(self):
1003 def remove_one_commit(self):
1004 assert len(self.pull_request.revisions) == 2
1004 assert len(self.pull_request.revisions) == 2
1005 source_vcs = self.source_repository.scm_instance()
1005 source_vcs = self.source_repository.scm_instance()
1006 removed_commit_id = source_vcs.commit_ids[-1]
1006 removed_commit_id = source_vcs.commit_ids[-1]
1007
1007
1008 # TODO: johbo: Git and Mercurial have an inconsistent vcs api here,
1008 # TODO: johbo: Git and Mercurial have an inconsistent vcs api here,
1009 # remove the if once that's sorted out.
1009 # remove the if once that's sorted out.
1010 if self.backend.alias == "git":
1010 if self.backend.alias == "git":
1011 kwargs = {'branch_name': self.backend.default_branch_name}
1011 kwargs = {'branch_name': self.backend.default_branch_name}
1012 else:
1012 else:
1013 kwargs = {}
1013 kwargs = {}
1014 source_vcs.strip(removed_commit_id, **kwargs)
1014 source_vcs.strip(removed_commit_id, **kwargs)
1015
1015
1016 PullRequestModel().update_commits(self.pull_request)
1016 PullRequestModel().update_commits(self.pull_request)
1017 assert len(self.pull_request.revisions) == 1
1017 assert len(self.pull_request.revisions) == 1
1018 return removed_commit_id
1018 return removed_commit_id
1019
1019
1020 def create_comment(self, linked_to=None):
1020 def create_comment(self, linked_to=None):
1021 comment = CommentsModel().create(
1021 comment = CommentsModel().create(
1022 text=u"Test comment",
1022 text=u"Test comment",
1023 repo=self.target_repository.repo_name,
1023 repo=self.target_repository.repo_name,
1024 user=self.author,
1024 user=self.author,
1025 pull_request=self.pull_request)
1025 pull_request=self.pull_request)
1026 assert comment.pull_request_version_id is None
1026 assert comment.pull_request_version_id is None
1027
1027
1028 if linked_to:
1028 if linked_to:
1029 PullRequestModel()._link_comments_to_version(linked_to)
1029 PullRequestModel()._link_comments_to_version(linked_to)
1030
1030
1031 return comment
1031 return comment
1032
1032
1033 def create_inline_comment(
1033 def create_inline_comment(
1034 self, linked_to=None, line_no=u'n1', file_path='file_1'):
1034 self, linked_to=None, line_no=u'n1', file_path='file_1'):
1035 comment = CommentsModel().create(
1035 comment = CommentsModel().create(
1036 text=u"Test comment",
1036 text=u"Test comment",
1037 repo=self.target_repository.repo_name,
1037 repo=self.target_repository.repo_name,
1038 user=self.author,
1038 user=self.author,
1039 line_no=line_no,
1039 line_no=line_no,
1040 f_path=file_path,
1040 f_path=file_path,
1041 pull_request=self.pull_request)
1041 pull_request=self.pull_request)
1042 assert comment.pull_request_version_id is None
1042 assert comment.pull_request_version_id is None
1043
1043
1044 if linked_to:
1044 if linked_to:
1045 PullRequestModel()._link_comments_to_version(linked_to)
1045 PullRequestModel()._link_comments_to_version(linked_to)
1046
1046
1047 return comment
1047 return comment
1048
1048
1049 def create_version_of_pull_request(self):
1049 def create_version_of_pull_request(self):
1050 pull_request = self.create_pull_request()
1050 pull_request = self.create_pull_request()
1051 version = PullRequestModel()._create_version_from_snapshot(
1051 version = PullRequestModel()._create_version_from_snapshot(
1052 pull_request)
1052 pull_request)
1053 return version
1053 return version
1054
1054
1055 def create_status_votes(self, status, *reviewers):
1055 def create_status_votes(self, status, *reviewers):
1056 for reviewer in reviewers:
1056 for reviewer in reviewers:
1057 ChangesetStatusModel().set_status(
1057 ChangesetStatusModel().set_status(
1058 repo=self.pull_request.target_repo,
1058 repo=self.pull_request.target_repo,
1059 status=status,
1059 status=status,
1060 user=reviewer.user_id,
1060 user=reviewer.user_id,
1061 pull_request=self.pull_request)
1061 pull_request=self.pull_request)
1062
1062
1063 def set_mergeable(self, value):
1063 def set_mergeable(self, value):
1064 if not self.mergeable_patcher:
1064 if not self.mergeable_patcher:
1065 self.mergeable_patcher = mock.patch.object(
1065 self.mergeable_patcher = mock.patch.object(
1066 VcsSettingsModel, 'get_general_settings')
1066 VcsSettingsModel, 'get_general_settings')
1067 self.mergeable_mock = self.mergeable_patcher.start()
1067 self.mergeable_mock = self.mergeable_patcher.start()
1068 self.mergeable_mock.return_value = {
1068 self.mergeable_mock.return_value = {
1069 'rhodecode_pr_merge_enabled': value}
1069 'rhodecode_pr_merge_enabled': value}
1070
1070
1071 def cleanup(self):
1071 def cleanup(self):
1072 # In case the source repository is already cleaned up, the pull
1072 # In case the source repository is already cleaned up, the pull
1073 # request will already be deleted.
1073 # request will already be deleted.
1074 pull_request = PullRequest().get(self.pull_request_id)
1074 pull_request = PullRequest().get(self.pull_request_id)
1075 if pull_request:
1075 if pull_request:
1076 PullRequestModel().delete(pull_request)
1076 PullRequestModel().delete(pull_request)
1077 Session().commit()
1077 Session().commit()
1078
1078
1079 if self.notification_patcher:
1079 if self.notification_patcher:
1080 self.notification_patcher.stop()
1080 self.notification_patcher.stop()
1081
1081
1082 if self.mergeable_patcher:
1082 if self.mergeable_patcher:
1083 self.mergeable_patcher.stop()
1083 self.mergeable_patcher.stop()
1084
1084
1085
1085
1086 @pytest.fixture
1086 @pytest.fixture
1087 def user_admin(pylonsapp):
1087 def user_admin(pylonsapp):
1088 """
1088 """
1089 Provides the default admin test user as an instance of `db.User`.
1089 Provides the default admin test user as an instance of `db.User`.
1090 """
1090 """
1091 user = UserModel().get_by_username(TEST_USER_ADMIN_LOGIN)
1091 user = UserModel().get_by_username(TEST_USER_ADMIN_LOGIN)
1092 return user
1092 return user
1093
1093
1094
1094
1095 @pytest.fixture
1095 @pytest.fixture
1096 def user_regular(pylonsapp):
1096 def user_regular(pylonsapp):
1097 """
1097 """
1098 Provides the default regular test user as an instance of `db.User`.
1098 Provides the default regular test user as an instance of `db.User`.
1099 """
1099 """
1100 user = UserModel().get_by_username(TEST_USER_REGULAR_LOGIN)
1100 user = UserModel().get_by_username(TEST_USER_REGULAR_LOGIN)
1101 return user
1101 return user
1102
1102
1103
1103
1104 @pytest.fixture
1104 @pytest.fixture
1105 def user_util(request, pylonsapp):
1105 def user_util(request, pylonsapp):
1106 """
1106 """
1107 Provides a wired instance of `UserUtility` with integrated cleanup.
1107 Provides a wired instance of `UserUtility` with integrated cleanup.
1108 """
1108 """
1109 utility = UserUtility(test_name=request.node.name)
1109 utility = UserUtility(test_name=request.node.name)
1110 request.addfinalizer(utility.cleanup)
1110 request.addfinalizer(utility.cleanup)
1111 return utility
1111 return utility
1112
1112
1113
1113
1114 # TODO: johbo: Split this up into utilities per domain or something similar
1114 # TODO: johbo: Split this up into utilities per domain or something similar
1115 class UserUtility(object):
1115 class UserUtility(object):
1116
1116
1117 def __init__(self, test_name="test"):
1117 def __init__(self, test_name="test"):
1118 self._test_name = self._sanitize_name(test_name)
1118 self._test_name = self._sanitize_name(test_name)
1119 self.fixture = Fixture()
1119 self.fixture = Fixture()
1120 self.repo_group_ids = []
1120 self.repo_group_ids = []
1121 self.repos_ids = []
1121 self.repos_ids = []
1122 self.user_ids = []
1122 self.user_ids = []
1123 self.user_group_ids = []
1123 self.user_group_ids = []
1124 self.user_repo_permission_ids = []
1124 self.user_repo_permission_ids = []
1125 self.user_group_repo_permission_ids = []
1125 self.user_group_repo_permission_ids = []
1126 self.user_repo_group_permission_ids = []
1126 self.user_repo_group_permission_ids = []
1127 self.user_group_repo_group_permission_ids = []
1127 self.user_group_repo_group_permission_ids = []
1128 self.user_user_group_permission_ids = []
1128 self.user_user_group_permission_ids = []
1129 self.user_group_user_group_permission_ids = []
1129 self.user_group_user_group_permission_ids = []
1130 self.user_permissions = []
1130 self.user_permissions = []
1131
1131
1132 def _sanitize_name(self, name):
1132 def _sanitize_name(self, name):
1133 for char in ['[', ']']:
1133 for char in ['[', ']']:
1134 name = name.replace(char, '_')
1134 name = name.replace(char, '_')
1135 return name
1135 return name
1136
1136
1137 def create_repo_group(
1137 def create_repo_group(
1138 self, owner=TEST_USER_ADMIN_LOGIN, auto_cleanup=True):
1138 self, owner=TEST_USER_ADMIN_LOGIN, auto_cleanup=True):
1139 group_name = "{prefix}_repogroup_{count}".format(
1139 group_name = "{prefix}_repogroup_{count}".format(
1140 prefix=self._test_name,
1140 prefix=self._test_name,
1141 count=len(self.repo_group_ids))
1141 count=len(self.repo_group_ids))
1142 repo_group = self.fixture.create_repo_group(
1142 repo_group = self.fixture.create_repo_group(
1143 group_name, cur_user=owner)
1143 group_name, cur_user=owner)
1144 if auto_cleanup:
1144 if auto_cleanup:
1145 self.repo_group_ids.append(repo_group.group_id)
1145 self.repo_group_ids.append(repo_group.group_id)
1146 return repo_group
1146 return repo_group
1147
1147
1148 def create_repo(self, owner=TEST_USER_ADMIN_LOGIN, parent=None, auto_cleanup=True):
1148 def create_repo(self, owner=TEST_USER_ADMIN_LOGIN, parent=None,
1149 auto_cleanup=True, repo_type='hg'):
1149 repo_name = "{prefix}_repository_{count}".format(
1150 repo_name = "{prefix}_repository_{count}".format(
1150 prefix=self._test_name,
1151 prefix=self._test_name,
1151 count=len(self.repos_ids))
1152 count=len(self.repos_ids))
1152
1153
1153 repository = self.fixture.create_repo(
1154 repository = self.fixture.create_repo(
1154 repo_name, cur_user=owner, repo_group=parent)
1155 repo_name, cur_user=owner, repo_group=parent, repo_type=repo_type)
1155 if auto_cleanup:
1156 if auto_cleanup:
1156 self.repos_ids.append(repository.repo_id)
1157 self.repos_ids.append(repository.repo_id)
1157 return repository
1158 return repository
1158
1159
1159 def create_user(self, auto_cleanup=True, **kwargs):
1160 def create_user(self, auto_cleanup=True, **kwargs):
1160 user_name = "{prefix}_user_{count}".format(
1161 user_name = "{prefix}_user_{count}".format(
1161 prefix=self._test_name,
1162 prefix=self._test_name,
1162 count=len(self.user_ids))
1163 count=len(self.user_ids))
1163 user = self.fixture.create_user(user_name, **kwargs)
1164 user = self.fixture.create_user(user_name, **kwargs)
1164 if auto_cleanup:
1165 if auto_cleanup:
1165 self.user_ids.append(user.user_id)
1166 self.user_ids.append(user.user_id)
1166 return user
1167 return user
1167
1168
1168 def create_user_with_group(self):
1169 def create_user_with_group(self):
1169 user = self.create_user()
1170 user = self.create_user()
1170 user_group = self.create_user_group(members=[user])
1171 user_group = self.create_user_group(members=[user])
1171 return user, user_group
1172 return user, user_group
1172
1173
1173 def create_user_group(self, owner=TEST_USER_ADMIN_LOGIN, members=None,
1174 def create_user_group(self, owner=TEST_USER_ADMIN_LOGIN, members=None,
1174 auto_cleanup=True, **kwargs):
1175 auto_cleanup=True, **kwargs):
1175 group_name = "{prefix}_usergroup_{count}".format(
1176 group_name = "{prefix}_usergroup_{count}".format(
1176 prefix=self._test_name,
1177 prefix=self._test_name,
1177 count=len(self.user_group_ids))
1178 count=len(self.user_group_ids))
1178 user_group = self.fixture.create_user_group(
1179 user_group = self.fixture.create_user_group(
1179 group_name, cur_user=owner, **kwargs)
1180 group_name, cur_user=owner, **kwargs)
1180
1181
1181 if auto_cleanup:
1182 if auto_cleanup:
1182 self.user_group_ids.append(user_group.users_group_id)
1183 self.user_group_ids.append(user_group.users_group_id)
1183 if members:
1184 if members:
1184 for user in members:
1185 for user in members:
1185 UserGroupModel().add_user_to_group(user_group, user)
1186 UserGroupModel().add_user_to_group(user_group, user)
1186 return user_group
1187 return user_group
1187
1188
1188 def grant_user_permission(self, user_name, permission_name):
1189 def grant_user_permission(self, user_name, permission_name):
1189 self._inherit_default_user_permissions(user_name, False)
1190 self._inherit_default_user_permissions(user_name, False)
1190 self.user_permissions.append((user_name, permission_name))
1191 self.user_permissions.append((user_name, permission_name))
1191
1192
1192 def grant_user_permission_to_repo_group(
1193 def grant_user_permission_to_repo_group(
1193 self, repo_group, user, permission_name):
1194 self, repo_group, user, permission_name):
1194 permission = RepoGroupModel().grant_user_permission(
1195 permission = RepoGroupModel().grant_user_permission(
1195 repo_group, user, permission_name)
1196 repo_group, user, permission_name)
1196 self.user_repo_group_permission_ids.append(
1197 self.user_repo_group_permission_ids.append(
1197 (repo_group.group_id, user.user_id))
1198 (repo_group.group_id, user.user_id))
1198 return permission
1199 return permission
1199
1200
1200 def grant_user_group_permission_to_repo_group(
1201 def grant_user_group_permission_to_repo_group(
1201 self, repo_group, user_group, permission_name):
1202 self, repo_group, user_group, permission_name):
1202 permission = RepoGroupModel().grant_user_group_permission(
1203 permission = RepoGroupModel().grant_user_group_permission(
1203 repo_group, user_group, permission_name)
1204 repo_group, user_group, permission_name)
1204 self.user_group_repo_group_permission_ids.append(
1205 self.user_group_repo_group_permission_ids.append(
1205 (repo_group.group_id, user_group.users_group_id))
1206 (repo_group.group_id, user_group.users_group_id))
1206 return permission
1207 return permission
1207
1208
1208 def grant_user_permission_to_repo(
1209 def grant_user_permission_to_repo(
1209 self, repo, user, permission_name):
1210 self, repo, user, permission_name):
1210 permission = RepoModel().grant_user_permission(
1211 permission = RepoModel().grant_user_permission(
1211 repo, user, permission_name)
1212 repo, user, permission_name)
1212 self.user_repo_permission_ids.append(
1213 self.user_repo_permission_ids.append(
1213 (repo.repo_id, user.user_id))
1214 (repo.repo_id, user.user_id))
1214 return permission
1215 return permission
1215
1216
1216 def grant_user_group_permission_to_repo(
1217 def grant_user_group_permission_to_repo(
1217 self, repo, user_group, permission_name):
1218 self, repo, user_group, permission_name):
1218 permission = RepoModel().grant_user_group_permission(
1219 permission = RepoModel().grant_user_group_permission(
1219 repo, user_group, permission_name)
1220 repo, user_group, permission_name)
1220 self.user_group_repo_permission_ids.append(
1221 self.user_group_repo_permission_ids.append(
1221 (repo.repo_id, user_group.users_group_id))
1222 (repo.repo_id, user_group.users_group_id))
1222 return permission
1223 return permission
1223
1224
1224 def grant_user_permission_to_user_group(
1225 def grant_user_permission_to_user_group(
1225 self, target_user_group, user, permission_name):
1226 self, target_user_group, user, permission_name):
1226 permission = UserGroupModel().grant_user_permission(
1227 permission = UserGroupModel().grant_user_permission(
1227 target_user_group, user, permission_name)
1228 target_user_group, user, permission_name)
1228 self.user_user_group_permission_ids.append(
1229 self.user_user_group_permission_ids.append(
1229 (target_user_group.users_group_id, user.user_id))
1230 (target_user_group.users_group_id, user.user_id))
1230 return permission
1231 return permission
1231
1232
1232 def grant_user_group_permission_to_user_group(
1233 def grant_user_group_permission_to_user_group(
1233 self, target_user_group, user_group, permission_name):
1234 self, target_user_group, user_group, permission_name):
1234 permission = UserGroupModel().grant_user_group_permission(
1235 permission = UserGroupModel().grant_user_group_permission(
1235 target_user_group, user_group, permission_name)
1236 target_user_group, user_group, permission_name)
1236 self.user_group_user_group_permission_ids.append(
1237 self.user_group_user_group_permission_ids.append(
1237 (target_user_group.users_group_id, user_group.users_group_id))
1238 (target_user_group.users_group_id, user_group.users_group_id))
1238 return permission
1239 return permission
1239
1240
1240 def revoke_user_permission(self, user_name, permission_name):
1241 def revoke_user_permission(self, user_name, permission_name):
1241 self._inherit_default_user_permissions(user_name, True)
1242 self._inherit_default_user_permissions(user_name, True)
1242 UserModel().revoke_perm(user_name, permission_name)
1243 UserModel().revoke_perm(user_name, permission_name)
1243
1244
1244 def _inherit_default_user_permissions(self, user_name, value):
1245 def _inherit_default_user_permissions(self, user_name, value):
1245 user = UserModel().get_by_username(user_name)
1246 user = UserModel().get_by_username(user_name)
1246 user.inherit_default_permissions = value
1247 user.inherit_default_permissions = value
1247 Session().add(user)
1248 Session().add(user)
1248 Session().commit()
1249 Session().commit()
1249
1250
1250 def cleanup(self):
1251 def cleanup(self):
1251 self._cleanup_permissions()
1252 self._cleanup_permissions()
1252 self._cleanup_repos()
1253 self._cleanup_repos()
1253 self._cleanup_repo_groups()
1254 self._cleanup_repo_groups()
1254 self._cleanup_user_groups()
1255 self._cleanup_user_groups()
1255 self._cleanup_users()
1256 self._cleanup_users()
1256
1257
1257 def _cleanup_permissions(self):
1258 def _cleanup_permissions(self):
1258 if self.user_permissions:
1259 if self.user_permissions:
1259 for user_name, permission_name in self.user_permissions:
1260 for user_name, permission_name in self.user_permissions:
1260 self.revoke_user_permission(user_name, permission_name)
1261 self.revoke_user_permission(user_name, permission_name)
1261
1262
1262 for permission in self.user_repo_permission_ids:
1263 for permission in self.user_repo_permission_ids:
1263 RepoModel().revoke_user_permission(*permission)
1264 RepoModel().revoke_user_permission(*permission)
1264
1265
1265 for permission in self.user_group_repo_permission_ids:
1266 for permission in self.user_group_repo_permission_ids:
1266 RepoModel().revoke_user_group_permission(*permission)
1267 RepoModel().revoke_user_group_permission(*permission)
1267
1268
1268 for permission in self.user_repo_group_permission_ids:
1269 for permission in self.user_repo_group_permission_ids:
1269 RepoGroupModel().revoke_user_permission(*permission)
1270 RepoGroupModel().revoke_user_permission(*permission)
1270
1271
1271 for permission in self.user_group_repo_group_permission_ids:
1272 for permission in self.user_group_repo_group_permission_ids:
1272 RepoGroupModel().revoke_user_group_permission(*permission)
1273 RepoGroupModel().revoke_user_group_permission(*permission)
1273
1274
1274 for permission in self.user_user_group_permission_ids:
1275 for permission in self.user_user_group_permission_ids:
1275 UserGroupModel().revoke_user_permission(*permission)
1276 UserGroupModel().revoke_user_permission(*permission)
1276
1277
1277 for permission in self.user_group_user_group_permission_ids:
1278 for permission in self.user_group_user_group_permission_ids:
1278 UserGroupModel().revoke_user_group_permission(*permission)
1279 UserGroupModel().revoke_user_group_permission(*permission)
1279
1280
1280 def _cleanup_repo_groups(self):
1281 def _cleanup_repo_groups(self):
1281 def _repo_group_compare(first_group_id, second_group_id):
1282 def _repo_group_compare(first_group_id, second_group_id):
1282 """
1283 """
1283 Gives higher priority to the groups with the most complex paths
1284 Gives higher priority to the groups with the most complex paths
1284 """
1285 """
1285 first_group = RepoGroup.get(first_group_id)
1286 first_group = RepoGroup.get(first_group_id)
1286 second_group = RepoGroup.get(second_group_id)
1287 second_group = RepoGroup.get(second_group_id)
1287 first_group_parts = (
1288 first_group_parts = (
1288 len(first_group.group_name.split('/')) if first_group else 0)
1289 len(first_group.group_name.split('/')) if first_group else 0)
1289 second_group_parts = (
1290 second_group_parts = (
1290 len(second_group.group_name.split('/')) if second_group else 0)
1291 len(second_group.group_name.split('/')) if second_group else 0)
1291 return cmp(second_group_parts, first_group_parts)
1292 return cmp(second_group_parts, first_group_parts)
1292
1293
1293 sorted_repo_group_ids = sorted(
1294 sorted_repo_group_ids = sorted(
1294 self.repo_group_ids, cmp=_repo_group_compare)
1295 self.repo_group_ids, cmp=_repo_group_compare)
1295 for repo_group_id in sorted_repo_group_ids:
1296 for repo_group_id in sorted_repo_group_ids:
1296 self.fixture.destroy_repo_group(repo_group_id)
1297 self.fixture.destroy_repo_group(repo_group_id)
1297
1298
1298 def _cleanup_repos(self):
1299 def _cleanup_repos(self):
1299 sorted_repos_ids = sorted(self.repos_ids)
1300 sorted_repos_ids = sorted(self.repos_ids)
1300 for repo_id in sorted_repos_ids:
1301 for repo_id in sorted_repos_ids:
1301 self.fixture.destroy_repo(repo_id)
1302 self.fixture.destroy_repo(repo_id)
1302
1303
1303 def _cleanup_user_groups(self):
1304 def _cleanup_user_groups(self):
1304 def _user_group_compare(first_group_id, second_group_id):
1305 def _user_group_compare(first_group_id, second_group_id):
1305 """
1306 """
1306 Gives higher priority to the groups with the most complex paths
1307 Gives higher priority to the groups with the most complex paths
1307 """
1308 """
1308 first_group = UserGroup.get(first_group_id)
1309 first_group = UserGroup.get(first_group_id)
1309 second_group = UserGroup.get(second_group_id)
1310 second_group = UserGroup.get(second_group_id)
1310 first_group_parts = (
1311 first_group_parts = (
1311 len(first_group.users_group_name.split('/'))
1312 len(first_group.users_group_name.split('/'))
1312 if first_group else 0)
1313 if first_group else 0)
1313 second_group_parts = (
1314 second_group_parts = (
1314 len(second_group.users_group_name.split('/'))
1315 len(second_group.users_group_name.split('/'))
1315 if second_group else 0)
1316 if second_group else 0)
1316 return cmp(second_group_parts, first_group_parts)
1317 return cmp(second_group_parts, first_group_parts)
1317
1318
1318 sorted_user_group_ids = sorted(
1319 sorted_user_group_ids = sorted(
1319 self.user_group_ids, cmp=_user_group_compare)
1320 self.user_group_ids, cmp=_user_group_compare)
1320 for user_group_id in sorted_user_group_ids:
1321 for user_group_id in sorted_user_group_ids:
1321 self.fixture.destroy_user_group(user_group_id)
1322 self.fixture.destroy_user_group(user_group_id)
1322
1323
1323 def _cleanup_users(self):
1324 def _cleanup_users(self):
1324 for user_id in self.user_ids:
1325 for user_id in self.user_ids:
1325 self.fixture.destroy_user(user_id)
1326 self.fixture.destroy_user(user_id)
1326
1327
1327
1328
1328 # TODO: Think about moving this into a pytest-pyro package and make it a
1329 # TODO: Think about moving this into a pytest-pyro package and make it a
1329 # pytest plugin
1330 # pytest plugin
1330 @pytest.hookimpl(tryfirst=True, hookwrapper=True)
1331 @pytest.hookimpl(tryfirst=True, hookwrapper=True)
1331 def pytest_runtest_makereport(item, call):
1332 def pytest_runtest_makereport(item, call):
1332 """
1333 """
1333 Adding the remote traceback if the exception has this information.
1334 Adding the remote traceback if the exception has this information.
1334
1335
1335 VCSServer attaches this information as the attribute `_vcs_server_traceback`
1336 VCSServer attaches this information as the attribute `_vcs_server_traceback`
1336 to the exception instance.
1337 to the exception instance.
1337 """
1338 """
1338 outcome = yield
1339 outcome = yield
1339 report = outcome.get_result()
1340 report = outcome.get_result()
1340 if call.excinfo:
1341 if call.excinfo:
1341 _add_vcsserver_remote_traceback(report, call.excinfo.value)
1342 _add_vcsserver_remote_traceback(report, call.excinfo.value)
1342
1343
1343
1344
1344 def _add_vcsserver_remote_traceback(report, exc):
1345 def _add_vcsserver_remote_traceback(report, exc):
1345 vcsserver_traceback = getattr(exc, '_vcs_server_traceback', None)
1346 vcsserver_traceback = getattr(exc, '_vcs_server_traceback', None)
1346
1347
1347 if vcsserver_traceback:
1348 if vcsserver_traceback:
1348 section = 'VCSServer remote traceback ' + report.when
1349 section = 'VCSServer remote traceback ' + report.when
1349 report.sections.append((section, vcsserver_traceback))
1350 report.sections.append((section, vcsserver_traceback))
1350
1351
1351
1352
1352 @pytest.fixture(scope='session')
1353 @pytest.fixture(scope='session')
1353 def testrun():
1354 def testrun():
1354 return {
1355 return {
1355 'uuid': uuid.uuid4(),
1356 'uuid': uuid.uuid4(),
1356 'start': datetime.datetime.utcnow().isoformat(),
1357 'start': datetime.datetime.utcnow().isoformat(),
1357 'timestamp': int(time.time()),
1358 'timestamp': int(time.time()),
1358 }
1359 }
1359
1360
1360
1361
1361 @pytest.fixture(autouse=True)
1362 @pytest.fixture(autouse=True)
1362 def collect_appenlight_stats(request, testrun):
1363 def collect_appenlight_stats(request, testrun):
1363 """
1364 """
1364 This fixture reports memory consumtion of single tests.
1365 This fixture reports memory consumtion of single tests.
1365
1366
1366 It gathers data based on `psutil` and sends them to Appenlight. The option
1367 It gathers data based on `psutil` and sends them to Appenlight. The option
1367 ``--ae`` has te be used to enable this fixture and the API key for your
1368 ``--ae`` has te be used to enable this fixture and the API key for your
1368 application has to be provided in ``--ae-key``.
1369 application has to be provided in ``--ae-key``.
1369 """
1370 """
1370 try:
1371 try:
1371 # cygwin cannot have yet psutil support.
1372 # cygwin cannot have yet psutil support.
1372 import psutil
1373 import psutil
1373 except ImportError:
1374 except ImportError:
1374 return
1375 return
1375
1376
1376 if not request.config.getoption('--appenlight'):
1377 if not request.config.getoption('--appenlight'):
1377 return
1378 return
1378 else:
1379 else:
1379 # Only request the pylonsapp fixture if appenlight tracking is
1380 # Only request the pylonsapp fixture if appenlight tracking is
1380 # enabled. This will speed up a test run of unit tests by 2 to 3
1381 # enabled. This will speed up a test run of unit tests by 2 to 3
1381 # seconds if appenlight is not enabled.
1382 # seconds if appenlight is not enabled.
1382 pylonsapp = request.getfuncargvalue("pylonsapp")
1383 pylonsapp = request.getfuncargvalue("pylonsapp")
1383 url = '{}/api/logs'.format(request.config.getoption('--appenlight-url'))
1384 url = '{}/api/logs'.format(request.config.getoption('--appenlight-url'))
1384 client = AppenlightClient(
1385 client = AppenlightClient(
1385 url=url,
1386 url=url,
1386 api_key=request.config.getoption('--appenlight-api-key'),
1387 api_key=request.config.getoption('--appenlight-api-key'),
1387 namespace=request.node.nodeid,
1388 namespace=request.node.nodeid,
1388 request=str(testrun['uuid']),
1389 request=str(testrun['uuid']),
1389 testrun=testrun)
1390 testrun=testrun)
1390
1391
1391 client.collect({
1392 client.collect({
1392 'message': "Starting",
1393 'message': "Starting",
1393 })
1394 })
1394
1395
1395 server_and_port = pylonsapp.config['vcs.server']
1396 server_and_port = pylonsapp.config['vcs.server']
1396 protocol = pylonsapp.config['vcs.server.protocol']
1397 protocol = pylonsapp.config['vcs.server.protocol']
1397 server = create_vcsserver_proxy(server_and_port, protocol)
1398 server = create_vcsserver_proxy(server_and_port, protocol)
1398 with server:
1399 with server:
1399 vcs_pid = server.get_pid()
1400 vcs_pid = server.get_pid()
1400 server.run_gc()
1401 server.run_gc()
1401 vcs_process = psutil.Process(vcs_pid)
1402 vcs_process = psutil.Process(vcs_pid)
1402 mem = vcs_process.memory_info()
1403 mem = vcs_process.memory_info()
1403 client.tag_before('vcsserver.rss', mem.rss)
1404 client.tag_before('vcsserver.rss', mem.rss)
1404 client.tag_before('vcsserver.vms', mem.vms)
1405 client.tag_before('vcsserver.vms', mem.vms)
1405
1406
1406 test_process = psutil.Process()
1407 test_process = psutil.Process()
1407 mem = test_process.memory_info()
1408 mem = test_process.memory_info()
1408 client.tag_before('test.rss', mem.rss)
1409 client.tag_before('test.rss', mem.rss)
1409 client.tag_before('test.vms', mem.vms)
1410 client.tag_before('test.vms', mem.vms)
1410
1411
1411 client.tag_before('time', time.time())
1412 client.tag_before('time', time.time())
1412
1413
1413 @request.addfinalizer
1414 @request.addfinalizer
1414 def send_stats():
1415 def send_stats():
1415 client.tag_after('time', time.time())
1416 client.tag_after('time', time.time())
1416 with server:
1417 with server:
1417 gc_stats = server.run_gc()
1418 gc_stats = server.run_gc()
1418 for tag, value in gc_stats.items():
1419 for tag, value in gc_stats.items():
1419 client.tag_after(tag, value)
1420 client.tag_after(tag, value)
1420 mem = vcs_process.memory_info()
1421 mem = vcs_process.memory_info()
1421 client.tag_after('vcsserver.rss', mem.rss)
1422 client.tag_after('vcsserver.rss', mem.rss)
1422 client.tag_after('vcsserver.vms', mem.vms)
1423 client.tag_after('vcsserver.vms', mem.vms)
1423
1424
1424 mem = test_process.memory_info()
1425 mem = test_process.memory_info()
1425 client.tag_after('test.rss', mem.rss)
1426 client.tag_after('test.rss', mem.rss)
1426 client.tag_after('test.vms', mem.vms)
1427 client.tag_after('test.vms', mem.vms)
1427
1428
1428 client.collect({
1429 client.collect({
1429 'message': "Finished",
1430 'message': "Finished",
1430 })
1431 })
1431 client.send_stats()
1432 client.send_stats()
1432
1433
1433 return client
1434 return client
1434
1435
1435
1436
1436 class AppenlightClient():
1437 class AppenlightClient():
1437
1438
1438 url_template = '{url}?protocol_version=0.5'
1439 url_template = '{url}?protocol_version=0.5'
1439
1440
1440 def __init__(
1441 def __init__(
1441 self, url, api_key, add_server=True, add_timestamp=True,
1442 self, url, api_key, add_server=True, add_timestamp=True,
1442 namespace=None, request=None, testrun=None):
1443 namespace=None, request=None, testrun=None):
1443 self.url = self.url_template.format(url=url)
1444 self.url = self.url_template.format(url=url)
1444 self.api_key = api_key
1445 self.api_key = api_key
1445 self.add_server = add_server
1446 self.add_server = add_server
1446 self.add_timestamp = add_timestamp
1447 self.add_timestamp = add_timestamp
1447 self.namespace = namespace
1448 self.namespace = namespace
1448 self.request = request
1449 self.request = request
1449 self.server = socket.getfqdn(socket.gethostname())
1450 self.server = socket.getfqdn(socket.gethostname())
1450 self.tags_before = {}
1451 self.tags_before = {}
1451 self.tags_after = {}
1452 self.tags_after = {}
1452 self.stats = []
1453 self.stats = []
1453 self.testrun = testrun or {}
1454 self.testrun = testrun or {}
1454
1455
1455 def tag_before(self, tag, value):
1456 def tag_before(self, tag, value):
1456 self.tags_before[tag] = value
1457 self.tags_before[tag] = value
1457
1458
1458 def tag_after(self, tag, value):
1459 def tag_after(self, tag, value):
1459 self.tags_after[tag] = value
1460 self.tags_after[tag] = value
1460
1461
1461 def collect(self, data):
1462 def collect(self, data):
1462 if self.add_server:
1463 if self.add_server:
1463 data.setdefault('server', self.server)
1464 data.setdefault('server', self.server)
1464 if self.add_timestamp:
1465 if self.add_timestamp:
1465 data.setdefault('date', datetime.datetime.utcnow().isoformat())
1466 data.setdefault('date', datetime.datetime.utcnow().isoformat())
1466 if self.namespace:
1467 if self.namespace:
1467 data.setdefault('namespace', self.namespace)
1468 data.setdefault('namespace', self.namespace)
1468 if self.request:
1469 if self.request:
1469 data.setdefault('request', self.request)
1470 data.setdefault('request', self.request)
1470 self.stats.append(data)
1471 self.stats.append(data)
1471
1472
1472 def send_stats(self):
1473 def send_stats(self):
1473 tags = [
1474 tags = [
1474 ('testrun', self.request),
1475 ('testrun', self.request),
1475 ('testrun.start', self.testrun['start']),
1476 ('testrun.start', self.testrun['start']),
1476 ('testrun.timestamp', self.testrun['timestamp']),
1477 ('testrun.timestamp', self.testrun['timestamp']),
1477 ('test', self.namespace),
1478 ('test', self.namespace),
1478 ]
1479 ]
1479 for key, value in self.tags_before.items():
1480 for key, value in self.tags_before.items():
1480 tags.append((key + '.before', value))
1481 tags.append((key + '.before', value))
1481 try:
1482 try:
1482 delta = self.tags_after[key] - value
1483 delta = self.tags_after[key] - value
1483 tags.append((key + '.delta', delta))
1484 tags.append((key + '.delta', delta))
1484 except Exception:
1485 except Exception:
1485 pass
1486 pass
1486 for key, value in self.tags_after.items():
1487 for key, value in self.tags_after.items():
1487 tags.append((key + '.after', value))
1488 tags.append((key + '.after', value))
1488 self.collect({
1489 self.collect({
1489 'message': "Collected tags",
1490 'message': "Collected tags",
1490 'tags': tags,
1491 'tags': tags,
1491 })
1492 })
1492
1493
1493 response = requests.post(
1494 response = requests.post(
1494 self.url,
1495 self.url,
1495 headers={
1496 headers={
1496 'X-appenlight-api-key': self.api_key},
1497 'X-appenlight-api-key': self.api_key},
1497 json=self.stats,
1498 json=self.stats,
1498 )
1499 )
1499
1500
1500 if not response.status_code == 200:
1501 if not response.status_code == 200:
1501 pprint.pprint(self.stats)
1502 pprint.pprint(self.stats)
1502 print response.headers
1503 print response.headers
1503 print response.text
1504 print response.text
1504 raise Exception('Sending to appenlight failed')
1505 raise Exception('Sending to appenlight failed')
1505
1506
1506
1507
1507 @pytest.fixture
1508 @pytest.fixture
1508 def gist_util(request, pylonsapp):
1509 def gist_util(request, pylonsapp):
1509 """
1510 """
1510 Provides a wired instance of `GistUtility` with integrated cleanup.
1511 Provides a wired instance of `GistUtility` with integrated cleanup.
1511 """
1512 """
1512 utility = GistUtility()
1513 utility = GistUtility()
1513 request.addfinalizer(utility.cleanup)
1514 request.addfinalizer(utility.cleanup)
1514 return utility
1515 return utility
1515
1516
1516
1517
1517 class GistUtility(object):
1518 class GistUtility(object):
1518 def __init__(self):
1519 def __init__(self):
1519 self.fixture = Fixture()
1520 self.fixture = Fixture()
1520 self.gist_ids = []
1521 self.gist_ids = []
1521
1522
1522 def create_gist(self, **kwargs):
1523 def create_gist(self, **kwargs):
1523 gist = self.fixture.create_gist(**kwargs)
1524 gist = self.fixture.create_gist(**kwargs)
1524 self.gist_ids.append(gist.gist_id)
1525 self.gist_ids.append(gist.gist_id)
1525 return gist
1526 return gist
1526
1527
1527 def cleanup(self):
1528 def cleanup(self):
1528 for id_ in self.gist_ids:
1529 for id_ in self.gist_ids:
1529 self.fixture.destroy_gists(str(id_))
1530 self.fixture.destroy_gists(str(id_))
1530
1531
1531
1532
1532 @pytest.fixture
1533 @pytest.fixture
1533 def enabled_backends(request):
1534 def enabled_backends(request):
1534 backends = request.config.option.backends
1535 backends = request.config.option.backends
1535 return backends[:]
1536 return backends[:]
1536
1537
1537
1538
1538 @pytest.fixture
1539 @pytest.fixture
1539 def settings_util(request):
1540 def settings_util(request):
1540 """
1541 """
1541 Provides a wired instance of `SettingsUtility` with integrated cleanup.
1542 Provides a wired instance of `SettingsUtility` with integrated cleanup.
1542 """
1543 """
1543 utility = SettingsUtility()
1544 utility = SettingsUtility()
1544 request.addfinalizer(utility.cleanup)
1545 request.addfinalizer(utility.cleanup)
1545 return utility
1546 return utility
1546
1547
1547
1548
1548 class SettingsUtility(object):
1549 class SettingsUtility(object):
1549 def __init__(self):
1550 def __init__(self):
1550 self.rhodecode_ui_ids = []
1551 self.rhodecode_ui_ids = []
1551 self.rhodecode_setting_ids = []
1552 self.rhodecode_setting_ids = []
1552 self.repo_rhodecode_ui_ids = []
1553 self.repo_rhodecode_ui_ids = []
1553 self.repo_rhodecode_setting_ids = []
1554 self.repo_rhodecode_setting_ids = []
1554
1555
1555 def create_repo_rhodecode_ui(
1556 def create_repo_rhodecode_ui(
1556 self, repo, section, value, key=None, active=True, cleanup=True):
1557 self, repo, section, value, key=None, active=True, cleanup=True):
1557 key = key or hashlib.sha1(
1558 key = key or hashlib.sha1(
1558 '{}{}{}'.format(section, value, repo.repo_id)).hexdigest()
1559 '{}{}{}'.format(section, value, repo.repo_id)).hexdigest()
1559
1560
1560 setting = RepoRhodeCodeUi()
1561 setting = RepoRhodeCodeUi()
1561 setting.repository_id = repo.repo_id
1562 setting.repository_id = repo.repo_id
1562 setting.ui_section = section
1563 setting.ui_section = section
1563 setting.ui_value = value
1564 setting.ui_value = value
1564 setting.ui_key = key
1565 setting.ui_key = key
1565 setting.ui_active = active
1566 setting.ui_active = active
1566 Session().add(setting)
1567 Session().add(setting)
1567 Session().commit()
1568 Session().commit()
1568
1569
1569 if cleanup:
1570 if cleanup:
1570 self.repo_rhodecode_ui_ids.append(setting.ui_id)
1571 self.repo_rhodecode_ui_ids.append(setting.ui_id)
1571 return setting
1572 return setting
1572
1573
1573 def create_rhodecode_ui(
1574 def create_rhodecode_ui(
1574 self, section, value, key=None, active=True, cleanup=True):
1575 self, section, value, key=None, active=True, cleanup=True):
1575 key = key or hashlib.sha1('{}{}'.format(section, value)).hexdigest()
1576 key = key or hashlib.sha1('{}{}'.format(section, value)).hexdigest()
1576
1577
1577 setting = RhodeCodeUi()
1578 setting = RhodeCodeUi()
1578 setting.ui_section = section
1579 setting.ui_section = section
1579 setting.ui_value = value
1580 setting.ui_value = value
1580 setting.ui_key = key
1581 setting.ui_key = key
1581 setting.ui_active = active
1582 setting.ui_active = active
1582 Session().add(setting)
1583 Session().add(setting)
1583 Session().commit()
1584 Session().commit()
1584
1585
1585 if cleanup:
1586 if cleanup:
1586 self.rhodecode_ui_ids.append(setting.ui_id)
1587 self.rhodecode_ui_ids.append(setting.ui_id)
1587 return setting
1588 return setting
1588
1589
1589 def create_repo_rhodecode_setting(
1590 def create_repo_rhodecode_setting(
1590 self, repo, name, value, type_, cleanup=True):
1591 self, repo, name, value, type_, cleanup=True):
1591 setting = RepoRhodeCodeSetting(
1592 setting = RepoRhodeCodeSetting(
1592 repo.repo_id, key=name, val=value, type=type_)
1593 repo.repo_id, key=name, val=value, type=type_)
1593 Session().add(setting)
1594 Session().add(setting)
1594 Session().commit()
1595 Session().commit()
1595
1596
1596 if cleanup:
1597 if cleanup:
1597 self.repo_rhodecode_setting_ids.append(setting.app_settings_id)
1598 self.repo_rhodecode_setting_ids.append(setting.app_settings_id)
1598 return setting
1599 return setting
1599
1600
1600 def create_rhodecode_setting(self, name, value, type_, cleanup=True):
1601 def create_rhodecode_setting(self, name, value, type_, cleanup=True):
1601 setting = RhodeCodeSetting(key=name, val=value, type=type_)
1602 setting = RhodeCodeSetting(key=name, val=value, type=type_)
1602 Session().add(setting)
1603 Session().add(setting)
1603 Session().commit()
1604 Session().commit()
1604
1605
1605 if cleanup:
1606 if cleanup:
1606 self.rhodecode_setting_ids.append(setting.app_settings_id)
1607 self.rhodecode_setting_ids.append(setting.app_settings_id)
1607
1608
1608 return setting
1609 return setting
1609
1610
1610 def cleanup(self):
1611 def cleanup(self):
1611 for id_ in self.rhodecode_ui_ids:
1612 for id_ in self.rhodecode_ui_ids:
1612 setting = RhodeCodeUi.get(id_)
1613 setting = RhodeCodeUi.get(id_)
1613 Session().delete(setting)
1614 Session().delete(setting)
1614
1615
1615 for id_ in self.rhodecode_setting_ids:
1616 for id_ in self.rhodecode_setting_ids:
1616 setting = RhodeCodeSetting.get(id_)
1617 setting = RhodeCodeSetting.get(id_)
1617 Session().delete(setting)
1618 Session().delete(setting)
1618
1619
1619 for id_ in self.repo_rhodecode_ui_ids:
1620 for id_ in self.repo_rhodecode_ui_ids:
1620 setting = RepoRhodeCodeUi.get(id_)
1621 setting = RepoRhodeCodeUi.get(id_)
1621 Session().delete(setting)
1622 Session().delete(setting)
1622
1623
1623 for id_ in self.repo_rhodecode_setting_ids:
1624 for id_ in self.repo_rhodecode_setting_ids:
1624 setting = RepoRhodeCodeSetting.get(id_)
1625 setting = RepoRhodeCodeSetting.get(id_)
1625 Session().delete(setting)
1626 Session().delete(setting)
1626
1627
1627 Session().commit()
1628 Session().commit()
1628
1629
1629
1630
1630 @pytest.fixture
1631 @pytest.fixture
1631 def no_notifications(request):
1632 def no_notifications(request):
1632 notification_patcher = mock.patch(
1633 notification_patcher = mock.patch(
1633 'rhodecode.model.notification.NotificationModel.create')
1634 'rhodecode.model.notification.NotificationModel.create')
1634 notification_patcher.start()
1635 notification_patcher.start()
1635 request.addfinalizer(notification_patcher.stop)
1636 request.addfinalizer(notification_patcher.stop)
1636
1637
1637
1638
1638 @pytest.fixture
1639 @pytest.fixture
1639 def silence_action_logger(request):
1640 def silence_action_logger(request):
1640 notification_patcher = mock.patch(
1641 notification_patcher = mock.patch(
1641 'rhodecode.lib.utils.action_logger')
1642 'rhodecode.lib.utils.action_logger')
1642 notification_patcher.start()
1643 notification_patcher.start()
1643 request.addfinalizer(notification_patcher.stop)
1644 request.addfinalizer(notification_patcher.stop)
1644
1645
1645
1646
1646 @pytest.fixture(scope='session')
1647 @pytest.fixture(scope='session')
1647 def repeat(request):
1648 def repeat(request):
1648 """
1649 """
1649 The number of repetitions is based on this fixture.
1650 The number of repetitions is based on this fixture.
1650
1651
1651 Slower calls may divide it by 10 or 100. It is chosen in a way so that the
1652 Slower calls may divide it by 10 or 100. It is chosen in a way so that the
1652 tests are not too slow in our default test suite.
1653 tests are not too slow in our default test suite.
1653 """
1654 """
1654 return request.config.getoption('--repeat')
1655 return request.config.getoption('--repeat')
1655
1656
1656
1657
1657 @pytest.fixture
1658 @pytest.fixture
1658 def rhodecode_fixtures():
1659 def rhodecode_fixtures():
1659 return Fixture()
1660 return Fixture()
1660
1661
1661
1662
1662 @pytest.fixture
1663 @pytest.fixture
1663 def request_stub():
1664 def request_stub():
1664 """
1665 """
1665 Stub request object.
1666 Stub request object.
1666 """
1667 """
1667 request = pyramid.testing.DummyRequest()
1668 request = pyramid.testing.DummyRequest()
1668 request.scheme = 'https'
1669 request.scheme = 'https'
1669 return request
1670 return request
1670
1671
1671
1672
1672 @pytest.fixture
1673 @pytest.fixture
1673 def config_stub(request, request_stub):
1674 def config_stub(request, request_stub):
1674 """
1675 """
1675 Set up pyramid.testing and return the Configurator.
1676 Set up pyramid.testing and return the Configurator.
1676 """
1677 """
1677 config = pyramid.testing.setUp(request=request_stub)
1678 config = pyramid.testing.setUp(request=request_stub)
1678
1679
1679 @request.addfinalizer
1680 @request.addfinalizer
1680 def cleanup():
1681 def cleanup():
1681 pyramid.testing.tearDown()
1682 pyramid.testing.tearDown()
1682
1683
1683 return config
1684 return config
1684
1685
1685
1686
1686 @pytest.fixture
1687 @pytest.fixture
1687 def StubIntegrationType():
1688 def StubIntegrationType():
1688 class _StubIntegrationType(IntegrationTypeBase):
1689 class _StubIntegrationType(IntegrationTypeBase):
1689 """ Test integration type class """
1690 """ Test integration type class """
1690
1691
1691 key = 'test'
1692 key = 'test'
1692 display_name = 'Test integration type'
1693 display_name = 'Test integration type'
1693 description = 'A test integration type for testing'
1694 description = 'A test integration type for testing'
1694 icon = 'test_icon_html_image'
1695 icon = 'test_icon_html_image'
1695
1696
1696 def __init__(self, settings):
1697 def __init__(self, settings):
1697 super(_StubIntegrationType, self).__init__(settings)
1698 super(_StubIntegrationType, self).__init__(settings)
1698 self.sent_events = [] # for testing
1699 self.sent_events = [] # for testing
1699
1700
1700 def send_event(self, event):
1701 def send_event(self, event):
1701 self.sent_events.append(event)
1702 self.sent_events.append(event)
1702
1703
1703 def settings_schema(self):
1704 def settings_schema(self):
1704 class SettingsSchema(colander.Schema):
1705 class SettingsSchema(colander.Schema):
1705 test_string_field = colander.SchemaNode(
1706 test_string_field = colander.SchemaNode(
1706 colander.String(),
1707 colander.String(),
1707 missing=colander.required,
1708 missing=colander.required,
1708 title='test string field',
1709 title='test string field',
1709 )
1710 )
1710 test_int_field = colander.SchemaNode(
1711 test_int_field = colander.SchemaNode(
1711 colander.Int(),
1712 colander.Int(),
1712 title='some integer setting',
1713 title='some integer setting',
1713 )
1714 )
1714 return SettingsSchema()
1715 return SettingsSchema()
1715
1716
1716
1717
1717 integration_type_registry.register_integration_type(_StubIntegrationType)
1718 integration_type_registry.register_integration_type(_StubIntegrationType)
1718 return _StubIntegrationType
1719 return _StubIntegrationType
1719
1720
1720 @pytest.fixture
1721 @pytest.fixture
1721 def stub_integration_settings():
1722 def stub_integration_settings():
1722 return {
1723 return {
1723 'test_string_field': 'some data',
1724 'test_string_field': 'some data',
1724 'test_int_field': 100,
1725 'test_int_field': 100,
1725 }
1726 }
1726
1727
1727
1728
1728 @pytest.fixture
1729 @pytest.fixture
1729 def repo_integration_stub(request, repo_stub, StubIntegrationType,
1730 def repo_integration_stub(request, repo_stub, StubIntegrationType,
1730 stub_integration_settings):
1731 stub_integration_settings):
1731 integration = IntegrationModel().create(
1732 integration = IntegrationModel().create(
1732 StubIntegrationType, settings=stub_integration_settings, enabled=True,
1733 StubIntegrationType, settings=stub_integration_settings, enabled=True,
1733 name='test repo integration',
1734 name='test repo integration',
1734 repo=repo_stub, repo_group=None, child_repos_only=None)
1735 repo=repo_stub, repo_group=None, child_repos_only=None)
1735
1736
1736 @request.addfinalizer
1737 @request.addfinalizer
1737 def cleanup():
1738 def cleanup():
1738 IntegrationModel().delete(integration)
1739 IntegrationModel().delete(integration)
1739
1740
1740 return integration
1741 return integration
1741
1742
1742
1743
1743 @pytest.fixture
1744 @pytest.fixture
1744 def repogroup_integration_stub(request, test_repo_group, StubIntegrationType,
1745 def repogroup_integration_stub(request, test_repo_group, StubIntegrationType,
1745 stub_integration_settings):
1746 stub_integration_settings):
1746 integration = IntegrationModel().create(
1747 integration = IntegrationModel().create(
1747 StubIntegrationType, settings=stub_integration_settings, enabled=True,
1748 StubIntegrationType, settings=stub_integration_settings, enabled=True,
1748 name='test repogroup integration',
1749 name='test repogroup integration',
1749 repo=None, repo_group=test_repo_group, child_repos_only=True)
1750 repo=None, repo_group=test_repo_group, child_repos_only=True)
1750
1751
1751 @request.addfinalizer
1752 @request.addfinalizer
1752 def cleanup():
1753 def cleanup():
1753 IntegrationModel().delete(integration)
1754 IntegrationModel().delete(integration)
1754
1755
1755 return integration
1756 return integration
1756
1757
1757
1758
1758 @pytest.fixture
1759 @pytest.fixture
1759 def repogroup_recursive_integration_stub(request, test_repo_group,
1760 def repogroup_recursive_integration_stub(request, test_repo_group,
1760 StubIntegrationType, stub_integration_settings):
1761 StubIntegrationType, stub_integration_settings):
1761 integration = IntegrationModel().create(
1762 integration = IntegrationModel().create(
1762 StubIntegrationType, settings=stub_integration_settings, enabled=True,
1763 StubIntegrationType, settings=stub_integration_settings, enabled=True,
1763 name='test recursive repogroup integration',
1764 name='test recursive repogroup integration',
1764 repo=None, repo_group=test_repo_group, child_repos_only=False)
1765 repo=None, repo_group=test_repo_group, child_repos_only=False)
1765
1766
1766 @request.addfinalizer
1767 @request.addfinalizer
1767 def cleanup():
1768 def cleanup():
1768 IntegrationModel().delete(integration)
1769 IntegrationModel().delete(integration)
1769
1770
1770 return integration
1771 return integration
1771
1772
1772
1773
1773 @pytest.fixture
1774 @pytest.fixture
1774 def global_integration_stub(request, StubIntegrationType,
1775 def global_integration_stub(request, StubIntegrationType,
1775 stub_integration_settings):
1776 stub_integration_settings):
1776 integration = IntegrationModel().create(
1777 integration = IntegrationModel().create(
1777 StubIntegrationType, settings=stub_integration_settings, enabled=True,
1778 StubIntegrationType, settings=stub_integration_settings, enabled=True,
1778 name='test global integration',
1779 name='test global integration',
1779 repo=None, repo_group=None, child_repos_only=None)
1780 repo=None, repo_group=None, child_repos_only=None)
1780
1781
1781 @request.addfinalizer
1782 @request.addfinalizer
1782 def cleanup():
1783 def cleanup():
1783 IntegrationModel().delete(integration)
1784 IntegrationModel().delete(integration)
1784
1785
1785 return integration
1786 return integration
1786
1787
1787
1788
1788 @pytest.fixture
1789 @pytest.fixture
1789 def root_repos_integration_stub(request, StubIntegrationType,
1790 def root_repos_integration_stub(request, StubIntegrationType,
1790 stub_integration_settings):
1791 stub_integration_settings):
1791 integration = IntegrationModel().create(
1792 integration = IntegrationModel().create(
1792 StubIntegrationType, settings=stub_integration_settings, enabled=True,
1793 StubIntegrationType, settings=stub_integration_settings, enabled=True,
1793 name='test global integration',
1794 name='test global integration',
1794 repo=None, repo_group=None, child_repos_only=True)
1795 repo=None, repo_group=None, child_repos_only=True)
1795
1796
1796 @request.addfinalizer
1797 @request.addfinalizer
1797 def cleanup():
1798 def cleanup():
1798 IntegrationModel().delete(integration)
1799 IntegrationModel().delete(integration)
1799
1800
1800 return integration
1801 return integration
1801
1802
1802
1803
1803 @pytest.fixture
1804 @pytest.fixture
1804 def local_dt_to_utc():
1805 def local_dt_to_utc():
1805 def _factory(dt):
1806 def _factory(dt):
1806 return dt.replace(tzinfo=dateutil.tz.tzlocal()).astimezone(
1807 return dt.replace(tzinfo=dateutil.tz.tzlocal()).astimezone(
1807 dateutil.tz.tzutc()).replace(tzinfo=None)
1808 dateutil.tz.tzutc()).replace(tzinfo=None)
1808 return _factory
1809 return _factory
General Comments 0
You need to be logged in to leave comments. Login now