def env():
env = Environment()
env.define("MyVar", "MyValue")
env.define("MyVar1", "MyValue1")
env.append("MyVar2", "MyValue2")
env.prepend("MyVar3", "MyValue3")
env.unset("MyVar4")
env.define("MyVar5", "MyValue5 With Space5=More Space5;:More")
env.append("MyVar6", "MyValue6") # Append, but previous not existing
env.define_path("MyPath1", "/Some/Path1/")
env.append_path("MyPath2", ["/Some/Path2/", "/Other/Path2/"])
env.prepend_path("MyPath3", "/Some/Path3/")
env.unset("MyPath4")
return env
def test_dict_access():
env = Environment()
env.append("MyVar", "MyValue", separator="@")
ret = env.vars(ConanFileMock()).items()
assert dict(ret) == {"MyVar": "MyValue"}
env = Environment()
env.prepend("MyVar", "MyValue", separator="@")
env_vars = env.vars(ConanFileMock())
assert dict(env_vars.items()) == {"MyVar": "MyValue"}
assert env_vars["MyVar"] == "MyValue"
env2 = Environment()
env2.define("MyVar", "MyValue2")
env.compose_env(env2)
env_vars = env.vars(ConanFileMock())
assert dict(env_vars.items()) == {"MyVar": "MyValue@MyValue2"}
with pytest.raises(KeyError):
_ = env_vars["Missing"]
# With previous values in the environment
env = Environment()
env.prepend("MyVar", "MyValue", separator="@")
old_env = dict(os.environ)
os.environ.update({"MyVar": "PreviousValue"})
env_vars = env.vars(ConanFileMock())
try:
assert env_vars["MyVar"] == "MyValue@PreviousValue"
finally:
os.environ.clear()
os.environ.update(old_env)
env = Environment()
env.append_path("MyVar", "MyValue")
old_env = dict(os.environ)
os.environ.update({"MyVar": "PreviousValue"})
env_vars = env.vars(ConanFileMock())
try:
assert env_vars["MyVar"] == "PreviousValue{}MyValue".format(os.pathsep)
with env_vars.apply():
assert os.getenv("MyVar") == "PreviousValue{}MyValue".format(
os.pathsep)
finally:
os.environ.clear()
os.environ.update(old_env)
assert list(env_vars.keys()) == ["MyVar"]
assert dict(env_vars.items()) == {"MyVar": "MyValue"}
def test_public_access():
env = Environment(ConanFileMock())
env.define("MyVar", "MyValue")
env.append("MyVar", "MyValue2")
env.define_path("MyPath", "c:/path/to/something")
env.append_path("MyPath", "D:/Otherpath")
for name, values in env.items():
if name == "MyVar":
assert values == "MyValue MyValue2"
if name == "MyPath":
assert values == "c:/path/to/something{}D:/Otherpath".format(
os.pathsep)
env.remove("MyPath", "D:/Otherpath")
assert env.get("MyPath") == "c:/path/to/something"
def test_env_files():
env = Environment()
env.define("MyVar", "MyValue")
env.define("MyVar1", "MyValue1")
env.append("MyVar2", "MyValue2")
env.prepend("MyVar3", "MyValue3")
env.unset("MyVar4")
env.define("MyVar5", "MyValue5 With Space5=More Space5;:More")
env.append("MyVar6", "MyValue6") # Append, but previous not existing
env.define_path("MyPath1", "/Some/Path1/")
env.append_path("MyPath2", ["/Some/Path2/", "/Other/Path2/"])
env.prepend_path("MyPath3", "/Some/Path3/")
env.unset("MyPath4")
folder = temp_folder()
prevenv = {"MyVar1": "OldVar1",
"MyVar2": "OldVar2",
"MyVar3": "OldVar3",
"MyVar4": "OldVar4",
"MyPath1": "OldPath1",
"MyPath2": "OldPath2",
"MyPath3": "OldPath3",
"MyPath4": "OldPath4",
}
display_bat = textwrap.dedent("""\
@echo off
echo MyVar=%MyVar%!!
echo MyVar1=%MyVar1%!!
echo MyVar2=%MyVar2%!!
echo MyVar3=%MyVar3%!!
echo MyVar4=%MyVar4%!!
echo MyVar5=%MyVar5%!!
echo MyVar6=%MyVar6%!!
echo MyPath1=%MyPath1%!!
echo MyPath2=%MyPath2%!!
echo MyPath3=%MyPath3%!!
echo MyPath4=%MyPath4%!!
""")
display_sh = textwrap.dedent("""\
echo MyVar=$MyVar!!
echo MyVar1=$MyVar1!!
echo MyVar2=$MyVar2!!
echo MyVar3=$MyVar3!!
echo MyVar4=$MyVar4!!
echo MyVar5=$MyVar5!!
echo MyVar6=$MyVar6!!
echo MyPath1=$MyPath1!!
echo MyPath2=$MyPath2!!
echo MyPath3=$MyPath3!!
echo MyPath4=$MyPath4!!
""")
def check(cmd_):
out, _ = subprocess.Popen(cmd_, stdout=subprocess.PIPE, stderr=subprocess.STDOUT,
env=prevenv, shell=True).communicate()
out = out.decode()
assert "MyVar=MyValue!!" in out
assert "MyVar1=MyValue1!!" in out
assert "MyVar2=OldVar2 MyValue2!!" in out
assert "MyVar3=MyValue3 OldVar3!!" in out
assert "MyVar4=!!" in out
assert "MyVar5=MyValue5 With Space5=More Space5;:More!!" in out
assert "MyVar6= MyValue6!!" in out # The previous is non existing, append has space
assert "MyPath1=/Some/Path1/!!" in out
assert "MyPath2=OldPath2:/Some/Path2/:/Other/Path2/!!" in out
assert "MyPath3=/Some/Path3/:OldPath3!!" in out
assert "MyPath4=!!" in out
# This should be output when deactivated
assert "MyVar=!!" in out
assert "MyVar1=OldVar1!!" in out
assert "MyVar2=OldVar2!!" in out
assert "MyVar3=OldVar3!!" in out
assert "MyVar4=OldVar4!!" in out
assert "MyVar5=!!" in out
assert "MyVar6=!!" in out
assert "MyPath1=OldPath1!!" in out
assert "MyPath2=OldPath2!!" in out
assert "MyPath3=OldPath3!!" in out
assert "MyPath4=OldPath4!!" in out
with chdir(folder):
if platform.system() == "Windows":
env.save_bat("test.bat", pathsep=":", generate_deactivate=True)
save("display.bat", display_bat)
cmd = "test.bat && display.bat && deactivate_test.bat && display.bat"
check(cmd)
# FIXME: Powershell still not working
# env.save_ps1("test.ps1", pathsep=":")
# cmd = 'powershell.exe -ExecutionPolicy ./test.ps1; gci env:'
# shell = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
# (stdout, stderr) = shell.communicate()
# stdout, stderr = decode_text(stdout), decode_text(stderr)
# check(cmd)
else:
env.save_sh("test.sh", generate_deactivate=True)
save("display.sh", display_sh)
os.chmod("display.sh", 0o777)
cmd = '. ./test.sh && ./display.sh && . ./deactivate_test.sh && ./display.sh'
check(cmd)