ceph/teuthology/task/watch_notify_stress.py
2012-01-03 13:02:31 -08:00

66 lines
1.7 KiB
Python

import contextlib
import logging
import proc_thrasher
from ..orchestra import run
log = logging.getLogger(__name__)
@contextlib.contextmanager
def task(ctx, config):
"""
Run test_stress_watch
The config should be as follows:
test_stress_watch:
clients: [client list]
example:
tasks:
- ceph:
- test_stress_watch:
clients: [client.0]
- interactive:
"""
log.info('Beginning test_stress_watch...')
assert isinstance(config, dict), \
"please list clients to run on"
testwatch = {}
remotes = []
for role in config.get('clients', ['client.0']):
assert isinstance(role, basestring)
PREFIX = 'client.'
assert role.startswith(PREFIX)
id_ = role[len(PREFIX):]
(remote,) = ctx.cluster.only(role).remotes.iterkeys()
remotes.append(remote)
args =['CEPH_CLIENT_ID={id_}'.format(id_=id_),
'CEPH_CONF=/tmp/cephtest/ceph.conf',
'CEPH_ARGS="{flags}"'.format(flags=config.get('flags', '')),
'LD_PRELOAD=/tmp/cephtest/binary/usr/local/lib/librados.so.2',
'/tmp/cephtest/daemon-helper', 'kill',
'/tmp/cephtest/binary/usr/local/bin/multi_stress_watch foo foo'
]
log.info("args are %s" % (args,))
proc = proc_thrasher.ProcThrasher({}, remote,
args=[run.Raw(i) for i in args],
logger=log.getChild('testwatch.{id}'.format(id=id_)),
stdin=run.PIPE,
wait=False
)
proc.start()
testwatch[id_] = proc
try:
yield
finally:
log.info('joining watch_notify_stress')
for i in testwatch.itervalues():
i.join()