Python init_db Examples

Example #1

File: test_preprocessor.py Project: vishnuravichand/wazuh

def db_setup():
    with patch('wazuh.common.ossec_uid'), patch('wazuh.common.ossec_gid'):
        with patch('sqlalchemy.create_engine', return_value=create_engine("sqlite://")):
            with patch('shutil.chown'), patch('os.chmod'):
                with patch('api.constants.SECURITY_PATH', new=test_data_path):
                    from wazuh.rbac.preprocessor import optimize_resources
    init_db('schema_security_test.sql', test_data_path)

    yield optimize_resources

Example #2

def db_setup():
    with patch('wazuh.common.ossec_uid'), patch('wazuh.common.ossec_gid'):
        with patch('sqlalchemy.create_engine', return_value=create_engine("sqlite://")):
            with patch('shutil.chown'), patch('os.chmod'):
                with patch('api.constants.SECURITY_PATH', new=test_data_path):
                    from wazuh.rbac.auth_context import RBAChecker
    init_db('schema_security_test.sql', test_data_path)

    yield RBAChecker

Example #3

File: test_orm.py Project: wisdark/wazuh

def db_setup():
    with patch('wazuh.core.common.wazuh_uid'), patch(
            'wazuh.core.common.wazuh_gid'):
        with patch('sqlalchemy.create_engine',
                   return_value=create_engine("sqlite://")):
            with patch('shutil.chown'), patch('os.chmod'):
                with patch('api.constants.SECURITY_PATH', new=test_data_path):
                    import wazuh.rbac.orm as rbac
    init_db('schema_security_test.sql', test_data_path)

    yield rbac