commit c8faa482fa3eae065f011505c4d244d5511171c0 · pyrox.dev/nixpkgs

+18 -5
nixos/tests/slurm.nix
···

       1
       1
        
       import ./make-test.nix ({ pkgs, ... }:

     

       2
       2
        
       let mungekey = "mungeverryweakkeybuteasytointegratoinatest";

     

       3
       3
        
           slurmconfig = {

     

       4
       4
       -
             client.enable = true;

     

       5
       4
        
             controlMachine = "control";

     

       6
       5
        
             nodeName = ''

     

       7
       6
        
               control

     
···

       20
       19
        
               # TODO slrumd port and slurmctld port should be configurations and

     

       21
       20
        
               # automatically allowed by the  firewall.

     

       22
       21
        
               networking.firewall.enable = false;

     

       23
       23
       -
               services.slurm = slurmconfig;

     

       22
       22
       +
               services.slurm = {

     

       23
       23
       +
                 client.enable = true;

     

       24
       24
       +
               } // slurmconfig;

     

       24
       25
        
             };

     

       25
       26
        
           in {

     

       27
       27
       +
       

     

       26
       28
        
           control =

     

       27
       29
        
             { config, pkgs, ...}:

     

       28
       30
        
             {

     
···

       31
       33
        
                 server.enable = true;

     

       32
       34
        
               } // slurmconfig;

     

       33
       35
        
             };

     

       36
       36
       +
       

     

       37
       37
       +
           submit =

     

       38
       38
       +
             { config, pkgs, ...}:

     

       39
       39
       +
             {

     

       40
       40
       +
               networking.firewall.enable = false;

     

       41
       41
       +
               services.slurm = {

     

       42
       42
       +
                 enableStools = true;

     

       43
       43
       +
               } // slurmconfig;

     

       44
       44
       +
             };

     

       45
       45
       +
       

     

       34
       46
        
           node1 = computeNode;

     

       35
       47
        
           node2 = computeNode;

     

       36
       48
        
           node3 = computeNode;

     

       37
       49
        
         };

     

       50
       50
       +
       

     

       38
       51
        
       

     

       39
       52
        
         testScript =

     

       40
       53
        
         ''

     

       41
       54
        
         startAll;

     

       42
       55
        
       

     

       43
       56
        
         # Set up authentification across the cluster

     

       44
       44
       -
         foreach my $node (($control,$node1,$node2,$node3))

     

       57
       57
       +
         foreach my $node (($submit,$control,$node1,$node2,$node3))

     

       45
       58
        
         {

     

       46
       59
        
           $node->waitForUnit("default.target");

     

       47
       60
        
       

     
···

       60
       73
        
         };

     

       61
       74
        
       

     

       62
       75
        
         subtest "can_start_slurmd", sub {

     

       63
       63
       -
           foreach my $node (($control,$node1,$node2,$node3))

     

       76
       76
       +
           foreach my $node (($node1,$node2,$node3))

     

       64
       77
        
           {

     

       65
       78
        
             $node->succeed("systemctl restart slurmd.service");

     

       66
       79
        
             $node->waitForUnit("slurmd");

     
···

       72
       85
        
         subtest "run_distributed_command", sub {

     

       73
       86
        
           # Run `hostname` on 3 nodes of the partition (so on all the 3 nodes).

     

       74
       87
        
           # The output must contain the 3 different names

     

       75
       75
       -
           $control->succeed("srun -N 3 hostname | sort | uniq | wc -l | xargs test 3 -eq");

     

       88
       88
       +
           $submit->succeed("srun -N 3 hostname | sort | uniq | wc -l | xargs test 3 -eq");

     

       76
       89
        
         };

     

       77
       90
        
         '';

     

       78
       91
        
       })