mirror of
https://github.com/nix-community/nix-direnv.git
synced 2025-11-08 19:46:11 +01:00
65 lines
1.9 KiB
Python
65 lines
1.9 KiB
Python
#!/usr/bin/env python2
|
|
|
|
from tempfile import TemporaryDirectory
|
|
import os
|
|
import sys
|
|
import subprocess
|
|
from pathlib import Path
|
|
import shutil
|
|
import unittest
|
|
from typing import List
|
|
|
|
|
|
TEST_ROOT = Path(__file__).resolve().parent
|
|
|
|
|
|
def run(cmd: List[str], **kwargs) -> subprocess.CompletedProcess:
|
|
print("$ " + " ".join(cmd))
|
|
return subprocess.run(cmd, **kwargs)
|
|
|
|
|
|
class IntegrationTest(unittest.TestCase):
|
|
def setUp(self) -> None:
|
|
self.env = os.environ.copy()
|
|
self.dir = TemporaryDirectory()
|
|
self.env["HOME"] = str(self.dir.name)
|
|
self.testenv = Path(self.dir.name).joinpath("testenv")
|
|
shutil.copytree(TEST_ROOT.joinpath("testenv"), self.testenv)
|
|
direnvrc = str(TEST_ROOT.parent.joinpath("direnvrc"))
|
|
with open(self.testenv.joinpath(".envrc"), "w") as f:
|
|
f.write(f"source {direnvrc}\n" "use nix")
|
|
|
|
def test_direnv(self) -> None:
|
|
run(["direnv", "allow"], cwd=str(self.testenv), env=self.env, check=True)
|
|
|
|
run(["nix-collect-garbage"], check=True)
|
|
|
|
out1 = run(
|
|
["direnv", "exec", str(self.testenv), "hello"],
|
|
env=self.env,
|
|
stderr=subprocess.PIPE,
|
|
text=True,
|
|
)
|
|
sys.stderr.write(out1.stderr)
|
|
self.assertIn("renewed cache and derivation link", out1.stderr)
|
|
self.assertEqual(out1.returncode, 0)
|
|
|
|
run(["nix-collect-garbage"], check=True)
|
|
run(["sh", "-c", "realpath /nix/var/nix/gcroots/per-user/$USER/*"], check=True)
|
|
|
|
out2 = run(
|
|
["direnv", "exec", str(self.testenv), "hello"],
|
|
env=self.env,
|
|
stderr=subprocess.PIPE,
|
|
text=True,
|
|
)
|
|
sys.stderr.write(out2.stderr)
|
|
self.assertIn("using cached derivation", out2.stderr)
|
|
self.assertEqual(out2.returncode, 0)
|
|
|
|
def tearDown(self) -> None:
|
|
self.dir.cleanup()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|