rhodecode/tests/server_utils.py

# Copyright (C) 2010-2024 RhodeCode GmbH
#
# 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 tempfile
import pytest
import subprocess
import logging
import requests
import configparser


from rhodecode.tests import TEST_USER_ADMIN_LOGIN, TEST_USER_ADMIN_PASS
from rhodecode.tests.utils import is_url_reachable
from rhodecode.tests import console_printer

log = logging.getLogger(__name__)


def get_port(pyramid_config):
    config = configparser.ConfigParser()
    config.read(pyramid_config)
    return config.get('server:main', 'port')


def get_host_url(pyramid_config):
    """Construct the host url using the port in the test configuration."""
    port = get_port(pyramid_config)
    return f'127.0.0.1:{port}'


def assert_no_running_instance(url):
    if is_url_reachable(url):
        console_printer(f"Hint: Usually this means another instance of server "
              f"is running in the background at {url}.")
        pytest.fail(f"Port is not free at {url}, cannot start server at")


class ServerBase(object):
    _args = []
    log_file_name = 'NOT_DEFINED.log'
    status_url_tmpl = 'http://{host}:{port}/_admin/ops/ping'
    console_marker = " :warning: [green]pytest-setup[/green] "

    def __init__(self, config_file, log_file, env):
        self.config_file = config_file
        config = configparser.ConfigParser()
        config.read(config_file)

        self._config = {k: v for k, v in config['server:main'].items()}

        self._args = []
        self.log_file = log_file or os.path.join(
            tempfile.gettempdir(), self.log_file_name)
        self.env = env
        self.process = None
        self.server_out = None
        log.info(f"Using the {self.__class__.__name__} configuration:{config_file}")

        if not os.path.isfile(config_file):
            raise RuntimeError(f'Failed to get config at {config_file}')

    @property
    def command(self):
        return ' '.join(self._args)

    @property
    def bind_addr(self):
        return '{host}:{port}'.format(**self._config)

    @property
    def http_url(self):
        template = 'http://{host}:{port}/'
        return template.format(**self._config)

    def host_url(self):
        host = get_host_url(self.config_file)
        return f'http://{host}'

    def get_rc_log(self):
        with open(self.log_file) as f:
            return f.read()

    def assert_message_in_server_logs(self, message):
        server_logs = self.get_rc_log()
        assert message in server_logs

    def wait_until_ready(self, timeout=30):
        host = self._config['host']
        port = self._config['port']
        status_url = self.status_url_tmpl.format(host=host, port=port)
        start = time.time()

        while time.time() - start < timeout:
            try:
                requests.get(status_url)
                break
            except requests.exceptions.ConnectionError:
                time.sleep(0.2)
        else:
            pytest.fail(
                f"Starting the {self.__class__.__name__} failed or took more than {timeout} seconds."
                f"cmd: `{self.command}`"
            )

        log.info(f'Server of {self.__class__.__name__} ready at url {status_url}')

    def shutdown(self):
        self.process.kill()
        self.server_out.flush()
        self.server_out.close()

    def get_log_file_with_port(self):
        log_file = list(self.log_file.partition('.log'))
        log_file.insert(1, f'-{get_port(self.config_file)}')
        log_file = ''.join(log_file)
        return log_file


class RcVCSServer(ServerBase):
    """
    Represents a running VCSServer instance.
    """

    log_file_name = 'rhodecode-vcsserver.log'
    status_url_tmpl = 'http://{host}:{port}/status'

    def __init__(self, config_file, log_file=None, workers='3', env=None, info_prefix=''):
        super(RcVCSServer, self).__init__(config_file, log_file, env)
        self.info_prefix = info_prefix
        self._args = [
            'gunicorn',
            '--bind', self.bind_addr,
            '--worker-class', 'sync',
            '--threads', '1',
            '--backlog', '8',
            '--timeout', '300',
            '--workers', workers,
            '--paste', self.config_file]

    def start(self):
        env = os.environ.copy()

        self.log_file = self.get_log_file_with_port()
        self.server_out = open(self.log_file, 'w')

        host_url = self.host_url()
        assert_no_running_instance(host_url)

        console_printer(f'{self.console_marker}{self.info_prefix}rhodecode-vcsserver starting at: {host_url}')
        console_printer(f'{self.console_marker}{self.info_prefix}rhodecode-vcsserver command: {self.command}')
        console_printer(f'{self.console_marker}{self.info_prefix}rhodecode-vcsserver logfile: {self.log_file}')
        console_printer()

        self.process = subprocess.Popen(
            self._args, bufsize=0, env=env,
            stdout=self.server_out, stderr=self.server_out)


