99 lines
		
	
	
		
			2.9 KiB
		
	
	
	
		
			Nix
		
	
	
	
	
	
			
		
		
	
	
			99 lines
		
	
	
		
			2.9 KiB
		
	
	
	
		
			Nix
		
	
	
	
	
	
| { system ? builtins.currentSystem,
 | |
|   config ? {},
 | |
|   pkgs ? import ../.. { inherit system config; }
 | |
| }:
 | |
| 
 | |
| with import ../lib/testing-python.nix { inherit system pkgs; };
 | |
| with pkgs.lib;
 | |
| 
 | |
| with import common/ec2.nix { inherit makeTest pkgs; };
 | |
| 
 | |
| let
 | |
|   image = (import ../lib/eval-config.nix {
 | |
|     inherit system;
 | |
|     modules = [
 | |
|       ../maintainers/scripts/openstack/openstack-image.nix
 | |
|       ../modules/testing/test-instrumentation.nix
 | |
|       ../modules/profiles/qemu-guest.nix
 | |
|       {
 | |
|         # Needed by nixos-rebuild due to lack of network access.
 | |
|         system.extraDependencies = with pkgs; [
 | |
|           stdenv
 | |
|         ];
 | |
|       }
 | |
|     ];
 | |
|   }).config.system.build.openstackImage + "/nixos.qcow2";
 | |
| 
 | |
|   sshKeys = import ./ssh-keys.nix pkgs;
 | |
|   snakeOilPrivateKey = sshKeys.snakeOilPrivateKey.text;
 | |
|   snakeOilPrivateKeyFile = pkgs.writeText "private-key" snakeOilPrivateKey;
 | |
|   snakeOilPublicKey = sshKeys.snakeOilPublicKey;
 | |
| 
 | |
| in {
 | |
|   metadata = makeEc2Test {
 | |
|     name = "openstack-ec2-metadata";
 | |
|     inherit image;
 | |
|     sshPublicKey = snakeOilPublicKey;
 | |
|     userData = ''
 | |
|       SSH_HOST_ED25519_KEY_PUB:${snakeOilPublicKey}
 | |
|       SSH_HOST_ED25519_KEY:${replaceStrings ["\n"] ["|"] snakeOilPrivateKey}
 | |
|     '';
 | |
|     script = ''
 | |
|       machine.start()
 | |
|       machine.wait_for_file("/etc/ec2-metadata/user-data")
 | |
|       machine.wait_for_unit("sshd.service")
 | |
| 
 | |
|       machine.succeed("grep unknown /etc/ec2-metadata/ami-manifest-path")
 | |
| 
 | |
|       # We have no keys configured on the client side yet, so this should fail
 | |
|       machine.fail("ssh -o BatchMode=yes localhost exit")
 | |
| 
 | |
|       # Let's install our client private key
 | |
|       machine.succeed("mkdir -p ~/.ssh")
 | |
| 
 | |
|       machine.copy_from_host_via_shell(
 | |
|           "${snakeOilPrivateKeyFile}", "~/.ssh/id_ed25519"
 | |
|       )
 | |
|       machine.succeed("chmod 600 ~/.ssh/id_ed25519")
 | |
| 
 | |
|       # We haven't configured the host key yet, so this should still fail
 | |
|       machine.fail("ssh -o BatchMode=yes localhost exit")
 | |
| 
 | |
|       # Add the host key; ssh should finally succeed
 | |
|       machine.succeed(
 | |
|           "echo localhost,127.0.0.1 ${snakeOilPublicKey} > ~/.ssh/known_hosts"
 | |
|       )
 | |
|       machine.succeed("ssh -o BatchMode=yes localhost exit")
 | |
| 
 | |
|       # Just to make sure resizing is idempotent.
 | |
|       machine.shutdown()
 | |
|       machine.start()
 | |
|       machine.wait_for_file("/etc/ec2-metadata/user-data")
 | |
|     '';
 | |
|   };
 | |
| 
 | |
|   userdata = makeEc2Test {
 | |
|     name = "openstack-ec2-metadata";
 | |
|     inherit image;
 | |
|     sshPublicKey = snakeOilPublicKey;
 | |
|     userData = ''
 | |
|       { pkgs, ... }:
 | |
|       {
 | |
|         imports = [
 | |
|           <nixpkgs/nixos/modules/virtualisation/openstack-config.nix>
 | |
|           <nixpkgs/nixos/modules/testing/test-instrumentation.nix>
 | |
|           <nixpkgs/nixos/modules/profiles/qemu-guest.nix>
 | |
|         ];
 | |
|         environment.etc.testFile = {
 | |
|           text = "whoa";
 | |
|         };
 | |
|       }
 | |
|     '';
 | |
|     script = ''
 | |
|       machine.start()
 | |
|       machine.wait_for_file("/etc/testFile")
 | |
|       assert "whoa" in machine.succeed("cat /etc/testFile")
 | |
|     '';
 | |
|   };
 | |
| }
 | 
