commit 28d41fc5912cc0526e9dae9388f9e7cadf9acb34 · pyrox.dev/nixpkgs

+30 -23
nixos/tests/systemd-repart.nix
···

       9
       9
        
       

     

       10
       10
        
       let

     

       11
       11
        
         # A testScript fragment that prepares a disk with some empty, unpartitioned

     

       12
       12
       -
         # space. and uses it to boot the test with. Takes a single argument `machine`

     

       13
       13
       -
         # from which the diskImage is extracted.

     

       14
       14
       -
         useDiskImage = machine: ''

     

       15
       15
       -
           import os

     

       16
       16
       -
           import shutil

     

       17
       17
       -
           import subprocess

     

       18
       18
       -
           import tempfile

     

       12
       12
       +
         # space. and uses it to boot the test with.

     

       13
       13
       +
         # Takes two arguments, `machine` from which the diskImage is extracted,

     

       14
       14
       +
         # as well an optional `sizeDiff` (defaulting to +32M), describing how should

     

       15
       15
       +
         # be resized.

     

       16
       16
       +
         useDiskImage =

     

       17
       17
       +
           {

     

       18
       18
       +
             machine,

     

       19
       19
       +
             sizeDiff ? "+32M",

     

       20
       20
       +
           }:

     

       21
       21
       +
           ''

     

       22
       22
       +
             import os

     

       23
       23
       +
             import shutil

     

       24
       24
       +
             import subprocess

     

       25
       25
       +
             import tempfile

     

       19
       26
        
       

     

       20
       20
       -
           tmp_disk_image = tempfile.NamedTemporaryFile()

     

       27
       27
       +
             tmp_disk_image = tempfile.NamedTemporaryFile()

     

       21
       28
        
       

     

       22
       22
       -
           shutil.copyfile("${machine.system.build.diskImage}/nixos.img", tmp_disk_image.name)

     

       29
       29
       +
             shutil.copyfile("${machine.system.build.diskImage}/nixos.img", tmp_disk_image.name)

     

       23
       30
        
       

     

       24
       24
       -
           subprocess.run([

     

       25
       25
       -
             "${machine.virtualisation.qemu.package}/bin/qemu-img",

     

       26
       26
       -
             "resize",

     

       27
       27
       -
             "-f",

     

       28
       28
       -
             "raw",

     

       29
       29
       -
             tmp_disk_image.name,

     

       30
       30
       -
             "+32M",

     

       31
       31
       -
           ])

     

       31
       31
       +
             subprocess.run([

     

       32
       32
       +
               "${machine.virtualisation.qemu.package}/bin/qemu-img",

     

       33
       33
       +
               "resize",

     

       34
       34
       +
               "-f",

     

       35
       35
       +
               "raw",

     

       36
       36
       +
               tmp_disk_image.name,

     

       37
       37
       +
               "${sizeDiff}",

     

       38
       38
       +
             ])

     

       32
       39
        
       

     

       33
       33
       -
           # Set NIX_DISK_IMAGE so that the qemu script finds the right disk image.

     

       34
       34
       -
           os.environ['NIX_DISK_IMAGE'] = tmp_disk_image.name

     

       35
       35
       -
         '';

     

       40
       40
       +
             # Set NIX_DISK_IMAGE so that the qemu script finds the right disk image.

     

       41
       41
       +
             os.environ['NIX_DISK_IMAGE'] = tmp_disk_image.name

     

       42
       42
       +
           '';

     

       36
       43
        
       

     

       37
       44
        
         common =

     

       38
       45
        
           {

     
···

       98
       105
        
           testScript =

     

       99
       106
        
             { nodes, ... }:

     

       100
       107
        
             ''

     

       101
       101
       -
               ${useDiskImage nodes.machine}

     

       108
       108
       +
               ${useDiskImage { inherit (nodes) machine; }}

     

       102
       109
        
       

     

       103
       110
        
               machine.start()

     

       104
       111
        
               machine.wait_for_unit("multi-user.target")

     
···

       128
       135
        
           testScript =

     

       129
       136
        
             { nodes, ... }:

     

       130
       137
        
             ''

     

       131
       131
       -
               ${useDiskImage nodes.machine}

     

       138
       138
       +
               ${useDiskImage { inherit (nodes) machine; }}

     

       132
       139
        
       

     

       133
       140
        
               machine.start()

     

       134
       141
        
               machine.wait_for_unit("multi-user.target")

     
···

       196
       203
        
           testScript =

     

       197
       204
        
             { nodes, ... }:

     

       198
       205
        
             ''

     

       199
       199
       -
               ${useDiskImage nodes.machine}

     

       206
       206
       +
               ${useDiskImage { inherit (nodes) machine; }}

     

       200
       207
        
       

     

       201
       208
        
               machine.start()

     

       202
       209
        
               machine.wait_for_unit("multi-user.target")