mirror of
https://github.com/nix-community/nix-direnv.git
synced 2024-09-11 11:35:54 +03:00
72 lines
2.1 KiB
Python
72 lines
2.1 KiB
Python
import logging
|
|
import os
|
|
import shlex
|
|
import subprocess
|
|
import sys
|
|
import unittest
|
|
|
|
import pytest
|
|
|
|
from .direnv_project import DirenvProject
|
|
from .procs import run
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
def direnv_exec(
|
|
direnv_project: DirenvProject, cmd: str, env: dict[str, str] | None = None
|
|
) -> None:
|
|
args = ["direnv", "exec", str(direnv_project.directory), "sh", "-c", cmd]
|
|
log.debug(f"$ {shlex.join(args)}")
|
|
out = run(
|
|
args,
|
|
stderr=subprocess.PIPE,
|
|
stdout=subprocess.PIPE,
|
|
check=False,
|
|
cwd=direnv_project.directory,
|
|
env=env,
|
|
)
|
|
sys.stdout.write(out.stdout)
|
|
sys.stderr.write(out.stderr)
|
|
assert out.returncode == 0
|
|
assert out.stdout == "OK\n"
|
|
assert "Renewed cache" in out.stderr
|
|
|
|
|
|
@pytest.mark.parametrize("strict_env", [False, True])
|
|
def test_attrs(direnv_project: DirenvProject, strict_env: bool) -> None:
|
|
direnv_project.setup_envrc("use nix -A subshell", strict_env=strict_env)
|
|
direnv_exec(direnv_project, "echo $THIS_IS_A_SUBSHELL")
|
|
|
|
|
|
@pytest.mark.parametrize("strict_env", [False, True])
|
|
def test_no_nix_path(direnv_project: DirenvProject, strict_env: bool) -> None:
|
|
direnv_project.setup_envrc("use nix --argstr someArg OK", strict_env=strict_env)
|
|
env = os.environ.copy()
|
|
del env["NIX_PATH"]
|
|
direnv_exec(direnv_project, "echo $SHOULD_BE_SET", env=env)
|
|
|
|
|
|
@pytest.mark.parametrize("strict_env", [False, True])
|
|
def test_args(direnv_project: DirenvProject, strict_env: bool) -> None:
|
|
direnv_project.setup_envrc("use nix --argstr someArg OK", strict_env=strict_env)
|
|
direnv_exec(direnv_project, "echo $SHOULD_BE_SET")
|
|
|
|
|
|
@pytest.mark.parametrize("strict_env", [False, True])
|
|
def test_no_files(direnv_project: DirenvProject, strict_env: bool) -> None:
|
|
direnv_project.setup_envrc("use nix -p hello", strict_env=strict_env)
|
|
out = run(
|
|
["direnv", "status"],
|
|
stderr=subprocess.PIPE,
|
|
stdout=subprocess.PIPE,
|
|
check=False,
|
|
cwd=direnv_project.directory,
|
|
)
|
|
assert out.returncode == 0
|
|
assert 'Loaded watch: "."' not in out.stdout
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|