nixos/tests/ec2.nix

   1 { system ? builtins.currentSystem,
   2   config ? {},
   3   pkgs ? import ../.. { inherit system config; }
   4 }:
   5
   6 with import ../lib/testing-python.nix { inherit system pkgs; };
   7 with pkgs.lib;
   8
   9 with import common/ec2.nix { inherit makeTest pkgs; };
  10
  11 let
  12   imageCfg = (import ../lib/eval-config.nix {
  13     inherit system;
  14     modules = [
  15       ../maintainers/scripts/ec2/amazon-image.nix
  16       ../modules/testing/test-instrumentation.nix
  17       ../modules/profiles/qemu-guest.nix
  18       {
  19         # Hack to make the partition resizing work in QEMU.
  20         boot.initrd.postDeviceCommands = mkBefore ''
  21           ln -s vda /dev/xvda
  22           ln -s vda1 /dev/xvda1
  23         '';
  24
  25         amazonImage.format = "qcow2";
  26
  27         # In a NixOS test the serial console is occupied by the "backdoor"
  28         # (see testing/test-instrumentation.nix) and is incompatible with
  29         # the configuration in virtualisation/amazon-image.nix.
  30         systemd.services."serial-getty@ttyS0".enable = mkForce false;
  31
  32         # Needed by nixos-rebuild due to the lack of network
  33         # access. Determined by trial and error.
  34         system.extraDependencies = with pkgs; ( [
  35           # Needed for a nixos-rebuild.
  36           busybox
  37           cloud-utils
  38           desktop-file-utils
  39           libxslt.bin
  40           mkinitcpio-nfs-utils
  41           stdenv
  42           stdenvNoCC
  43           texinfo
  44           unionfs-fuse
  45           xorg.lndir
  46
  47           # These are used in the configure-from-userdata tests
  48           # for EC2. Httpd and valgrind are requested by the
  49           # configuration.
  50           apacheHttpd
  51           apacheHttpd.doc
  52           apacheHttpd.man
  53           valgrind.doc
  54         ]);
  55       }
  56     ];
  57   }).config;
  58   image = "${imageCfg.system.build.amazonImage}/${imageCfg.amazonImage.name}.qcow2";
  59
  60   sshKeys = import ./ssh-keys.nix pkgs;
  61   snakeOilPrivateKey = sshKeys.snakeOilPrivateKey.text;
  62   snakeOilPrivateKeyFile = pkgs.writeText "private-key" snakeOilPrivateKey;
  63   snakeOilPublicKey = sshKeys.snakeOilPublicKey;
  64
  65 in {
  66   boot-ec2-nixops = makeEc2Test {
  67     name         = "nixops-userdata";
  68     meta.timeout = 600;
  69     inherit image;
  70     sshPublicKey = snakeOilPublicKey; # That's right folks! My user's key is also the host key!
  71
  72     userData = ''
  73       SSH_HOST_ED25519_KEY_PUB:${snakeOilPublicKey}
  74       SSH_HOST_ED25519_KEY:${replaceStrings ["\n"] ["|"] snakeOilPrivateKey}
  75     '';
  76     script = ''
  77       machine.start()
  78       machine.wait_for_file("/etc/ec2-metadata/user-data")
  79       machine.wait_for_unit("sshd.service")
  80
  81       machine.succeed("grep unknown /etc/ec2-metadata/ami-manifest-path")
  82
  83       # We have no keys configured on the client side yet, so this should fail
  84       machine.fail("ssh -o BatchMode=yes localhost exit")
  85
  86       # Let's install our client private key
  87       machine.succeed("mkdir -p ~/.ssh")
  88
  89       machine.copy_from_host_via_shell(
  90           "${snakeOilPrivateKeyFile}", "~/.ssh/id_ed25519"
  91       )
  92       machine.succeed("chmod 600 ~/.ssh/id_ed25519")
  93
  94       # We haven't configured the host key yet, so this should still fail
  95       machine.fail("ssh -o BatchMode=yes localhost exit")
  96
  97       # Add the host key; ssh should finally succeed
  98       machine.succeed(
  99           "echo localhost,127.0.0.1 ${snakeOilPublicKey} > ~/.ssh/known_hosts"
 100       )
 101       machine.succeed("ssh -o BatchMode=yes localhost exit", timeout=120)
 102
 103       # Test whether the root disk was resized.
 104       blocks, block_size = map(int, machine.succeed("stat -c %b:%S -f /").split(":"))
 105       GB = 1024 ** 3
 106       assert 9.7 * GB <= blocks * block_size <= 10 * GB
 107
 108       # Just to make sure resizing is idempotent.
 109       machine.shutdown()
 110       machine.start()
 111       machine.wait_for_file("/etc/ec2-metadata/user-data")
 112     '';
 113   };
 114
 115   boot-ec2-config = makeEc2Test {
 116     name         = "config-userdata";
 117     meta.broken = true; # amazon-init wants to download from the internet while building the system
 118     inherit image;
 119     sshPublicKey = snakeOilPublicKey;
 120
 121     # ### https://nixos.org/channels/nixos-unstable nixos
 122     userData = ''
 123       { pkgs, ... }:
 124
 125       {
 126         imports = [
 127           <nixpkgs/nixos/modules/virtualisation/amazon-image.nix>
 128           <nixpkgs/nixos/modules/testing/test-instrumentation.nix>
 129           <nixpkgs/nixos/modules/profiles/qemu-guest.nix>
 130         ];
 131         environment.etc.testFile = {
 132           text = "whoa";
 133         };
 134
 135         networking.hostName = "ec2-test-vm"; # required by services.httpd
 136
 137         services.httpd = {
 138           enable = true;
 139           adminAddr = "test@example.org";
 140           virtualHosts.localhost.documentRoot = "''${pkgs.valgrind.doc}/share/doc/valgrind/html";
 141         };
 142         networking.firewall.allowedTCPPorts = [ 80 ];
 143       }
 144     '';
 145     script = ''
 146       machine.start()
 147
 148       # amazon-init must succeed. if it fails, make the test fail
 149       # immediately instead of timing out in wait_for_file.
 150       machine.wait_for_unit("amazon-init.service")
 151
 152       machine.wait_for_file("/etc/testFile")
 153       assert "whoa" in machine.succeed("cat /etc/testFile")
 154
 155       machine.wait_for_unit("httpd.service")
 156       assert "Valgrind" in machine.succeed("curl http://localhost")
 157     '';
 158   };
 159 }