repo2docker/tests/unit/test_env.py

38 wiersze
947 B
Python

"""
Test that environment variables may be defined
"""
import os
import subprocess
import tempfile
import time
def test_env():
"""
Validate that you can define environment variables
"""
ts = str(time.time())
with tempfile.TemporaryDirectory() as tmpdir:
username = os.getlogin()
subprocess.check_call(
[
"repo2docker",
"-v",
"{}:/home/{}".format(tmpdir, username),
"-e",
"FOO={}".format(ts),
"--env",
"BAR=baz",
"--",
tmpdir,
"/bin/bash",
"-c",
"echo -n $FOO > ts && echo -n $BAR > bar",
]
)
with open(os.path.join(tmpdir, "ts")) as f:
assert f.read().strip() == ts
with open(os.path.join(tmpdir, "bar")) as f:
assert f.read().strip() == "baz"