verifyDeviceIDSet() helper

[syncthing-autoregister] / tests / automatic-enrollment.nix

diff --git a/tests/automatic-enrollment.nix b/tests/automatic-enrollment.nix
index cac1b28ba83956bea3ea77f67fbe355ea4b81411..8cbcdc446b5743b57dcc2458e000651a1e0576f9 100644 (file)
--- a/tests/automatic-enrollment.nix
+++ b/tests/automatic-enrollment.nix
@@ -9,24 +9,52 @@ import <nixpkgs/nixos/tests/make-test-python.nix> ({ pkgs, lib, ... }:
         services.syncthing = {
           enable = true;
           openDefaultPorts = true;
         services.syncthing = {
           enable = true;
           openDefaultPorts = true;

+          autoRegister = {
+            enable = true;
+            path = "/tmp/syncthing-auto-register-test-device-ids.nix";
+          };

         };
         environment.systemPackages = with pkgs; [ nix ];
       }
     '';
         };
         environment.systemPackages = with pkgs; [ nix ];
       }
     '';

+    configurationWithDeviceIDs =
+      pkgs.writeText "configuration-with-device-ids.nix" ''
+        {
+          imports = [
+            ${configuration}
+            /tmp/syncthing-auto-register-test-device-ids.nix
+          ];
+        }
+      '';

   in {
 
     name = "syncthing";
 
   in {
 
     name = "syncthing";
 

-    machine = "${configuration}";
+    nodes = {
+      a = "${configuration}";
+      b = "${configuration}";
+    };

 
     testScript = ''
 
     testScript = ''

-      machine.copy_from_host(
-          "${configuration}",
-          "/etc/nixos/configuration.nix",
-      )
-      machine.wait_for_unit("syncthing-autoregister.service")
-      machine.succeed(
-          '(( "$(nix eval --raw -f /tmp/syncthing-auto-register-test-device-ids.nix services.syncthing.declarative.devices.machine.id | wc -c)" == 63 ))'
+      def setConfig(machine, config):
+          machine.copy_from_host(config, "/etc/nixos/configuration.nix")
+
+
+      def verifyDeviceIDSet(machine, expectedDevice):
+          machine.succeed(
+              '(( "$(nix eval --raw -f "<nixos/nixos>" config.services.syncthing.declarative.devices.%s.id | wc -c)" == 63 ))'
+              % expectedDevice
+          )
+
+
+      initialConfiguration = "${configuration}"
+      configurationWithDeviceIDs = (
+          "${configurationWithDeviceIDs}"

       )
       )

+
+      setConfig(a, initialConfiguration)
+      a.wait_for_unit("syncthing-autoregister.service")
+      setConfig(a, configurationWithDeviceIDs)
+      verifyDeviceIDSet(a, "a")

     '';
   })
     '';
   })