def test_set_cfg():
lpc = LinchpinCliContext()
lpc.load_config(config_path)
lpc.set_cfg('test', 'key', 'value')
assert_equal(lpc.get_cfg('test', 'key'), lpc.cfgs['test']['key'])
def setup_lp_api():
"""
Perform setup of LinchpinContext, lpc.load_config, and LinchPinAPI
"""
global lpc
global lpa
global target
global pinfile
base_path = '{0}'.format(os.path.dirname(
os.path.realpath(__file__))).rstrip('/')
lib_path = os.path.realpath(os.path.join(base_path, os.pardir))
setup_load_config()
pinfile = 'PinFile'
mock_path = '{0}/{1}/{2}'.format(lib_path, 'mockdata', provider)
lpc = LinchpinCliContext()
lpc.load_config(lpconfig=config_path)
lpc.load_global_evars()
lpc.setup_logging()
lpc.workspace = os.path.realpath(mock_path)
lpa = LinchpinAPI(lpc)
def setup_lp_cli():
"""
Perform setup of LinchpinContext, lpc.load_config, and LinchPinAPI
"""
global lpctx
global lpc
global target
global pinfile
base_path = '{0}'.format(os.path.dirname(
os.path.realpath(__file__))).rstrip('/')
lib_path = os.path.realpath(os.path.join(base_path, os.pardir))
setup_load_config()
pinfile = 'PinFile'
mock_path = '{0}/{1}/{2}'.format(lib_path, 'mockdata', provider)
lpctx = LinchpinCliContext()
lpctx.load_config(lpconfig=config_path)
lpctx.load_global_evars()
lpctx.setup_logging()
lpctx.workspace = os.path.realpath(mock_path)
lpc = LinchpinCli(lpctx)
def test_load_config():
lpc = LinchpinCliContext()
lpc.load_config(lpconfig=config_path)
# assert_dict_equal.__self__.maxDiff = None
assert_dict_equal(config_data, lpc.cfgs)
def test_get_cfg():
lpc = LinchpinCliContext()
lpc.load_config()
cfg_value = lpc.get_cfg(section='lp', key='pkg')
assert_equal(cfg_value, config_data['lp']['pkg'])
def test_load_global_evars():
lpc = LinchpinCliContext()
lpc.load_config(config_path)
lpc.load_global_evars()
assert_dict_equal(evars_data, lpc.evars)
def test_get_cfg():
lpc = LinchpinCliContext()
lpc.load_config(config_path)
cfg_value = lpc.get_cfg('hookstates', 'up')
assert_equal(cfg_value, config_data['hookstates']['up'])
def test_logging_setup():
lpc = LinchpinCliContext()
lpc.load_config(config_path)
lpc.setup_logging()
assert os.path.isfile(logfile)
def test_get_cfg():
lpc = LinchpinCliContext()
lpc.load_config()
cfg_value = lpc.get_cfg(section='lp', key='pkg')
assert_equal(cfg_value, config_data['lp']['pkg'])
def test_set_cfg():
lpc = LinchpinCliContext()
lpc.load_config(config_path)
lpc.set_cfg('test', 'key', 'value')
assert_equal(lpc.get_cfg('test', 'key'), lpc.cfgs['test']['key'])
def setup_lp_cli():
"""
Perform setup of LinchpinContext, lpc.load_config, and LinchPinAPI
"""
global lpctx
global lpc
global target
global pinfile
base_path = '{0}'.format(os.path.dirname(
os.path.realpath(__file__))).rstrip('/')
lib_path = os.path.realpath(os.path.join(base_path, os.pardir))
setup_load_config()
pinfile = 'PinFile'
mock_path = '{0}/{1}/{2}'.format(lib_path, 'mockdata', provider)
lpctx = LinchpinCliContext()
lpctx.load_config(lpconfig=config_path)
lpctx.load_global_evars()
lpctx.setup_logging()
lpctx.workspace = os.path.realpath(mock_path)
lpctx.pinfile = pinfile
lpctx.pf_data = None
lpctx.no_monitor = True
lpctx.set_evar("no_monitor", True)
lpctx.set_cfg("progress_bar", "no_progress", str(True))
lpc = LinchpinCli(lpctx)
lpc.disable_pbar = True
def setup_lp_fetch_env():
global lpc
global lpcli
global target
global pinfile
global mockpath
base_path = '{0}'.format(os.path.dirname(
os.path.realpath(__file__))).rstrip('/')
lib_path = os.path.realpath(os.path.join(base_path, os.pardir))
setup_load_config()
pinfile = 'PinFile'
mock_path = '{0}/{1}/{2}'.format(lib_path, 'mockdata', provider)
lpc = LinchpinCliContext()
lpc.load_config(lpconfig=config_path)
lpc.load_global_evars()
lpc.setup_logging()
lpc.workspace = '/tmp/workspace/'
mockpath = os.path.realpath(mock_path)
if not os.path.exists(lpc.workspace):
os.mkdir(lpc.workspace)
lpcli = LinchpinCli(lpc)
def test_cli_create():
lpctx = LinchpinCliContext()
lpctx.load_config()
lpctx.load_global_evars()
lpc = LinchpinCli(lpctx)
assert_equal(isinstance(lpc, LinchpinCli), True)
def test_set_evar():
lpc = LinchpinCliContext()
lpc.load_config(lpconfig=config_path)
lpc.load_global_evars()
lpc.set_evar('test', 'me')
assert_equal('me', lpc.evars['test'])
def test_get_evar():
lpc = LinchpinCliContext()
lpc.load_config(lpconfig=config_path)
lpc.load_global_evars()
evar_value = lpc.get_evar('_async')
assert_equal(evar_value, evars_data['_async'])
def test_cli_create():
lpctx = LinchpinCliContext()
lpctx.load_config()
lpctx.load_global_evars()
lpc = LinchpinCli(lpctx)
assert_equal(isinstance(lpc, LinchpinCli), True)
def test_logging_setup():
lpc = LinchpinCliContext()
lpc.load_config(lpconfig=config_path)
lpc.set_cfg('logger', 'file', config_data['logger']['file'])
lpc.setup_logging()
assert os.path.isfile(logfile)
def test_get_evar():
lpc = LinchpinCliContext()
lpc.load_config(lpconfig=config_path)
lpc.load_global_evars()
evar_value = lpc.get_evar('_async')
assert_equal(evar_value, evars_data['_async'])
def test_set_evar():
lpc = LinchpinCliContext()
lpc.load_config(lpconfig=config_path)
lpc.load_global_evars()
lpc.set_evar('test', 'me')
assert_equal('me', lpc.evars['test'])
def test_logging_setup():
lpc = LinchpinCliContext()
lpc.load_config(lpconfig=config_path)
lpc.set_cfg('logger',
'file',
config_data['logger']['file'])
lpc.setup_logging()
assert os.path.isfile(logfile)
def test_load_config():
lpc = LinchpinCliContext()
lpc.load_config(lpconfig=config_path)
cfg_data = dict(config_data)
# remove things that will always differ
for cfg in ['lp', 'evars']:
lpc.cfgs.pop(cfg)
cfg_data.pop(cfg)
assert_dict_equal(cfg_data['console'], lpc.cfgs['console'])
def test_load_config():
lpc = LinchpinCliContext()
lpc.load_config(lpconfig=config_path)
cfg_data = dict(config_data)
# remove things that will always differ
for cfg in ['lp', 'evars']:
lpc.cfgs.pop(cfg)
cfg_data.pop(cfg)
assert_dict_equal(cfg_data['console'],
lpc.cfgs['console'])
def test_load_global_evars():
lpc = LinchpinCliContext()
lpc.load_config(lpconfig=config_path)
lpc.load_global_evars()
# remove workspace key as it's not from the config file
for k in lpc.evars.keys():
if k == 'workspace':
if lpc.evars.get(k):
lpc.evars.pop(k)
if evars_data.get(k):
evars_data.pop(k)
assert_dict_equal(evars_data, lpc.evars)
def test_log_info():
lvl=logging.INFO
msg = 'Info Msg'
regex = '^{0}.*{1}'.format(logging.getLevelName(lvl), msg)
lpc = LinchpinCliContext()
lpc.load_config(config_path)
lpc.setup_logging()
lpc.log_info(msg)
with open(logfile) as f:
line = f.readline()
assert_regexp_matches(line, regex)
def test_log_state():
lvl=logging.DEBUG
msg = '{0}: State Msg'.format(logging.getLevelName(lvl))
regex = '^{0}.*STATE - {1}'.format(logging.getLevelName(lvl), msg)
lpc = LinchpinCliContext()
lpc.load_config(config_path)
lpc.setup_logging()
lpc.log_state(msg)
with open(logfile) as f:
line = f.readline()
assert_regexp_matches(line, regex)
def test_load_global_evars():
lpc = LinchpinCliContext()
lpc.load_config(lpconfig=config_path)
lpc.load_global_evars()
# remove workspace key as it's not from the config file
for k in lpc.evars.keys():
if k == 'workspace':
if lpc.evars.get(k):
lpc.evars.pop(k)
if evars_data.get(k):
evars_data.pop(k)
assert_dict_equal(evars_data, lpc.evars)
def test_log_debug():
lvl = logging.DEBUG
msg = 'Debug Msg'
regex = '^{0}.*{1}'.format(logging.getLevelName(lvl), msg)
lpc = LinchpinCliContext()
lpc.load_config(lpconfig=config_path)
lpc.set_cfg('logger', 'file', config_data['logger']['file'])
lpc.setup_logging()
lpc.log_debug(msg)
with open(logfile) as f:
line = f.readline()
assertRegex(line, regex)
def test_log_debug():
lvl=logging.DEBUG
msg = 'Debug Msg'
regex = '^{0}.*{1}'.format(logging.getLevelName(lvl), msg)
lpc = LinchpinCliContext()
lpc.load_config(lpconfig=config_path)
lpc.set_cfg('logger',
'file',
config_data['logger']['file'])
lpc.setup_logging()
lpc.log_debug(msg)
with open(logfile) as f:
line = f.readline()
assert_regexp_matches(line, regex)
def test_log_msg():
# This test assumes the default message format found around line 139
# of linchpin/cli/context.py
lvl=logging.DEBUG
msg = 'Test Msg'
regex = '^{0}.*{1}'.format(logging.getLevelName(lvl), msg)
lpc = LinchpinCliContext()
lpc.load_config(config_path)
lpc.setup_logging()
lpc.log(msg, level=lvl)
with open(logfile) as f:
line = f.readline()
assert_regexp_matches(line, regex)
def test_log_msg():
# This test assumes the default message format found around line 139
# of linchpin/cli/context.py
lvl=logging.DEBUG
msg = 'Test Msg'
regex = '^{0}.*{1}'.format(logging.getLevelName(lvl), msg)
lpc = LinchpinCliContext()
lpc.load_config(lpconfig=config_path)
lpc.set_cfg('logger',
'file',
config_data['logger']['file'])
lpc.setup_logging()
lpc.log(msg, level=lvl)
with open(logfile) as f:
line = f.readline()
assert_regexp_matches(line, regex)
def setup_lp_fetch_env():
global lpc
global lpcli
global target
global pinfile
global mockpath
base_path = '{0}'.format(os.path.dirname(
os.path.realpath(__file__))).rstrip('/')
lib_path = os.path.realpath(os.path.join(base_path, os.pardir))
setup_load_config()
lpc = LinchpinCliContext()
lpc.load_config(lpconfig=config_path)
lpc.load_global_evars()
lpc.setup_logging()
lpc.workspace = tempfile.mkdtemp(prefix='fetch_')
print('workspace: {0}'.format(lpc.workspace))
lpcli = LinchpinCli(lpc)
def setup_lp_fetch_env():
global lpc
global lpcli
global target
global pinfile
global mockpath
base_path = '{0}'.format(os.path.dirname(
os.path.realpath(__file__))).rstrip('/')
lib_path = os.path.realpath(os.path.join(base_path, os.pardir))
setup_load_config()
lpc = LinchpinCliContext()
lpc.load_config(lpconfig=config_path)
lpc.load_global_evars()
lpc.setup_logging()
lpc.workspace = tempfile.mkdtemp(prefix='fetch_')
print(('workspace: {0}'.format(lpc.workspace)))
lpcli = LinchpinCli(lpc)
def _get_hosts(ctx, args, incomplete):
lpctx = LinchpinCliContext()
lpctx.load_config(lpconfig=None)
lpctx.load_global_evars()
lpcli = LinchpinCli(lpctx)
return [k for k in lpcli.ctx.inventory.hosts.keys() if incomplete in k]