Verify b's device ID gets set also

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

diff --git a/tests/automatic-enrollment.nix b/tests/automatic-enrollment.nix
index 49325fa90e5f71d19270683c93f0964087b978fb..951a4153e000be6bb453522fab7f4bcde415d1f4 100644 (file)
--- a/tests/automatic-enrollment.nix
+++ b/tests/automatic-enrollment.nix
@@ -1,5 +1,6 @@
 import <nixpkgs/nixos/tests/make-test-python.nix> ({ pkgs, lib, ... }:
   let
 import <nixpkgs/nixos/tests/make-test-python.nix> ({ pkgs, lib, ... }:
   let

+    deviceIDFilename = "/tmp/syncthing-auto-register-test-device-ids.nix";

     configuration = pkgs.writeText "configuration.nix" ''
       { pkgs, ... }: {
         imports = [
     configuration = pkgs.writeText "configuration.nix" ''
       { pkgs, ... }: {
         imports = [


@@ -11,18 +12,30 @@ import <nixpkgs/nixos/tests/make-test-python.nix> ({ pkgs, lib, ... }:
           openDefaultPorts = true;
           autoRegister = {
             enable = true;
           openDefaultPorts = true;
           autoRegister = {
             enable = true;

-            path = "/tmp/syncthing-auto-register-test-device-ids.nix";
+            path = "${deviceIDFilename}";

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

+    initialConfiguration = pkgs.writeText "initial-configuration.nix" ''
+      {
+        imports = [ "${configuration}" ];
+        system.activationScripts.installInitialConfiguration = {
+          text = '''
+            mkdir -p /etc/nixos
+            cp "${configuration}" /etc/nixos/configuration.nix
+          ''';
+          deps = [ ];
+        };
+      }
+    '';

     configurationWithDeviceIDs =
       pkgs.writeText "configuration-with-device-ids.nix" ''
         {
           imports = [
             ${configuration}
     configurationWithDeviceIDs =
       pkgs.writeText "configuration-with-device-ids.nix" ''
         {
           imports = [
             ${configuration}

-            /tmp/syncthing-auto-register-test-device-ids.nix
+            ${deviceIDFilename}

           ];
         }
       '';
           ];
         }
       '';


@@ -30,23 +43,55 @@ import <nixpkgs/nixos/tests/make-test-python.nix> ({ pkgs, lib, ... }:
 
     name = "syncthing";
 
 
     name = "syncthing";
 

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

 
     testScript = ''
 
     testScript = ''

+      import os
+
+

       def setConfig(machine, config):
           machine.copy_from_host(config, "/etc/nixos/configuration.nix")
 
 
       def setConfig(machine, config):
           machine.copy_from_host(config, "/etc/nixos/configuration.nix")
 
 

-      initialConfiguration = "${configuration}"
+      def verifyDeviceIDSet(machine, expectedDevice):
+          machine.succeed(
+              '(( "$(nix eval --raw -f "<nixos/nixos>" config.services.syncthing.declarative.devices.%s.id | wc -c)" == 63 ))'
+              % expectedDevice
+          )
+
+
+      def getDeviceFile(machine):
+          machine.copy_from_vm("${deviceIDFilename}")
+
+
+      def putDeviceFile(machine):
+          hostDeviceIDFilename = os.path.join(
+              os.environ["out"],
+              os.path.basename("${deviceIDFilename}"),
+          )
+          machine.copy_from_host(
+              hostDeviceIDFilename, "${deviceIDFilename}"
+          )
+          machine.succeed("chown syncthing ${deviceIDFilename}")
+
+

       configurationWithDeviceIDs = (
           "${configurationWithDeviceIDs}"
       )
 
       configurationWithDeviceIDs = (
           "${configurationWithDeviceIDs}"
       )
 

-      setConfig(machine, initialConfiguration)
-      machine.wait_for_unit("syncthing-autoregister.service")
-      setConfig(machine, configurationWithDeviceIDs)
-      machine.succeed(
-          '(( "$(nix eval --raw -f "<nixos/nixos>" config.services.syncthing.declarative.devices.machine.id | wc -c)" == 63 ))'
-      )
+      a.wait_for_unit("syncthing-autoregister.service")
+      setConfig(a, configurationWithDeviceIDs)
+      verifyDeviceIDSet(a, "a")
+      getDeviceFile(a)
+
+      putDeviceFile(b)
+      setConfig(b, configurationWithDeviceIDs)
+      verifyDeviceIDSet(b, "a")
+      b.wait_for_unit("syncthing-autoregister.service")
+      verifyDeviceIDSet(b, "a")
+      verifyDeviceIDSet(b, "b")

     '';
   })
     '';
   })