testing

Vagrantfile

@@ -10,60 +10,93 @@ if File.exist?(settings_path)
 end
 
 # Can't use `||` here because `false || true` evaluates to true
-VAGRANT_PROV = settings.key?('VAGRANT_PROV') ? settings['VAGRANT_PROV'] : true
-VAGRANT_NAME = settings['VAGRANT_NAME'] || 'zfstest'
-VAGRANT_BOX  = settings['VAGRANT_BOX']  || 'debian/bookworm64'
-VAGRANT_CPUS = settings['VAGRANT_CPUS'] || 4
-VAGRANT_MEM  = settings['VAGRANT_MEM']  || 4096
-VAGRANT_SH   = settings['VAGRANT_SH']   || ''
-NUM_DISKS    = settings['NUM_DISKS']    || 16
-# 18 TB decimal/SI to binary/IEC conversion:
+VAGRANT_PROV  = settings.key?('VAGRANT_PROV') ? settings['VAGRANT_PROV'] : true
+VAGRANT_NODES = settings['VAGRANT_NODES'] || 1
+VAGRANT_NAME  = settings['VAGRANT_NAME'] || 'zfstest'
+VAGRANT_BOX   = settings['VAGRANT_BOX']  || 'debian/bookworm64'
+VAGRANT_CPUS  = settings['VAGRANT_CPUS'] || 4
+VAGRANT_MEM   = settings['VAGRANT_MEM']  || 4096
+VAGRANT_SH    = settings['VAGRANT_SH']   || ''
+NUM_DISKS     = settings['NUM_DISKS']    || 16
+# 18 TB decimal/SI to binary/IEC conversion
 # 18 TB x (1000^4 / 1024^4) = 16.37 TiB
 # 16.37 TiB x 1024 = ~16763 GiB
-DISK_SIZE    = settings['DISK_SIZE']    || '16763G'
+DISK_SIZE     = settings['DISK_SIZE']    || '16763G'
+
+NODES = settings['NODES'] || {}
 
 Vagrant.configure("2") do |config|
   config.vm.box = VAGRANT_BOX
-  config.vm.hostname = VAGRANT_NAME
   config.vm.synced_folder ".", "/vagrant", type: "rsync",
     rsync__exclude: ".git"
 
-  config.vm.provider :libvirt do |libvirt|
-    libvirt.cpus   = VAGRANT_CPUS
-    libvirt.memory = VAGRANT_MEM
-    # Doesn't boot without this on rockylinux/9
-    libvirt.machine_virtual_size = 10
+  HOSTS = Array(1..VAGRANT_NODES)
+  HOSTS.each do |count|
+    DEFAULT_NODE_NAME = "#{VAGRANT_NAME}#{count}"
+    NODE_NAME = NODES.dig(DEFAULT_NODE_NAME, 'NAME') || DEFAULT_NODE_NAME
 
-    (1..NUM_DISKS).each do |i|
-      if i <= 676
-        first_char = (97 + ((i - 1) / 26)).chr
-        second_char = (97 + ((i - 1) % 26)).chr
-        device_suffix = first_char + second_char
-      else
-        raise "Error: Exceeded number of supported disks (#{NUM_DISKS}/676))"
+    config.vm.define NODE_NAME do |node_config|
+      node_config.vm.hostname = NODE_NAME
+      node_config.vm.box = NODES.dig(DEFAULT_NODE_NAME, 'BOX') || VAGRANT_BOX
+
+      node_config.vm.provider :libvirt do |libvirt|
+        libvirt.cpus   = NODES.dig(DEFAULT_NODE_NAME, 'CPUS') || VAGRANT_CPUS
+        libvirt.memory = NODES.dig(DEFAULT_NODE_NAME, 'MEM') || VAGRANT_MEM
+        # Doesn't boot without this on rockylinux/9
+        libvirt.machine_virtual_size = 10
+
+        NODE_DISKS = NODES.dig(DEFAULT_NODE_NAME, 'DISKS') || {}
+        NODE_NUM_DISKS = NODE_DISKS['NUM'] || NUM_DISKS
+        NODE_DISK_SIZE = NODE_DISKS['SIZE'] || DISK_SIZE
+
+        if !NODE_DISKS.key?('VAGRANT_DISKS') || NODE_DISKS['VAGRANT_DISKS']
+          (1..NODE_NUM_DISKS).each do |i|
+            if i <= 676
+              FIRST_CHAR = (97 + ((i - 1) / 26)).chr
+              SECOND_CHAR = (97 + ((i - 1) % 26)).chr
+              DEVICE_SUFFIX = FIRST_CHAR + SECOND_CHAR
+            else
+              raise "Error: Exceeded number of supported disks (#{NODE_NUM_DISKS}/676))"
+            end
+
+            libvirt.storage :file, \
+              :size => "#{NODE_DISK_SIZE}", \
+              :type => 'qcow2', \
+              :sparse => true, \
+              :device => "vdz#{DEVICE_SUFFIX}", \
+              :serial => "VDZ#{DEVICE_SUFFIX.upcase}"
+          end
+        end
+
+        if NODE_DISKS.key?('CUSTOM')
+          CUSTOM_DISKS = NODE_DISKS['CUSTOM']
+          CUSTOM_DISKS.each do |disk|
+            libvirt.storage :file, \
+              :size => disk['SIZE'] || NODE_DISK_SIZE, \
+              :type => disk['TYPE'] || 'qcow2', \
+              :sparse => disk.key?('SPARSE') ? disk['SPARSE'] : true, \
+              :device => disk['DEVICE'], \
+              :serial => disk['SERIAL'] || disk['DEVICE'].upcase
+          end
+        end
       end
 
-      libvirt.storage :file, \
-        :size => "#{DISK_SIZE}", \
-        :type => 'qcow2', \
-        :sparse => true, \
-        :device => "vdz#{device_suffix}", \
-        :serial => "VDZ#{device_suffix.upcase}"
+      NODE_PROV = NODES.dig(DEFAULT_NODE_NAME, 'PROV')
+      if NODE_PROV.nil? ? VAGRANT_PROV : NODE_PROV
+        node_config.vm.provision "shell", path: "provision.sh"
+      end
+
+      NODE_SH = NODES.dig(DEFAULT_NODE_NAME, 'SH') || VAGRANT_SH
+      if NODE_SH != ''
+        node_config.vm.provision "shell", inline: <<-SHELL
+          if [ -f /vagrant/scratch/#{NODE_SH} ]; then
+            /bin/bash /vagrant/scratch/#{NODE_SH}
+          else
+            echo "ERROR: /vagrant/scratch/#{NODE_SH} not found"
+            exit 1
+          fi
+        SHELL
       end
     end
-
-  if VAGRANT_PROV
-    config.vm.provision "shell", path: "provision.sh"
-  end
-
-  if VAGRANT_SH != ''
-    config.vm.provision "shell", inline: <<-SHELL
-      if [ -f /vagrant/scratch/#{VAGRANT_SH} ]; then
-        /bin/bash /vagrant/scratch/#{VAGRANT_SH}
-      else
-        echo "ERROR: /vagrant/scratch/#{VAGRANT_SH} not found"
-        exit 1
-      fi
-    SHELL
   end
 end