vcsserver/tests/fixture.py

# RhodeCode VCSServer provides access to different vcs backends via network.
# Copyright (C) 2014-2018 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)