repo2docker/tests/env.py

32 wiersze
825 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'