Show More
__init__.py
236 lines
| 7.4 KiB
| text/x-python
|
PythonLexer
r1 | # -*- coding: utf-8 -*- | |||
r2487 | # Copyright (C) 2010-2018 RhodeCode GmbH | |||
r1 | # | |||
# This program is free software: you can redistribute it and/or modify | ||||
# it under the terms of the GNU Affero General Public License, version 3 | ||||
# (only), as published by the Free Software Foundation. | ||||
# | ||||
# This program is distributed in the hope that it will be useful, | ||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||||
# GNU General Public License for more details. | ||||
# | ||||
# You should have received a copy of the GNU Affero General Public License | ||||
# along with this program. If not, see <http://www.gnu.org/licenses/>. | ||||
# | ||||
# This program is dual-licensed. If you wish to learn more about the | ||||
# RhodeCode Enterprise Edition, including its added features, Support services, | ||||
# and proprietary license terms, please see https://rhodecode.com/licenses/ | ||||
import os | ||||
import time | ||||
import logging | ||||
import datetime | ||||
import hashlib | ||||
import tempfile | ||||
from os.path import join as jn | ||||
from tempfile import _RandomNameSequence | ||||
import pytest | ||||
from rhodecode.model.db import User | ||||
from rhodecode.lib import auth | ||||
r1941 | from rhodecode.lib import helpers as h | |||
r1 | from rhodecode.lib.helpers import flash, link_to | |||
r2095 | from rhodecode.lib.utils2 import safe_str | |||
r1 | ||||
log = logging.getLogger(__name__) | ||||
__all__ = [ | ||||
r2278 | 'get_new_dir', 'TestController', | |||
r2650 | 'link_to', 'clear_all_caches', | |||
r1720 | 'assert_session_flash', 'login_user', 'no_newline_id_generator', | |||
r1 | 'TESTS_TMP_PATH', 'HG_REPO', 'GIT_REPO', 'SVN_REPO', | |||
'NEW_HG_REPO', 'NEW_GIT_REPO', | ||||
'HG_FORK', 'GIT_FORK', 'TEST_USER_ADMIN_LOGIN', 'TEST_USER_ADMIN_PASS', | ||||
'TEST_USER_REGULAR_LOGIN', 'TEST_USER_REGULAR_PASS', | ||||
'TEST_USER_REGULAR_EMAIL', 'TEST_USER_REGULAR2_LOGIN', | ||||
'TEST_USER_REGULAR2_PASS', 'TEST_USER_REGULAR2_EMAIL', 'TEST_HG_REPO', | ||||
'TEST_HG_REPO_CLONE', 'TEST_HG_REPO_PULL', 'TEST_GIT_REPO', | ||||
'TEST_GIT_REPO_CLONE', 'TEST_GIT_REPO_PULL', 'SCM_TESTS', | ||||
] | ||||
# SOME GLOBALS FOR TESTS | ||||
TEST_DIR = tempfile.gettempdir() | ||||
TESTS_TMP_PATH = jn(TEST_DIR, 'rc_test_%s' % _RandomNameSequence().next()) | ||||
TEST_USER_ADMIN_LOGIN = 'test_admin' | ||||
TEST_USER_ADMIN_PASS = 'test12' | ||||
TEST_USER_ADMIN_EMAIL = 'test_admin@mail.com' | ||||
TEST_USER_REGULAR_LOGIN = 'test_regular' | ||||
TEST_USER_REGULAR_PASS = 'test12' | ||||
TEST_USER_REGULAR_EMAIL = 'test_regular@mail.com' | ||||
TEST_USER_REGULAR2_LOGIN = 'test_regular2' | ||||
TEST_USER_REGULAR2_PASS = 'test12' | ||||
TEST_USER_REGULAR2_EMAIL = 'test_regular2@mail.com' | ||||
HG_REPO = 'vcs_test_hg' | ||||
GIT_REPO = 'vcs_test_git' | ||||
SVN_REPO = 'vcs_test_svn' | ||||
NEW_HG_REPO = 'vcs_test_hg_new' | ||||
NEW_GIT_REPO = 'vcs_test_git_new' | ||||
HG_FORK = 'vcs_test_hg_fork' | ||||
GIT_FORK = 'vcs_test_git_fork' | ||||
## VCS | ||||
SCM_TESTS = ['hg', 'git'] | ||||
uniq_suffix = str(int(time.mktime(datetime.datetime.now().timetuple()))) | ||||
TEST_GIT_REPO = jn(TESTS_TMP_PATH, GIT_REPO) | ||||
TEST_GIT_REPO_CLONE = jn(TESTS_TMP_PATH, 'vcsgitclone%s' % uniq_suffix) | ||||
TEST_GIT_REPO_PULL = jn(TESTS_TMP_PATH, 'vcsgitpull%s' % uniq_suffix) | ||||
TEST_HG_REPO = jn(TESTS_TMP_PATH, HG_REPO) | ||||
TEST_HG_REPO_CLONE = jn(TESTS_TMP_PATH, 'vcshgclone%s' % uniq_suffix) | ||||
TEST_HG_REPO_PULL = jn(TESTS_TMP_PATH, 'vcshgpull%s' % uniq_suffix) | ||||
TEST_REPO_PREFIX = 'vcs-test' | ||||
def clear_all_caches(): | ||||
from beaker.cache import cache_managers | ||||
for _cache in cache_managers.values(): | ||||
_cache.clear() | ||||
def get_new_dir(title): | ||||
""" | ||||
Returns always new directory path. | ||||
""" | ||||
from rhodecode.tests.vcs.utils import get_normalized_path | ||||
name_parts = [TEST_REPO_PREFIX] | ||||
if title: | ||||
name_parts.append(title) | ||||
hex_str = hashlib.sha1('%s %s' % (os.getpid(), time.time())).hexdigest() | ||||
name_parts.append(hex_str) | ||||
name = '-'.join(name_parts) | ||||
path = os.path.join(TEST_DIR, name) | ||||
return get_normalized_path(path) | ||||
@pytest.mark.usefixtures('app', 'index_location') | ||||
class TestController(object): | ||||
maxDiff = None | ||||
def log_user(self, username=TEST_USER_ADMIN_LOGIN, | ||||
password=TEST_USER_ADMIN_PASS): | ||||
self._logged_username = username | ||||
self._session = login_user_session(self.app, username, password) | ||||
self.csrf_token = auth.get_csrf_token(self._session) | ||||
return self._session['rhodecode_user'] | ||||
def logout_user(self): | ||||
logout_user_session(self.app, auth.get_csrf_token(self._session)) | ||||
self.csrf_token = None | ||||
self._logged_username = None | ||||
self._session = None | ||||
def _get_logged_user(self): | ||||
return User.get_by_username(self._logged_username) | ||||
def login_user_session( | ||||
app, username=TEST_USER_ADMIN_LOGIN, password=TEST_USER_ADMIN_PASS): | ||||
r1941 | ||||
r40 | response = app.post( | |||
r1941 | h.route_path('login'), | |||
r40 | {'username': username, 'password': password}) | |||
r1 | if 'invalid user name' in response.body: | |||
pytest.fail('could not login using %s %s' % (username, password)) | ||||
assert response.status == '302 Found' | ||||
response = response.follow() | ||||
r41 | assert response.status == '200 OK' | |||
r1 | ||||
r1774 | session = response.get_session_from_response() | |||
r41 | assert 'rhodecode_user' in session | |||
rc_user = session['rhodecode_user'] | ||||
assert rc_user.get('username') == username | ||||
assert rc_user.get('is_authenticated') | ||||
return session | ||||
r1 | ||||
def logout_user_session(app, csrf_token): | ||||
r1941 | app.post(h.route_path('logout'), {'csrf_token': csrf_token}, status=302) | |||
r1 | ||||
def login_user(app, username=TEST_USER_ADMIN_LOGIN, | ||||
password=TEST_USER_ADMIN_PASS): | ||||
return login_user_session(app, username, password)['rhodecode_user'] | ||||
r2095 | def assert_session_flash(response, msg=None, category=None, no_=None): | |||
r1 | """ | |||
Assert on a flash message in the current session. | ||||
r2095 | :param response: Response from give calll, it will contain flash | |||
messages or bound session with them. | ||||
:param msg: The expected message. Will be evaluated if a | ||||
r1 | :class:`LazyString` is passed in. | |||
:param category: Optional. If passed, the message category will be | ||||
checked as well. | ||||
r2095 | :param no_: Optional. If passed, the message will be checked to NOT | |||
be in the flash session | ||||
r1 | """ | |||
r1925 | if msg is None and no_ is None: | |||
raise ValueError("Parameter msg or no_ is required.") | ||||
if msg and no_: | ||||
raise ValueError("Please specify either msg or no_, but not both") | ||||
r1 | ||||
r2095 | session = response.get_session_from_response() | |||
messages = flash.pop_messages(session=session) | ||||
r1097 | msg = _eval_if_lazy(msg) | |||
r2349 | if no_: | |||
error_msg = 'unable to detect no_ message `%s` in empty flash list' % no_ | ||||
else: | ||||
error_msg = 'unable to find message `%s` in empty flash list' % msg | ||||
assert messages, error_msg | ||||
r1 | message = messages[0] | |||
r1925 | message_text = _eval_if_lazy(message.message) or '' | |||
r1 | ||||
r1925 | if no_: | |||
if no_ in message_text: | ||||
msg = u'msg `%s` found in session flash.' % (no_,) | ||||
pytest.fail(safe_str(msg)) | ||||
else: | ||||
if msg not in message_text: | ||||
fail_msg = u'msg `%s` not found in session ' \ | ||||
u'flash: got `%s` (type:%s) instead' % ( | ||||
msg, message_text, type(message_text)) | ||||
r1312 | ||||
r1925 | pytest.fail(safe_str(fail_msg)) | |||
if category: | ||||
assert category == message.category | ||||
r1 | ||||
def _eval_if_lazy(value): | ||||
return value.eval() if hasattr(value, 'eval') else value | ||||
r1720 | def no_newline_id_generator(test_name): | |||
""" | ||||
Generates a test name without spaces or newlines characters. Used for | ||||
nicer output of progress of test | ||||
""" | ||||
org_name = test_name | ||||
test_name = test_name\ | ||||
.replace('\n', '_N') \ | ||||
r1930 | .replace('\r', '_N') \ | |||
r1720 | .replace('\t', '_T') \ | |||
.replace(' ', '_S') | ||||
return test_name or 'test-with-empty-name' | ||||