71 lines
		
	
	
		
			2.4 KiB
		
	
	
	
		
			Nix
		
	
	
	
	
	
			
		
		
	
	
			71 lines
		
	
	
		
			2.4 KiB
		
	
	
	
		
			Nix
		
	
	
	
	
	
/* This test checks that
 | 
						|
    - multiple config files can be loaded
 | 
						|
    - the storage backend can be in a file outside the nix store
 | 
						|
      as is required for security (required because while confidentiality is
 | 
						|
      always covered, availability isn't)
 | 
						|
    - the postgres integration works
 | 
						|
 */
 | 
						|
import ./make-test-python.nix ({ pkgs, ... }:
 | 
						|
{
 | 
						|
  name = "vault-postgresql";
 | 
						|
  meta = with pkgs.lib.maintainers; {
 | 
						|
    maintainers = [ lnl7 roberth ];
 | 
						|
  };
 | 
						|
  machine = { lib, pkgs, ... }: {
 | 
						|
    virtualisation.memorySize = 512;
 | 
						|
    environment.systemPackages = [ pkgs.vault ];
 | 
						|
    environment.variables.VAULT_ADDR = "http://127.0.0.1:8200";
 | 
						|
    services.vault.enable = true;
 | 
						|
    services.vault.extraSettingsPaths = [ "/run/vault.hcl" ];
 | 
						|
 | 
						|
    systemd.services.vault = {
 | 
						|
      after = [
 | 
						|
        "postgresql.service"
 | 
						|
      ];
 | 
						|
      # Try for about 10 minutes rather than the default of 5 attempts.
 | 
						|
      serviceConfig.RestartSec = 1;
 | 
						|
      serviceConfig.StartLimitBurst = 600;
 | 
						|
    };
 | 
						|
    # systemd.services.vault.unitConfig.RequiresMountsFor = "/run/keys/";
 | 
						|
 | 
						|
    services.postgresql.enable = true;
 | 
						|
    services.postgresql.initialScript = pkgs.writeText "init.psql" ''
 | 
						|
      CREATE USER vaultuser WITH ENCRYPTED PASSWORD 'thisisthepass';
 | 
						|
      GRANT CONNECT ON DATABASE postgres TO vaultuser;
 | 
						|
 | 
						|
      -- https://www.vaultproject.io/docs/configuration/storage/postgresql
 | 
						|
      CREATE TABLE vault_kv_store (
 | 
						|
        parent_path TEXT COLLATE "C" NOT NULL,
 | 
						|
        path        TEXT COLLATE "C",
 | 
						|
        key         TEXT COLLATE "C",
 | 
						|
        value       BYTEA,
 | 
						|
        CONSTRAINT pkey PRIMARY KEY (path, key)
 | 
						|
      );
 | 
						|
      CREATE INDEX parent_path_idx ON vault_kv_store (parent_path);
 | 
						|
 | 
						|
      GRANT SELECT, INSERT, UPDATE, DELETE ON ALL TABLES IN SCHEMA public TO vaultuser;
 | 
						|
    '';
 | 
						|
  };
 | 
						|
 | 
						|
  testScript =
 | 
						|
    ''
 | 
						|
      secretConfig = """
 | 
						|
          storage "postgresql" {
 | 
						|
            connection_url = "postgres://vaultuser:thisisthepass@localhost/postgres?sslmode=disable"
 | 
						|
          }
 | 
						|
          """
 | 
						|
 | 
						|
      start_all()
 | 
						|
 | 
						|
      machine.wait_for_unit("multi-user.target")
 | 
						|
      machine.succeed("cat >/root/vault.hcl <<EOF\n%s\nEOF\n" % secretConfig)
 | 
						|
      machine.succeed(
 | 
						|
          "install --owner vault --mode 0400 /root/vault.hcl /run/vault.hcl; rm /root/vault.hcl"
 | 
						|
      )
 | 
						|
      machine.wait_for_unit("vault.service")
 | 
						|
      machine.wait_for_open_port(8200)
 | 
						|
      machine.succeed("vault operator init")
 | 
						|
      machine.succeed("vault status | grep Sealed | grep true")
 | 
						|
    '';
 | 
						|
})
 |