The library does not depend on stdenv, that `stdenv` exposes `lib` is an artifact of the ancient origins of nixpkgs.
		
			
				
	
	
		
			85 lines
		
	
	
		
			2.4 KiB
		
	
	
	
		
			Nix
		
	
	
	
	
	
			
		
		
	
	
			85 lines
		
	
	
		
			2.4 KiB
		
	
	
	
		
			Nix
		
	
	
	
	
	
{ system ? builtins.currentSystem,
 | 
						|
  config ? {},
 | 
						|
  pkgs ? import ../.. { inherit system config; }
 | 
						|
}:
 | 
						|
 | 
						|
with import ../lib/testing-python.nix { inherit system pkgs; };
 | 
						|
with pkgs.lib;
 | 
						|
 | 
						|
let
 | 
						|
  inherit (import ./ssh-keys.nix pkgs)
 | 
						|
    snakeOilPrivateKey snakeOilPublicKey;
 | 
						|
 | 
						|
  metadataDrive = pkgs.stdenv.mkDerivation {
 | 
						|
    name = "metadata";
 | 
						|
    buildCommand = ''
 | 
						|
      mkdir -p $out/iso
 | 
						|
 | 
						|
      cat << EOF > $out/iso/user-data
 | 
						|
      #cloud-config
 | 
						|
      write_files:
 | 
						|
      -   content: |
 | 
						|
                cloudinit
 | 
						|
          path: /tmp/cloudinit-write-file
 | 
						|
 | 
						|
      users:
 | 
						|
        - default
 | 
						|
        - name: nixos
 | 
						|
          ssh_authorized_keys:
 | 
						|
            - "${snakeOilPublicKey}"
 | 
						|
      EOF
 | 
						|
 | 
						|
      cat << EOF > $out/iso/meta-data
 | 
						|
      instance-id: iid-local01
 | 
						|
      local-hostname: "test"
 | 
						|
      public-keys:
 | 
						|
        - "${snakeOilPublicKey}"
 | 
						|
      EOF
 | 
						|
      ${pkgs.cdrkit}/bin/genisoimage -volid cidata -joliet -rock -o $out/metadata.iso $out/iso
 | 
						|
      '';
 | 
						|
  };
 | 
						|
in makeTest {
 | 
						|
  name = "cloud-init";
 | 
						|
  meta = with pkgs.lib.maintainers; {
 | 
						|
    maintainers = [ lewo ];
 | 
						|
  };
 | 
						|
  machine = { ... }:
 | 
						|
  {
 | 
						|
    virtualisation.qemu.options = [ "-cdrom" "${metadataDrive}/metadata.iso" ];
 | 
						|
    services.cloud-init.enable = true;
 | 
						|
    services.openssh.enable = true;
 | 
						|
    networking.hostName = "";
 | 
						|
  };
 | 
						|
  testScript = ''
 | 
						|
    # To wait until cloud-init terminates its run
 | 
						|
    unnamed.wait_for_unit("cloud-final.service")
 | 
						|
 | 
						|
    unnamed.succeed("cat /tmp/cloudinit-write-file | grep -q 'cloudinit'")
 | 
						|
 | 
						|
    # install snakeoil ssh key and provision .ssh/config file
 | 
						|
    unnamed.succeed("mkdir -p ~/.ssh")
 | 
						|
    unnamed.succeed(
 | 
						|
        "cat ${snakeOilPrivateKey} > ~/.ssh/id_snakeoil"
 | 
						|
    )
 | 
						|
    unnamed.succeed("chmod 600 ~/.ssh/id_snakeoil")
 | 
						|
 | 
						|
    unnamed.wait_for_unit("sshd.service")
 | 
						|
 | 
						|
    # we should be able to log in as the root user, as well as the created nixos user
 | 
						|
    unnamed.succeed(
 | 
						|
        "ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no -o IdentityFile=~/.ssh/id_snakeoil root@localhost 'true'"
 | 
						|
    )
 | 
						|
    unnamed.succeed(
 | 
						|
        "ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no -o IdentityFile=~/.ssh/id_snakeoil nixos@localhost 'true'"
 | 
						|
    )
 | 
						|
 | 
						|
    # test changing hostname via cloud-init worked
 | 
						|
    assert (
 | 
						|
        unnamed.succeed(
 | 
						|
            "ssh -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no -o IdentityFile=~/.ssh/id_snakeoil nixos@localhost 'hostname'"
 | 
						|
        ).strip()
 | 
						|
        == "test"
 | 
						|
    )
 | 
						|
  '';
 | 
						|
}
 |