mongo.py

"""
Plugin that imports the required fixtures to start the database
and adjusts the mongo "dbconnection" static config value.
"""
 
import re
 
import pytest
 
pytest_plugins = [
    'testsuite.databases.mongo.pytest_plugin',
    'pytest_userver.plugins.core',
]
 
 
USERVER_CONFIG_HOOKS = ['userver_mongo_config']
 
 
@pytest.fixture(scope='session')
def userver_mongo_config(mongo_connection_info):
    """
    Returns a function that adjusts the static configuration file for
    the testsuite.
    Sets the `dbconnection` to the testsuite started MongoDB credentials if
    the `dbconnection` starts with `mongodb://`. Additionally
    increases MongoDB connection timeouts to 30 seconds.
 
    @ingroup userver_testsuite_fixtures
    """
 
    port = mongo_connection_info.port
    host = mongo_connection_info.host
    new_uri = f'mongodb://{host}:{port}/'
 
    def _patch_config(config_yaml, config_vars):
        components = config_yaml['components_manager']['components']
        for _, params in components.items():
            uri = ''
 
            if params and 'dbconnection#fallback' in params:
                uri = params['dbconnection#fallback']
 
            if params and 'dbconnection' in params:
                uri = params['dbconnection']
 
            if not uri or not uri.startswith('mongodb://'):
                continue
 
            uri = re.sub('mongodb://[^:]+:\\d+/', new_uri, uri)
            uri = re.sub('mongodb://[^:]+/', new_uri, uri)
 
            params['dbconnection'] = uri
            params['conn_timeout'] = '30s'
            params['so_timeout'] = '30s'
            params.pop('dbalias', None)
 
    return _patch_config