def test(target=None, t=None, cluster='.*', c=None, fabrun='.*', f=None,
bootstrap=True, b=None, fablib=None, l=None):
target = t if t is not None else target
cluster = c if c is not None else cluster
re_cluster = re.compile(cluster)
fabrun = f if f is not None else fabrun
bootstrap = (not b == 'false') if b is not None else bootstrap
fablib = l if l is not None else fablib
sys.path.remove(CONF._repo_dir)
FABTEST_DIR = os.path.dirname(os.path.abspath(__file__))
if fablib is not None:
fablib_dir = os.path.join(CONF._fablib_module_dir, fablib)
CONF._repo_dir = os.path.join(fablib_dir, 'test-repo')
else:
CONF._repo_dir = os.path.join(FABTEST_DIR, 'test-repo')
EX_CONF = cfg.ConfigOpts()
conf_file = os.path.join(CONF._repo_dir, 'fabfile.ini')
if os.path.exists(conf_file):
EX_CONF([], default_config_files=[conf_file])
else:
EX_CONF([])
from fabkit.conf import conf_base, conf_fabric, conf_web, conf_test # noqa
EX_CONF.register_opts(conf_base.default_opts)
EX_CONF.register_opts(conf_test.test_opts, group="test")
CONF._storage_dir = os.path.join(CONF._repo_dir, 'storage')
CONF._databag_dir = os.path.join(CONF._repo_dir, 'databag')
CONF._tmp_dir = os.path.join(CONF._storage_dir, 'tmp')
CONF._log_dir = os.path.join(CONF._storage_dir, 'log')
CONF._node_dir = os.path.join(CONF._repo_dir, 'nodes')
CONF._node_meta_pickle = os.path.join(CONF._node_dir, 'meta.pickle')
CONF._fabscript_module_dir = os.path.join(CONF._repo_dir, 'fabscript')
CONF._fablib_module_dir = os.path.join(CONF._repo_dir, 'fablib')
sys.path.extend([
CONF._repo_dir,
])
CONF.fablib = EX_CONF.fablib
CONF.test.fablib = EX_CONF.test.fablib
CONF.test.clusters = EX_CONF.test.clusters
util.create_required_dirs()
util.git_clone_required_fablib(is_test=True)
CONF._unittests_dir = os.path.join(FABTEST_DIR, 'unittests')
if target is None or target == 'fab':
if bootstrap:
node('bootstrap/')
setup()
CONF.user = CONF.test.user
CONF.password = CONF.test.password
conf_fabric.init()
env.disable_known_hosts = True
env.tasks.append('node:{0}'.format(cluster))
env.tasks.append('setup')
for cluster in CONF.test.clusters:
if re_cluster.search(cluster):
node(cluster, 'yes')
setup(f=fabrun)
if fablib is None:
if target is None:
suites = unittest.TestLoader().discover(CONF._unittests_dir,
pattern='test_*')
else:
suites = unittest.TestLoader().discover(CONF._unittests_dir,
pattern='test_{0}*'.format(target))
alltests = unittest.TestSuite(suites)
result = unittest.TextTestRunner(verbosity=2).run(alltests)
exit(len(result.errors) + len(result.failures))
def test(target=None, t=None, cluster='.*', c=None, fabrun='.*', f=None,
pipeline_pattern='.*', p='.*', fablib=None, l=None):
target = t if t is not None else target
cluster = c if c is not None else cluster
re_cluster = re.compile(cluster)
fabrun = f if f is not None else fabrun
fablib = l if l is not None else fablib
pipeline_pattern = p if p is not None else pipeline_pattern
if target is None and fablib is None:
print 'Bad args.'
print 'Prease set test target (t=xxx) or fablib target (l=yyy).'
print test.__doc__
return
sys.path.remove(CONF._repo_dir)
FABTEST_DIR = os.path.dirname(os.path.abspath(__file__))
if fablib is not None:
fablib_dir = os.path.join(CONF._fablib_module_dir, fablib)
CONF._repo_dir = os.path.join(fablib_dir, 'test-repo')
else:
CONF._repo_dir = os.path.join(FABTEST_DIR, 'test-repo')
EX_CONF = cfg.ConfigOpts()
conf_file = os.path.join(CONF._repo_dir, 'fabfile.ini')
if os.path.exists(conf_file):
EX_CONF([], default_config_files=[conf_file])
else:
EX_CONF([])
from fabkit.conf import conf_base, conf_fabric, conf_web, conf_test # noqa
EX_CONF.register_opts(conf_base.default_opts)
EX_CONF.register_opts(conf_test.test_opts, group="test")
CONF._storage_dir = os.path.join(CONF._repo_dir, 'storage')
CONF._databag_dir = os.path.join(CONF._repo_dir, 'databag')
CONF._tmp_dir = os.path.join(CONF._storage_dir, 'tmp')
CONF._log_dir = os.path.join(CONF._storage_dir, 'log')
CONF._node_dir = os.path.join(CONF._repo_dir, 'nodes')
CONF._node_meta_pickle = os.path.join(CONF._node_dir, 'meta.pickle')
CONF._fabscript_module_dir = os.path.join(CONF._repo_dir, 'fabscript')
CONF._fablib_module_dir = os.path.join(CONF._repo_dir, 'fablib')
sys.path.extend([
CONF._repo_dir,
])
CONF.fablib = EX_CONF.fablib
CONF.test.fablib = EX_CONF.test.fablib
CONF.test.clusters = EX_CONF.test.clusters
util.create_required_dirs()
util.git_clone_required_fablib(is_test=True)
CONF._unittests_dir = os.path.join(FABTEST_DIR, 'unittests')
if target is None or target == 'fab':
CONF.user = CONF.test.user
CONF.password = CONF.test.password
conf_fabric.init()
env.forward_agent = False
env.disable_known_hosts = True
env.tasks.append('node:{0}'.format(cluster))
env.tasks.append("job:local,'{0}'".format(pipeline_pattern))
for cluster in CONF.test.clusters:
if re_cluster.search(cluster):
node(cluster, 'yes')
job('local', pipeline_pattern, f=fabrun)
if fablib is None:
# Test fabkit
if target == 'all':
suites = unittest.TestLoader().discover(CONF._unittests_dir,
pattern='test_*')
else:
suites = unittest.TestLoader().discover(CONF._unittests_dir,
pattern='test_{0}*'.format(target))
alltests = unittest.TestSuite(suites)
result = unittest.TextTestRunner(verbosity=2).run(alltests)
exit(len(result.errors) + len(result.failures))
# initialize config
from fabkit import util, env
from fabkit.conf import conf_base, conf_fabric, conf_web, conf_test # noqa
log_file = None
tasks = env.tasks
if len(tasks) > 0:
if tasks[0] == 'agent':
log_file = 'agent.log'
elif tasks[0] == 'agent_central':
log_file = 'agent-central.log'
conf_base.init(REPO_DIR, log_file)
conf_fabric.init()
util.create_required_dirs()
util.event_handler.init()
# util.git_clone_required_fablib()
# register fabric tasks
from node import node, nodestr # noqa
from setup import setup, manage, check # noqa
from job import job # noqa
from databag import databag # noqa
from runserver import runserver # noqa
from agent import agent, agent_central, agent_manager # noqa
from fabtest import test # noqa
from doc import doc # noqa
from util import genconfig, sync_fablib, client, sync_db, create_dns_domain, create_dns_record # noqa
# initialize config
from fabkit import util, env
from fabkit.conf import conf_base, conf_fabric, conf_web, conf_test # noqa
log_file = None
tasks = env.tasks
if len(tasks) > 0:
if tasks[0] == 'agent':
log_file = 'agent.log'
elif tasks[0] == 'agent_central':
log_file = 'agent-central.log'
conf_base.init(REPO_DIR, log_file)
conf_fabric.init()
util.create_required_dirs()
util.event_handler.init()
# util.git_clone_required_fablib()
# register fabric tasks
from node import node, nodestr # noqa
from setup import setup, manage, check # noqa
from job import job # noqa
from databag import databag # noqa
from runserver import runserver # noqa
from agent import agent, agent_central, agent_manager # noqa
from fabtest import test # noqa
from doc import doc # noqa
from util import genconfig, sync_fablib, client, sync_db, create_dns_domain, create_dns_record # noqa
def test(target=None,
t=None,
cluster='.*',
c=None,
fabrun='.*',
f=None,
pipeline_pattern='.*',
p='.*',
fablib=None,
l=None):
target = t if t is not None else target
cluster = c if c is not None else cluster
re_cluster = re.compile(cluster)
fabrun = f if f is not None else fabrun
fablib = l if l is not None else fablib
pipeline_pattern = p if p is not None else pipeline_pattern
if target is None and fablib is None:
print 'Bad args.'
print 'Prease set test target (t=xxx) or fablib target (l=yyy).'
print test.__doc__
return
sys.path.remove(CONF._repo_dir)
FABTEST_DIR = os.path.dirname(os.path.abspath(__file__))
if fablib is not None:
fablib_dir = os.path.join(CONF._fablib_module_dir, fablib)
CONF._repo_dir = os.path.join(fablib_dir, 'test-repo')
else:
CONF._repo_dir = os.path.join(FABTEST_DIR, 'test-repo')
EX_CONF = cfg.ConfigOpts()
conf_file = os.path.join(CONF._repo_dir, 'fabfile.ini')
if os.path.exists(conf_file):
EX_CONF([], default_config_files=[conf_file])
else:
EX_CONF([])
from fabkit.conf import conf_base, conf_fabric, conf_web, conf_test # noqa
EX_CONF.register_opts(conf_base.default_opts)
EX_CONF.register_opts(conf_test.test_opts, group="test")
CONF._storage_dir = os.path.join(CONF._repo_dir, 'storage')
CONF._databag_dir = os.path.join(CONF._repo_dir, 'databag')
CONF._tmp_dir = os.path.join(CONF._storage_dir, 'tmp')
CONF._log_dir = os.path.join(CONF._storage_dir, 'log')
CONF._node_dir = os.path.join(CONF._repo_dir, 'nodes')
CONF._node_meta_pickle = os.path.join(CONF._node_dir, 'meta.pickle')
CONF._fabscript_module_dir = os.path.join(CONF._repo_dir, 'fabscript')
CONF._fablib_module_dir = os.path.join(CONF._repo_dir, 'fablib')
sys.path.extend([
CONF._repo_dir,
])
CONF.fablib = EX_CONF.fablib
CONF.test.fablib = EX_CONF.test.fablib
CONF.test.clusters = EX_CONF.test.clusters
util.create_required_dirs()
util.git_clone_required_fablib(is_test=True)
CONF._unittests_dir = os.path.join(FABTEST_DIR, 'unittests')
if target is None or target == 'fab':
CONF.user = CONF.test.user
CONF.password = CONF.test.password
conf_fabric.init()
env.forward_agent = False
env.disable_known_hosts = True
env.tasks.append('node:{0}'.format(cluster))
env.tasks.append("job:local,'{0}'".format(pipeline_pattern))
for cluster in CONF.test.clusters:
if re_cluster.search(cluster):
node(cluster, 'yes')
job('local', pipeline_pattern, f=fabrun)
if fablib is None:
# Test fabkit
if target == 'all':
suites = unittest.TestLoader().discover(CONF._unittests_dir,
pattern='test_*')
else:
suites = unittest.TestLoader().discover(
CONF._unittests_dir, pattern='test_{0}*'.format(target))
alltests = unittest.TestSuite(suites)
result = unittest.TextTestRunner(verbosity=2).run(alltests)
exit(len(result.errors) + len(result.failures))
