def test_file_obj(self, mocker):
e = IOError()
e.errno = errno.EPERM
def side_effect(*args, **kwargs):
if args[0] == 'nofile':
raise e
file_mock(mocker, side_effect=side_effect)
with pytest.raises(IOError):
pv_cli_file_utility.get_file_obj('nofile')
e.errno = errno.ENOENT
pv_cli_file_utility.get_file_obj('nofile')
def get_raw_profile():
profile_file = file_utils.get_profile_file()
fp = file_utils.get_file_obj(profile_file, 'rb')
cfg = None
if fp:
cfg = six.moves.configparser.SafeConfigParser()
with fp:
cfg.readfp(fp)
return cfg
def write_profile_config(cfg, mode='w'):
profile_file = file_utils.get_profile_file()
fp = file_utils.get_file_obj(profile_file, mode=mode)
with fp:
cfg.write(fp)
return cfg
