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]