integration-tests: Add flakes test

integration-tests/default.nix

@@ -1,6 +1,7 @@
 {
   apply = import ./apply;
   apply-local = import ./apply-local;
-  parallel = import ./parallel;
   exec = import ./exec;
+  flakes = import ./flakes;
+  parallel = import ./parallel;
 }

integration-tests/flakes/default.nix

@@ -0,0 +1,34 @@
+let
+  tools = import ../tools.nix {
+    targets = [ "alpha" ];
+  };
+in tools.makeTest {
+  name = "colmena-flakes";
+
+  bundle = ./.;
+
+  testScript = ''
+    import re
+
+    with subtest("Lock flake dependencies"):
+        # --impure required for path:/nixpkgs which is a symlink to a store path
+        deployer.succeed("cd /tmp/bundle && nix --experimental-features \"nix-command flakes\" flake lock --impure")
+
+    with subtest("Deploy with a plain flake without git"):
+        deployer.succeed("cd /tmp/bundle && ${tools.colmenaExec} apply --on @target")
+        alpha.succeed("grep FIRST /etc/deployment")
+
+    with subtest("Deploy with a git flake"):
+        deployer.succeed("sed -i s/FIRST/SECOND/g /tmp/bundle/probe.nix")
+
+        # don't put probe.nix in source control - should fail
+        deployer.succeed("cd /tmp/bundle && git init && git add flake.nix flake.lock hive.nix tools.nix")
+        logs = deployer.fail("cd /tmp/bundle && run-copy-stderr ${tools.colmenaExec} apply --on @target")
+        assert re.search(r"probe.nix.*No such file or directory", logs)
+
+        # now it should succeed
+        deployer.succeed("cd /tmp/bundle && git add probe.nix")
+        deployer.succeed("cd /tmp/bundle && ${tools.colmenaExec} apply --on @target")
+        alpha.succeed("grep SECOND /etc/deployment")
+  '';
+}

integration-tests/flakes/flake.nix

@@ -0,0 +1,15 @@
+{
+  description = "A simple deployment";
+
+  inputs = {
+    nixpkgs.url = "path:/nixpkgs";
+  };
+
+  outputs = { self, nixpkgs }: let
+    pkgs = import nixpkgs {
+      system = "x86_64-linux";
+    };
+  in {
+    colmena = import ./hive.nix { inherit pkgs; };
+  };
+}

integration-tests/flakes/hive.nix

@@ -0,0 +1,21 @@
+{ pkgs }:
+
+let
+  tools = import ./tools.nix {
+    insideVm = true;
+    targets = [ "alpha" ];
+  };
+in {
+  meta = {
+    nixpkgs = tools.pkgs;
+  };
+
+  deployer = tools.getStandaloneConfigFor "deployer";
+  alpha = {
+    imports = [
+      (tools.getStandaloneConfigFor "alpha")
+    ];
+
+    environment.etc."deployment".text = import ./probe.nix;
+  };
+}

integration-tests/flakes/probe.nix

@@ -0,0 +1 @@
+"FIRST"

integration-tests/tools.nix

@@ -61,10 +61,12 @@ let
         ];
       };
 
-      environment.systemPackages = [
+      environment.systemPackages = with pkgs; [
+        git # for git flake tests
+
         # HACK: copy stderr to both stdout and stderr
         # (the test framework only captures stdout, and only stderr appears on screen during the build)
-        (pkgs.writeShellScriptBin "run-copy-stderr" ''
+        (writeShellScriptBin "run-copy-stderr" ''
           exec "$@" 2> >(tee /dev/stderr)
         '')
       ];