##// END OF EJS Templates
release: Finish preparation for 4.25.2
release: Finish preparation for 4.25.2

File last commit:

r850:cbc05af2 default
r942:8610c4bf tip stable
Show More
fixture.py
86 lines | 2.9 KiB | text/x-python | PythonLexer
# RhodeCode VCSServer provides access to different vcs backends via network.
# Copyright (C) 2014-2020 RhodeCode GmbH
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 3 of the License, or
# (at your option) any later version.
#
# 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 General Public License
# along with this program; if not, write to the Free Software Foundation,
# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
import os
import shutil
import tempfile
import configobj
class ContextINI(object):
"""
Allows to create a new test.ini file as a copy of existing one with edited
data. If existing file is not present, it creates a new one. Example usage::
with TestINI('test.ini', [{'section': {'key': 'val'}}]) as new_test_ini_path:
print 'vcsserver --config=%s' % new_test_ini
"""
def __init__(self, ini_file_path, ini_params, new_file_prefix=None,
destroy=True):
self.ini_file_path = ini_file_path
self.ini_params = ini_params
self.new_path = None
self.new_path_prefix = new_file_prefix or 'test'
self.destroy = destroy
def __enter__(self):
_, pref = tempfile.mkstemp()
loc = tempfile.gettempdir()
self.new_path = os.path.join(loc, '{}_{}_{}'.format(
pref, self.new_path_prefix, self.ini_file_path))
# copy ini file and modify according to the params, if we re-use a file
if os.path.isfile(self.ini_file_path):
shutil.copy(self.ini_file_path, self.new_path)
else:
# create new dump file for configObj to write to.
with open(self.new_path, 'wb'):
pass
config = configobj.ConfigObj(
self.new_path, file_error=True, write_empty_values=True)
for data in self.ini_params:
section, ini_params = data.items()[0]
key, val = ini_params.items()[0]
if section not in config:
config[section] = {}
config[section][key] = val
config.write()
return self.new_path
def __exit__(self, exc_type, exc_val, exc_tb):
if self.destroy:
os.remove(self.new_path)
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 = str(test_name)\
.replace('\n', '_N') \
.replace('\r', '_N') \
.replace('\t', '_T') \
.replace(' ', '_S')
return test_name or 'test-with-empty-name'