From 4d030a8d96a3baf7855d840f0ca475d43cbdb18a Mon Sep 17 00:00:00 2001
From: Nick Van den Broeck <nick.van.den.broeck666@gmail.com>
Date: Tue, 28 May 2019 10:51:45 +0200
Subject: [PATCH] Added nonFlakeRequires test Fixes #2888

---
 src/libexpr/primops/flake.cc |  6 +++---
 tests/flakes.sh              | 40 +++++++++++++++++++++++++++++++++++-
 2 files changed, 42 insertions(+), 4 deletions(-)

diff --git a/src/libexpr/primops/flake.cc b/src/libexpr/primops/flake.cc
index 3c3d5e0c7..d82c2389d 100644
--- a/src/libexpr/primops/flake.cc
+++ b/src/libexpr/primops/flake.cc
@@ -366,9 +366,9 @@ Flake getFlake(EvalState & state, const FlakeRef & flakeRef, bool impureIsAllowe
 }
 
 // Get the `NonFlake` corresponding to a `FlakeRef`.
-NonFlake getNonFlake(EvalState & state, const FlakeRef & flakeRef, FlakeAlias alias)
+NonFlake getNonFlake(EvalState & state, const FlakeRef & flakeRef, FlakeAlias alias, bool impureIsAllowed = false)
 {
-    SourceInfo sourceInfo = fetchFlake(state, flakeRef);
+    SourceInfo sourceInfo = fetchFlake(state, flakeRef, impureIsAllowed);
     debug("got non-flake source '%s' with flakeref %s", sourceInfo.storePath, sourceInfo.resolvedRef.to_string());
 
     FlakeRef resolvedRef = sourceInfo.resolvedRef;
@@ -449,7 +449,7 @@ ResolvedFlake resolveFlakeFromLockFile(EvalState & state, const FlakeRef & flake
         } else {
             if (handleLockFile == AllPure || handleLockFile == TopRefUsesRegistries)
                 throw Error("cannot update non-flake dependency '%s' in pure mode", nonFlakeInfo.first);
-            deps.nonFlakeDeps.push_back(getNonFlake(state, nonFlakeInfo.second, nonFlakeInfo.first));
+            deps.nonFlakeDeps.push_back(getNonFlake(state, nonFlakeInfo.second, nonFlakeInfo.first, allowedToUseRegistries(handleLockFile, false)));
         }
     }
 
diff --git a/tests/flakes.sh b/tests/flakes.sh
index 179fef320..6081e8939 100644
--- a/tests/flakes.sh
+++ b/tests/flakes.sh
@@ -15,8 +15,9 @@ registry=$TEST_ROOT/registry.json
 flake1Dir=$TEST_ROOT/flake1
 flake2Dir=$TEST_ROOT/flake2
 flake3Dir=$TEST_ROOT/flake3
+nonFlakeDir=$TEST_ROOT/nonFlake
 
-for repo in $flake1Dir $flake2Dir $flake3Dir; do
+for repo in $flake1Dir $flake2Dir $flake3Dir $nonFlakeDir; do
     rm -rf $repo $repo.tmp
     mkdir $repo
     git -C $repo init
@@ -81,6 +82,13 @@ EOF
 git -C $flake3Dir add flake.nix
 git -C $flake3Dir commit -m 'Initial'
 
+cat > $nonFlakeDir/README.md <<EOF
+Not much
+EOF
+
+git -C $nonFlakeDir add README.md
+git -C $nonFlakeDir commit -m 'Initial'
+
 cat > $registry <<EOF
 {
     "flakes": {
@@ -199,3 +207,33 @@ nix build -o $TEST_ROOT/result --flake-registry file://$registry file://$flake2D
 nix build -o $TEST_ROOT/result --flake-registry file://$registry file://$flake2Dir:bar --tarball-ttl 0
 mv $flake1Dir.tmp $flake1Dir
 mv $flake2Dir.tmp $flake2Dir
+
+# Add nonFlakeRequires to flake3.
+rm $flake3Dir/flake.nix
+
+cat > $flake3Dir/flake.nix <<EOF
+{
+  name = "flake3";
+
+  epoch = 2019;
+
+  requires = [ "flake1" "flake2" ];
+
+  nonFlakeRequires = {
+    nonFlake = "$nonFlakeDir";
+  };
+
+  description = "Fnord";
+
+  provides = deps: rec {
+    packages.xyzzy = deps.flake2.provides.packages.bar;
+    packages.sth = deps.flake1.provides.packages.foo;
+  };
+}
+EOF
+
+git -C $flake3Dir add flake.nix
+git -C $flake3Dir commit -m 'Add nonFlakeRequires'
+
+# Check whether `nix build` works with a lockfile which is missing a nonFlakeRequires
+nix build -o $TEST_ROOT/result --flake-registry $registry $flake3Dir:sth