def test_vt_terminal_environ_cleanup(envvar, meipass):
with PatchedEnviron(**{envvar: meipass}):
original_env = dict(os.environ)
assert envvar in original_env
instance = overrides.PyinstallerTerminal(
[
sys.executable,
"-c",
"import os, json; print(json.dumps(dict(os.environ)))",
],
stream_stdout=False,
stream_stderr=False,
)
buffer_o = buffer_e = ""
while instance.has_unread_data:
stdout, stderr = instance.recv()
if stdout:
buffer_o += stdout
if stderr:
buffer_e += stderr
instance.terminate()
assert instance.exitstatus == 0
returned_env = json.loads(buffer_o)
assert returned_env != original_env
assert envvar in returned_env
assert returned_env[envvar] == ""
def test_vt_terminal_environ_cleanup_original_passed_directly(envvar, meipass):
orig_envvar = "{}_ORIG".format(envvar)
env = {
orig_envvar: meipass,
}
original_env = dict(os.environ)
instance = overrides.PyinstallerTerminal(
[
sys.executable, "-c",
"import os, json; print(json.dumps(dict(os.environ)))"
],
env=env.copy(),
stream_stdout=False,
stream_stderr=False,
)
buffer_o = buffer_e = ""
while instance.has_unread_data:
stdout, stderr = instance.recv()
if stdout:
buffer_o += stdout
if stderr:
buffer_e += stderr
instance.terminate()
assert instance.exitstatus == 0
returned_env = json.loads(buffer_o)
assert returned_env != original_env
assert envvar in returned_env
assert orig_envvar not in returned_env
assert returned_env[envvar] == meipass