commit cdd4d4fadf0d2d0b8ce9c92fabf2a808d8a97e07 · pyrox.dev/nixpkgs

+24 -20
nixos/tests/wireguard/generated.nix
···

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

     

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

     

       2
       2
        
         name = "wireguard-generated";

     

       3
       3
        
         meta = with pkgs.stdenv.lib.maintainers; {

     

       4
       4
        
           maintainers = [ ma27 grahamc ];

     
···

       28
       28
        
         };

     

       29
       29
        
       

     

       30
       30
        
         testScript = ''

     

       31
       31
       -
           startAll;

     

       31
       31
       +
           start_all()

     

       32
       32
        
       

     

       33
       33
       -
           $peer1->waitForUnit("wireguard-wg0.service");

     

       34
       34
       -
           $peer2->waitForUnit("wireguard-wg0.service");

     

       33
       33
       +
           peer1.wait_for_unit("wireguard-wg0.service")

     

       34
       34
       +
           peer2.wait_for_unit("wireguard-wg0.service")

     

       35
       35
        
       

     

       36
       36
       -
           my ($retcode, $peer1pubkey) = $peer1->execute("wg pubkey < /etc/wireguard/private");

     

       37
       37
       -
           $peer1pubkey =~ s/\s+$//;

     

       38
       38
       -
           if ($retcode != 0) {

     

       39
       39
       -
             die "Could not read public key from peer1";

     

       40
       40
       -
           }

     

       36
       36
       +
           retcode, peer1pubkey = peer1.execute("wg pubkey < /etc/wireguard/private")

     

       37
       37
       +
           if retcode != 0:

     

       38
       38
       +
               raise Exception("Could not read public key from peer1")

     

       41
       39
        
       

     

       42
       42
       -
           my ($retcode, $peer2pubkey) = $peer2->execute("wg pubkey < /etc/wireguard/private");

     

       43
       43
       -
           $peer2pubkey =~ s/\s+$//;

     

       44
       44
       -
           if ($retcode != 0) {

     

       45
       45
       -
             die "Could not read public key from peer2";

     

       46
       46
       -
           }

     

       40
       40
       +
           retcode, peer2pubkey = peer2.execute("wg pubkey < /etc/wireguard/private")

     

       41
       41
       +
           if retcode != 0:

     

       42
       42
       +
               raise Exception("Could not read public key from peer2")

     

       47
       43
        
       

     

       48
       48
       -
           $peer1->succeed("wg set wg0 peer $peer2pubkey allowed-ips 10.10.10.2/32 endpoint 192.168.1.2:12345 persistent-keepalive 1");

     

       49
       49
       -
           $peer1->succeed("ip route replace 10.10.10.2/32 dev wg0 table main");

     

       44
       44
       +
           peer1.succeed(

     

       45
       45
       +
               "wg set wg0 peer {} allowed-ips 10.10.10.2/32 endpoint 192.168.1.2:12345 persistent-keepalive 1".format(

     

       46
       46
       +
                   peer2pubkey.strip()

     

       47
       47
       +
               )

     

       48
       48
       +
           )

     

       49
       49
       +
           peer1.succeed("ip route replace 10.10.10.2/32 dev wg0 table main")

     

       50
       50
        
       

     

       51
       51
       -
           $peer2->succeed("wg set wg0 peer $peer1pubkey allowed-ips 10.10.10.1/32 endpoint 192.168.1.1:12345 persistent-keepalive 1");

     

       52
       52
       -
           $peer2->succeed("ip route replace 10.10.10.1/32 dev wg0 table main");

     

       51
       51
       +
           peer2.succeed(

     

       52
       52
       +
               "wg set wg0 peer {} allowed-ips 10.10.10.1/32 endpoint 192.168.1.1:12345 persistent-keepalive 1".format(

     

       53
       53
       +
                   peer1pubkey.strip()

     

       54
       54
       +
               )

     

       55
       55
       +
           )

     

       56
       56
       +
           peer2.succeed("ip route replace 10.10.10.1/32 dev wg0 table main")

     

       53
       57
        
       

     

       54
       54
       -
           $peer1->succeed("ping -c1 10.10.10.2");

     

       55
       55
       -
           $peer2->succeed("ping -c1 10.10.10.1");

     

       58
       58
       +
           peer1.succeed("ping -c1 10.10.10.2")

     

       59
       59
       +
           peer2.succeed("ping -c1 10.10.10.1")

     

       56
       60
        
         '';

     

       57
       61
        
       })