tests: migrate export-graph.sh

Change-Id: I0de67ba6b1f710434f504814dd9b5eb7d486e65a

tests/functional2/store/test_export_graph.py

@@ -0,0 +1,46 @@
+from testlib.utils import get_global_asset_pack
+from pathlib import Path
+from testlib.fixtures.file_helper import with_files
+from testlib.fixtures.file_helper import CopyFile
+from testlib.fixtures.nix import Nix
+
+export_files = {"export-graph.nix": CopyFile("assets/export-graph.nix")} | get_global_asset_pack(
+    "dependencies"
+)
+
+
+def check_ref(nix: Nix, file: str):
+    res = nix.nix_store(["-q", "--references", nix.env.dirs.home / "result"]).run().ok()
+    assert file in res.stdout_plain
+
+
+def check_out_path_refs(nix: Nix, out_path: str):
+    for file in Path(out_path).read_text().splitlines():
+        check_ref(nix, file)
+
+
+@with_files(export_files)
+def test_export_runtime(nix: Nix, files: Path):
+    result = files / "result"
+    out_path = (
+        nix.nix_build(["./export-graph.nix", "-A", 'foo."bar.runtimeGraph"', "-o", result])
+        .run()
+        .ok()
+    )
+    references = nix.nix_store(["-q", "--references", result]).run().ok().stdout_plain
+    assert len(references.splitlines()) == 4
+
+    check_ref(nix, "input-2")
+    check_out_path_refs(nix, out_path.stdout_plain)
+
+
+@with_files(export_files)
+def test_built_time(nix: Nix, files: Path):
+    result = files / "result"
+    out_path = (
+        nix.nix_build(["./export-graph.nix", "-A", 'foo."bar.buildGraph"', "-o", result]).run().ok()
+    )
+
+    for file in ["input-1", "input-1.drv", "input-2", "input-2.drv"]:
+        check_ref(nix, file)
+    check_out_path_refs(nix, out_path.stdout_plain)