def load_config(cfg_file='globus-config.yml', logger=logging.log):
config = None
config_dir = get_config_dirs()[0]
if os.path.isfile(os.path.join(config_dir, cfg_file)):
config = os.path.join(config_dir, cfg_file)
else:
logger(logging.ERROR, 'Could not find globus config file')
raise Exception
return yaml.safe_load(open(config).read())
def load_test_conf_file(file_name):
config_dir = next(
filter(lambda d: os.path.exists(os.path.join(d, file_name)),
get_config_dirs()))
with open(os.path.join(config_dir, file_name)) as f:
return json.load(f)
import json
import os
import tempfile
from random import choice
from string import ascii_uppercase
import pytest
from six import PY3
from rucio.common.config import config_get, config_get_bool, get_config_dirs
from rucio.common.utils import generate_uuid as uuid, execute
skip_rse_tests_with_accounts = pytest.mark.skipif(
not any(
os.path.exists(os.path.join(d, 'rse-accounts.cfg'))
for d in get_config_dirs()),
reason='fails if no rse-accounts.cfg found')
skiplimitedsql = pytest.mark.skipif(
'RDBMS' in os.environ and os.environ['RDBMS'] == 'sqlite',
reason="does not work in SQLite because of missing features")
skip_multivo = pytest.mark.skipif('SUITE' in os.environ
and os.environ['SUITE'] == 'multi_vo',
reason="does not work for multiVO")
def get_long_vo():
""" Get the VO name from the config file for testing.
Don't map the name to a short version.
:returns: VO name string.
"""
vo_name = 'def'
import html.parser as HTMLParser
import datetime
import glob
import hashlib
import logging
import operator
import os
import re
import requests
import gfal2
CHUNK_SIZE = 10485760
__DUMPERCONFIGDIRS = (os.path.join(confdir, 'auditor')
for confdir in get_config_dirs())
__DUMPERCONFIGDIRS = list(filter(os.path.exists, __DUMPERCONFIGDIRS))
class Parser(ConfigParser.RawConfigParser, object):
'''
RawConfigParser subclass that doesn't modify the the name of the options
and removes any quotes arround the string values.
'''
remove_quotes_re = re.compile(r"^'(.+)'$")
remove_double_quotes_re = re.compile(r'^"(.+)"$')
def optionxform(self, optionstr):
return optionstr
def get(self, section, option):
type = 'USER'
try:
client.get_account(account)
print 'Account \'' + account + '\' is already registered as Rucio account'
except exception.AccountNotFound:
client.add_account(account, type, None)
pass
except exception.InvalidObject as e:
print e[0][0]
pass
# Get LDAP Config
config = ConfigParser.ConfigParser()
has_config = False
for configfile in (os.path.join(confdir, 'ldap.cfg') for confdir in get_config_dirs()):
has_config = config.read(configfile) == [configfile]
if has_config:
break
# Global Variables
client = Client()
server_url = config.get('ldap', 'ldap_host')
baseDN = config.get('ldap', 'baseDN')
searchScope = ldapurl.LDAP_SCOPE_SUBTREE
retrieveAttributes = ['uid', 'gecos', 'mail']
if config.get('attributes', 'account') != 'uid':
retrieveAttributes.append('account')
if config.get('attributes', 'auth_type') != 'DEFAULT':
retrieveAttributes.append('auth_type')
searchFilter = config.get('ldap', 'searchFilter')
import logging
import operator
import os
import re
import requests
try:
import gfal2
except ImportError as e:
import sys
if 'nose' not in sys.modules and 'py.test' not in sys.modules and 'sphinx' not in sys.modules:
raise e
CHUNK_SIZE = 10485760
__DUMPERCONFIGDIRS = (os.path.join(confdir, 'auditor') for confdir in get_config_dirs())
__DUMPERCONFIGDIRS = list(filter(os.path.exists, __DUMPERCONFIGDIRS))
class Parser(ConfigParser.RawConfigParser, object):
'''
RawConfigParser subclass that doesn't modify the the name of the options
and removes any quotes arround the string values.
'''
remove_quotes_re = re.compile(r"^'(.+)'$")
remove_double_quotes_re = re.compile(r'^"(.+)"$')
def optionxform(self, optionstr):
return optionstr
def get(self, section, option):