class RcWebServer(ServerBase):
    """
    Represents a running RCE web server used as a test fixture.
    """

    log_file_name = 'rhodecode-ce.log'
    status_url_tmpl = 'http://{host}:{port}/_admin/ops/ping'

    def __init__(self, config_file, log_file=None, workers='2', env=None, info_prefix=''):
        super(RcWebServer, self).__init__(config_file, log_file, env)
        self.info_prefix = info_prefix
        self._args = [
            'gunicorn',
            '--bind', self.bind_addr,
            '--worker-class', 'gthread',
            '--threads', '4',
            '--backlog', '8',
            '--timeout', '300',
            '--workers', workers,
            '--paste', self.config_file]

    def start(self):
        env = os.environ.copy()
        if self.env:
            env.update(self.env)

        self.log_file = self.get_log_file_with_port()
        self.server_out = open(self.log_file, 'w')

        host_url = self.host_url()
        assert_no_running_instance(host_url)

        console_printer(f'{self.console_marker}{self.info_prefix}rhodecode-ce starting at: {host_url}')
        console_printer(f'{self.console_marker}{self.info_prefix}rhodecode-ce command: {self.command}')
        console_printer(f'{self.console_marker}{self.info_prefix}rhodecode-ce logfile: {self.log_file}')
        console_printer()

        self.process = subprocess.Popen(
            self._args, bufsize=0, env=env,
            stdout=self.server_out, stderr=self.server_out)

    def repo_clone_url(self, repo_name, **kwargs):
        params = {
            'user': TEST_USER_ADMIN_LOGIN,
            'passwd': TEST_USER_ADMIN_PASS,
            'host': get_host_url(self.config_file),
            'cloned_repo': repo_name,
        }
        params.update(**kwargs)
        _url = f"http://{params['user']}:{params['passwd']}@{params['host']}/{params['cloned_repo']}"
        return _url

    def repo_clone_credentials(self, **kwargs):
        params = {
            'user': TEST_USER_ADMIN_LOGIN,
            'passwd': TEST_USER_ADMIN_PASS,
        }
        params.update(**kwargs)
        return params['user'], params['passwd']

class CeleryServer(ServerBase):
    log_file_name = 'rhodecode-celery.log'
    status_url_tmpl = 'http://{host}:{port}/_admin/ops/ping'

    def __init__(self, config_file, log_file=None, workers='2', env=None, info_prefix=''):
        super(CeleryServer, self).__init__(config_file, log_file, env)
        self.info_prefix = info_prefix
        self._args = \
            ['celery',
             '--no-color',
             '--app=rhodecode.lib.celerylib.loader',
             'worker',
             '--autoscale=4,2',
             '--max-tasks-per-child=30',
             '--task-events',
             '--loglevel=DEBUG',
             '--ini=' + self.config_file]

    def start(self):
        env = os.environ.copy()
        env['RC_NO_TEST_ENV'] = '1'

        self.log_file = self.get_log_file_with_port()
        self.server_out = open(self.log_file, 'w')

        host_url = "Celery" #self.host_url()
        #assert_no_running_instance(host_url)

        console_printer(f'{self.console_marker}{self.info_prefix}rhodecode-celery starting at: {host_url}')
        console_printer(f'{self.console_marker}{self.info_prefix}rhodecode-celery command: {self.command}')
        console_printer(f'{self.console_marker}{self.info_prefix}rhodecode-celery logfile: {self.log_file}')
        console_printer()

        self.process = subprocess.Popen(
            self._args, bufsize=0, env=env,
            stdout=self.server_out, stderr=self.server_out)


    def wait_until_ready(self, timeout=30):
        time.sleep(2)