"""
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"