commit cdab66d5b896347e796c009878f75a0930e28d85 · pyrox.dev/nixpkgs

+1 -1

nixos/tests/all-tests.nix

···

       613
       613
        
         gokapi = runTest ./gokapi.nix;

     

       614
       614
        
         gollum = runTest ./gollum.nix;

     

       615
       615
        
         gonic = runTest ./gonic.nix;

     

       616
       616
       -
         google-oslogin = handleTest ./google-oslogin { };

     

       616
       616
       +
         google-oslogin = runTest ./google-oslogin;

     

       617
       617
        
         gopro-tool = runTest ./gopro-tool.nix;

     

       618
       618
        
         goss = runTest ./goss.nix;

     

       619
       619
        
         gotenberg = runTest ./gotenberg.nix;

+70 -67

nixos/tests/google-oslogin/default.nix

···

       1
       1
       -
       import ../make-test-python.nix (

     

       2
       2
       -
         { pkgs, ... }:

     

       3
       3
       -
         let

     

       4
       4
       -
           inherit (import ./../ssh-keys.nix pkgs)

     

       5
       5
       -
             snakeOilPrivateKey

     

       6
       6
       -
             snakeOilPublicKey

     

       7
       7
       -
             ;

     

       1
       1
       +
       {

     

       2
       2
       +
         lib,

     

       3
       3
       +
         pkgs,

     

       4
       4
       +
         hostPkgs,

     

       5
       5
       +
         ...

     

       6
       6
       +
       }:

     

       7
       7
       +
       let

     

       8
       8
       +
         inherit (import ./../ssh-keys.nix hostPkgs)

     

       9
       9
       +
           snakeOilPrivateKey

     

       10
       10
       +
           snakeOilPublicKey

     

       11
       11
       +
           ;

     

       8
       12
        
       

     

       9
       9
       -
           # don't check host keys or known hosts, use the snakeoil ssh key

     

       10
       10
       -
           ssh-config = builtins.toFile "ssh.conf" ''

     

       11
       11
       -
             UserKnownHostsFile=/dev/null

     

       12
       12
       -
             StrictHostKeyChecking=no

     

       13
       13
       -
             IdentityFile=~/.ssh/id_snakeoil

     

       14
       14
       -
           '';

     

       15
       15
       -
         in

     

       16
       16
       -
         {

     

       17
       17
       -
           name = "google-oslogin";

     

       18
       18
       -
           meta = with pkgs.lib.maintainers; {

     

       19
       19
       -
             maintainers = [ ];

     

       20
       20
       -
           };

     

       13
       13
       +
         # don't check host keys or known hosts, use the snakeoil ssh key

     

       14
       14
       +
         ssh-config = builtins.toFile "ssh.conf" ''

     

       15
       15
       +
           UserKnownHostsFile=/dev/null

     

       16
       16
       +
           StrictHostKeyChecking=no

     

       17
       17
       +
           IdentityFile=~/.ssh/id_snakeoil

     

       18
       18
       +
         '';

     

       19
       19
       +
       in

     

       20
       20
       +
       {

     

       21
       21
       +
         name = "google-oslogin";

     

       22
       22
       +
         meta = with lib.maintainers; {

     

       23
       23
       +
           maintainers = [ ];

     

       24
       24
       +
         };

     

       21
       25
        
       

     

       22
       22
       -
           nodes = {

     

       23
       23
       -
             # the server provides both the the mocked google metadata server and the ssh server

     

       24
       24
       -
             server = (import ./server.nix pkgs);

     

       26
       26
       +
         nodes = {

     

       27
       27
       +
           # the server provides both the the mocked google metadata server and the ssh server

     

       28
       28
       +
           server = ./server.nix;

     

       25
       29
        
       

     

       26
       26
       -
             client = { ... }: { };

     

       27
       27
       -
           };

     

       28
       28
       -
           testScript = ''

     

       29
       29
       -
             MOCKUSER = "mockuser_nixos_org"

     

       30
       30
       -
             MOCKADMIN = "mockadmin_nixos_org"

     

       31
       31
       -
             start_all()

     

       30
       30
       +
           client = { ... }: { };

     

       31
       31
       +
         };

     

       32
       32
       +
         testScript = ''

     

       33
       33
       +
           MOCKUSER = "mockuser_nixos_org"

     

       34
       34
       +
           MOCKADMIN = "mockadmin_nixos_org"

     

       35
       35
       +
           start_all()

     

       32
       36
        
       

     

       33
       33
       -
             server.wait_for_unit("mock-google-metadata.service")

     

       34
       34
       -
             server.wait_for_open_port(80)

     

       37
       37
       +
           server.wait_for_unit("mock-google-metadata.service")

     

       38
       38
       +
           server.wait_for_open_port(80)

     

       35
       39
        
       

     

       36
       36
       -
             # mockserver should return a non-expired ssh key for both mockuser and mockadmin

     

       37
       37
       -
             server.succeed(

     

       38
       38
       -
                 f'${pkgs.google-guest-oslogin}/bin/google_authorized_keys {MOCKUSER} | grep -q "${snakeOilPublicKey}"'

     

       39
       39
       -
             )

     

       40
       40
       -
             server.succeed(

     

       41
       41
       -
                 f'${pkgs.google-guest-oslogin}/bin/google_authorized_keys {MOCKADMIN} | grep -q "${snakeOilPublicKey}"'

     

       42
       42
       -
             )

     

       40
       40
       +
           # mockserver should return a non-expired ssh key for both mockuser and mockadmin

     

       41
       41
       +
           server.succeed(

     

       42
       42
       +
               f'${pkgs.google-guest-oslogin}/bin/google_authorized_keys {MOCKUSER} | grep -q "${snakeOilPublicKey}"'

     

       43
       43
       +
           )

     

       44
       44
       +
           server.succeed(

     

       45
       45
       +
               f'${pkgs.google-guest-oslogin}/bin/google_authorized_keys {MOCKADMIN} | grep -q "${snakeOilPublicKey}"'

     

       46
       46
       +
           )

     

       43
       47
        
       

     

       44
       44
       -
             # install snakeoil ssh key on the client, and provision .ssh/config file

     

       45
       45
       -
             client.succeed("mkdir -p ~/.ssh")

     

       46
       46
       -
             client.succeed(

     

       47
       47
       -
                 "cat ${snakeOilPrivateKey} > ~/.ssh/id_snakeoil"

     

       48
       48
       -
             )

     

       49
       49
       -
             client.succeed("chmod 600 ~/.ssh/id_snakeoil")

     

       50
       50
       -
             client.succeed("cp ${ssh-config} ~/.ssh/config")

     

       48
       48
       +
           # install snakeoil ssh key on the client, and provision .ssh/config file

     

       49
       49
       +
           client.succeed("mkdir -p ~/.ssh")

     

       50
       50
       +
           client.succeed(

     

       51
       51
       +
               "cat ${snakeOilPrivateKey} > ~/.ssh/id_snakeoil"

     

       52
       52
       +
           )

     

       53
       53
       +
           client.succeed("chmod 600 ~/.ssh/id_snakeoil")

     

       54
       54
       +
           client.succeed("cp ${ssh-config} ~/.ssh/config")

     

       51
       55
        
       

     

       52
       52
       -
             client.wait_for_unit("network.target")

     

       53
       53
       -
             server.wait_for_unit("sshd.service")

     

       56
       56
       +
           client.wait_for_unit("network.target")

     

       57
       57
       +
           server.wait_for_unit("sshd.service")

     

       54
       58
        
       

     

       55
       55
       -
             # we should not be able to connect as non-existing user

     

       56
       56
       -
             client.fail("ssh ghost@server 'true'")

     

       59
       59
       +
           # we should not be able to connect as non-existing user

     

       60
       60
       +
           client.fail("ssh ghost@server 'true'")

     

       57
       61
        
       

     

       58
       58
       -
             # we should be able to connect as mockuser

     

       59
       59
       -
             client.succeed(f"ssh {MOCKUSER}@server 'true'")

     

       60
       60
       -
             # but we shouldn't be able to sudo

     

       61
       61
       -
             client.fail(

     

       62
       62
       -
                 f"ssh {MOCKUSER}@server '/run/wrappers/bin/sudo /run/current-system/sw/bin/id' | grep -q 'root'"

     

       63
       63
       -
             )

     

       62
       62
       +
           # we should be able to connect as mockuser

     

       63
       63
       +
           client.succeed(f"ssh {MOCKUSER}@server 'true'")

     

       64
       64
       +
           # but we shouldn't be able to sudo

     

       65
       65
       +
           client.fail(

     

       66
       66
       +
               f"ssh {MOCKUSER}@server '/run/wrappers/bin/sudo /run/current-system/sw/bin/id' | grep -q 'root'"

     

       67
       67
       +
           )

     

       64
       68
        
       

     

       65
       65
       -
             # we should also be able to log in as mockadmin

     

       66
       66
       -
             client.succeed(f"ssh {MOCKADMIN}@server 'true'")

     

       67
       67
       -
             # pam_oslogin_admin.so should now have generated a sudoers file

     

       68
       68
       -
             server.succeed(

     

       69
       69
       -
                 f"find /run/google-sudoers.d | grep -q '/run/google-sudoers.d/{MOCKADMIN}'"

     

       70
       70
       -
             )

     

       69
       69
       +
           # we should also be able to log in as mockadmin

     

       70
       70
       +
           client.succeed(f"ssh {MOCKADMIN}@server 'true'")

     

       71
       71
       +
           # pam_oslogin_admin.so should now have generated a sudoers file

     

       72
       72
       +
           server.succeed(

     

       73
       73
       +
               f"find /run/google-sudoers.d | grep -q '/run/google-sudoers.d/{MOCKADMIN}'"

     

       74
       74
       +
           )

     

       71
       75
        
       

     

       72
       72
       -
             # and we should be able to sudo

     

       73
       73
       -
             client.succeed(

     

       74
       74
       -
                 f"ssh {MOCKADMIN}@server '/run/wrappers/bin/sudo /run/current-system/sw/bin/id' | grep -q 'root'"

     

       75
       75
       -
             )

     

       76
       76
       -
           '';

     

       77
       77
       -
         }

     

       78
       78
       -
       )

     

       76
       76
       +
           # and we should be able to sudo

     

       77
       77
       +
           client.succeed(

     

       78
       78
       +
               f"ssh {MOCKADMIN}@server '/run/wrappers/bin/sudo /run/current-system/sw/bin/id' | grep -q 'root'"

     

       79
       79
       +
           )

     

       80
       80
       +
         '';

     

       81
       81
       +
       }