def test_env_set_pre(self):
"""
TEST: set predefined environment variable from [env] entry
unconditionally
EXP: the old value gets overwritten
"""
self.dbgfunc()
sname = 'env'
evname = 'UTIL_TEST'
pre_val = "one:two:three"
add = "four:five:six"
exp = add
# make sure the target env variable is set to a known value
with U.tmpenv(evname, pre_val):
# create a config object with an 'env' section and a non-'+' option
cfg = hx.cfg.config()
cfg.add_section(sname)
cfg.set(sname, evname, add)
# pass the config object to U.env_update()
U.env_update(cfg)
# verify that the target env variable now contains the new value
# and the old value is gone
self.expected(exp, os.environ[evname])
self.assertTrue(
pre_val not in os.environ[evname],
"The old value should be gone but still seems " +
"to be hanging around")
def test_env_set_folded_none(self):
"""
TEST: set undefined environment variable from a folded [env] entry
unconditionally
EXP: the value gets set
"""
self.dbgfunc()
sname = 'env'
evname = 'UTIL_TEST'
newval = "one:\n two:\n three"
exp = re.sub("\n\s*", "", newval)
# make sure the target env variable is not defined
with U.tmpenv(evname, None):
# create a config object with an 'env' section and a non-'+' option
cfg = hx.cfg.config()
cfg.add_section(sname)
cfg.set(sname, evname, newval)
# pass the config object to U.env_update()
U.env_update(cfg)
# verify that the variable was set to the expected value
self.expected(exp, os.environ[evname])
def test_env_add_folded_pre(self):
"""
TEST: add to a preset environment variable from a folded [env]
entry
EXP: the value gets set to the payload with the whitespace squeezed out
"""
self.dbgfunc()
sname = 'env'
evname = 'UTIL_TEST'
pre_val = "one:two:three"
add = "four:\n five:\n six"
exp = ":".join([pre_val, re.sub("\n\s*", "", add)])
# make sure the target env variable has the expected value
with U.tmpenv(evname, pre_val):
# create a config object with an 'env' section and a folded '+'
# option
cfg = hx.cfg.config()
cfg.add_section(sname)
cfg.set(sname, evname, '+' + add)
# pass the config object to U.env_update()
U.env_update(cfg)
# verify that the variable was set to the expected value
self.expected(exp, os.environ[evname])
def test_env_add_pre(self):
"""
TEST: add to a predefined environment variable from [env] entry
EXP: payload is appended to the old value
"""
self.dbgfunc()
sname = 'env'
evname = 'UTIL_TEST'
pre_val = "one:two:three"
add = "four:five:six"
exp = ":".join([pre_val, add])
# make sure the target env variable is set to a known value
with U.tmpenv(evname, pre_val):
# create a config object with an 'env' section and a '+' option
cfg = hx.cfg.config()
cfg.add_section(sname)
cfg.set(sname, evname, "+" + add)
# pass the config object to U.env_update()
U.env_update(cfg)
# verify that the target env variable now contains both old and
# added values
self.expected(exp, os.environ[evname])
def test_env_add_folded_none(self):
"""
TEST: add to an undefined environment variable from a folded [env]
entry
EXP: the value gets set to the payload with the whitespace squeezed out
"""
self.dbgfunc()
sname = 'env'
evname = 'UTIL_TEST'
add = "four:\n five:\n six"
exp = re.sub("\n\s*", "", add)
# make sure the target env variable is not defined
with U.tmpenv(evname, None):
# create a config object with an 'env' section and a '+' option
cfg = hx.cfg.config()
cfg.add_section(sname)
cfg.set(sname, evname, '+' + add)
# pass the config object to U.env_update()
U.env_update(cfg)
# verify that the variable was set to the expected value
self.expected(exp, os.environ[evname])