mirror of
https://github.com/nix-community/nix-direnv.git
synced 2024-09-11 11:35:54 +03:00
49 lines
1.1 KiB
Python
49 lines
1.1 KiB
Python
import shutil
|
|
import textwrap
|
|
from collections.abc import Iterator
|
|
from dataclasses import dataclass
|
|
from pathlib import Path
|
|
from tempfile import TemporaryDirectory
|
|
|
|
import pytest
|
|
|
|
from .procs import run
|
|
|
|
|
|
@dataclass
|
|
class DirenvProject:
|
|
directory: Path
|
|
nix_direnv: Path
|
|
|
|
@property
|
|
def envrc(self) -> Path:
|
|
return self.directory / ".envrc"
|
|
|
|
def setup_envrc(self, content: str, strict_env: bool) -> None:
|
|
text = textwrap.dedent(
|
|
f"""
|
|
{'strict_env' if strict_env else ''}
|
|
source {self.nix_direnv}
|
|
{content}
|
|
"""
|
|
)
|
|
self.envrc.write_text(text)
|
|
run(["direnv", "allow"], cwd=self.directory)
|
|
|
|
|
|
@pytest.fixture()
|
|
def direnv_project(test_root: Path, project_root: Path) -> Iterator[DirenvProject]:
|
|
"""
|
|
Setups a direnv test project
|
|
"""
|
|
with TemporaryDirectory() as _dir:
|
|
directory = Path(_dir) / "proj"
|
|
shutil.copytree(test_root / "testenv", directory)
|
|
nix_direnv = project_root / "direnvrc"
|
|
|
|
c = DirenvProject(Path(directory), nix_direnv)
|
|
try:
|
|
yield c
|
|
finally:
|
|
pass
|