mirror of
https://github.com/ceph/ceph
synced 2024-12-12 14:39:05 +00:00
72 lines
1.8 KiB
Python
72 lines
1.8 KiB
Python
|
import argparse
|
||
|
|
import yaml
|
||
|
|
|
||
|
|
def config_file(string):
|
||
|
|
config = {}
|
||
|
|
try:
|
||
|
|
with file(string) as f:
|
||
|
|
g = yaml.safe_load_all(f)
|
||
|
|
for new in g:
|
||
|
|
config.update(new)
|
||
|
|
except IOError, e:
|
||
|
|
raise argparse.ArgumentTypeError(str(e))
|
||
|
|
return config
|
||
|
|
|
||
|
|
class MergeConfig(argparse.Action):
|
||
|
|
def __call__(self, parser, namespace, values, option_string=None):
|
||
|
|
config = getattr(namespace, self.dest)
|
||
|
|
for new in values:
|
||
|
|
config.update(new)
|
||
|
|
|
||
|
|
def parse_args():
|
||
|
|
parser = argparse.ArgumentParser(description='Run ceph integration tests')
|
||
|
|
parser.add_argument(
|
||
|
|
'-v', '--verbose',
|
||
|
|
action='store_true', default=None,
|
||
|
|
help='be more verbose',
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
'config',
|
||
|
|
metavar='CONFFILE',
|
||
|
|
nargs='+',
|
||
|
|
type=config_file,
|
||
|
|
action=MergeConfig,
|
||
|
|
default={},
|
||
|
|
help='config file to read',
|
||
|
|
)
|
||
|
|
|
||
|
|
args = parser.parse_args()
|
||
|
|
return args
|
||
|
|
|
||
|
|
def main():
|
||
|
|
from gevent import monkey; monkey.patch_all()
|
||
|
|
from orchestra import monkey; monkey.patch_all()
|
||
|
|
|
||
|
|
import logging
|
||
|
|
|
||
|
|
log = logging.getLogger(__name__)
|
||
|
|
ctx = parse_args()
|
||
|
|
|
||
|
|
loglevel = logging.INFO
|
||
|
|
if ctx.verbose:
|
||
|
|
loglevel = logging.DEBUG
|
||
|
|
|
||
|
|
logging.basicConfig(
|
||
|
|
level=loglevel,
|
||
|
|
)
|
||
|
|
|
||
|
|
log.debug('\n '.join(['Config:', ] + yaml.safe_dump(ctx.config).splitlines()))
|
||
|
|
log.info('Opening connections...')
|
||
|
|
|
||
|
|
from orchestra import connection, remote
|
||
|
|
import orchestra.cluster
|
||
|
|
|
||
|
|
remotes = [remote.Remote(name=t, ssh=connection.connect(t))
|
||
|
|
for t in ctx.config['targets']]
|
||
|
|
ctx.cluster = orchestra.cluster.Cluster()
|
||
|
|
for rem, roles in zip(remotes, ctx.config['roles']):
|
||
|
|
ctx.cluster.add(rem, roles)
|
||
|
|
|
||
|
|
from teuthology.run_tasks import run_tasks
|
||
|
|
run_tasks(tasks=ctx.config['tasks'], ctx=ctx)
|