From ba0bd8fae63e4f38cd735d8586f86a7a6e7b363a Mon Sep 17 00:00:00 2001
From: Cameron Dart <cdart2@illinois.edu>
Date: Tue, 2 Apr 2024 18:10:36 -0700
Subject: [PATCH] Add functional tests for include directive in nix config file

---
 tests/functional/config.sh                         | 12 +++++++++++-
 tests/functional/config/extra-config.conf          |  1 +
 tests/functional/config/nix-with-bang-include.conf |  2 ++
 tests/functional/config/nix-with-include.conf      |  2 ++
 4 files changed, 16 insertions(+), 1 deletion(-)
 create mode 100644 tests/functional/config/extra-config.conf
 create mode 100644 tests/functional/config/nix-with-bang-include.conf
 create mode 100644 tests/functional/config/nix-with-include.conf

diff --git a/tests/functional/config.sh b/tests/functional/config.sh
index 324fe95bd..efdf2a958 100644
--- a/tests/functional/config.sh
+++ b/tests/functional/config.sh
@@ -43,6 +43,16 @@ export NIX_USER_CONF_FILES=$here/config/nix-with-substituters.conf
 var=$(nix config show | grep '^substituters =' | cut -d '=' -f 2 | xargs)
 [[ $var == https://example.com ]]
 
+# Test that we can include a file.
+export NIX_USER_CONF_FILES=$here/config/nix-with-include.conf
+var=$(nix config show | grep '^allowed-uris =' | cut -d '=' -f 2 | xargs)
+[[ $var == https://github.com/NixOS/nix ]]
+
+# Test that we can !include a file.
+export NIX_USER_CONF_FILES=$here/config/nix-with-bang-include.conf
+var=$(nix config show | grep '^experimental-features =' | cut -d '=' -f 2 | xargs)
+[[ $var == nix-command ]]
+
 # Test that it's possible to load config from the environment
 prev=$(nix config show | grep '^cores' | cut -d '=' -f 2 | xargs)
 export NIX_CONFIG="cores = 4242"$'\n'"experimental-features = nix-command flakes"
@@ -56,4 +66,4 @@ exp_features=$(nix config show | grep '^experimental-features' | cut -d '=' -f 2
 # Test that it's possible to retrieve a single setting's value
 val=$(nix config show | grep '^warn-dirty' | cut -d '=' -f  2 | xargs)
 val2=$(nix config show warn-dirty)
-[[ $val == $val2 ]]
+[[ $val == $val2 ]]
\ No newline at end of file
diff --git a/tests/functional/config/extra-config.conf b/tests/functional/config/extra-config.conf
new file mode 100644
index 000000000..d110f06e4
--- /dev/null
+++ b/tests/functional/config/extra-config.conf
@@ -0,0 +1 @@
+allowed-uris = https://github.com/NixOS/nix
\ No newline at end of file
diff --git a/tests/functional/config/nix-with-bang-include.conf b/tests/functional/config/nix-with-bang-include.conf
new file mode 100644
index 000000000..fa600e6ff
--- /dev/null
+++ b/tests/functional/config/nix-with-bang-include.conf
@@ -0,0 +1,2 @@
+experimental-features = nix-command
+!include ./missing-extra-config.conf
\ No newline at end of file
diff --git a/tests/functional/config/nix-with-include.conf b/tests/functional/config/nix-with-include.conf
new file mode 100644
index 000000000..17b8958ba
--- /dev/null
+++ b/tests/functional/config/nix-with-include.conf
@@ -0,0 +1,2 @@
+experimental-features = nix-command
+include ./extra-config.conf
\ No newline at end of file