From 5a6b08caab513bace301f53f4dd2c9d5a17420fc Mon Sep 17 00:00:00 2001
From: Seth Vargo <seth@sethvargo.com>
Date: Mon, 4 Sep 2017 23:55:58 -0400
Subject: [PATCH] Add helper for decrypting via PGP in tests

---
 command/pgp_test.go | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/command/pgp_test.go b/command/pgp_test.go
index c368e3113..4cfda985b 100644
--- a/command/pgp_test.go
+++ b/command/pgp_test.go
@@ -62,6 +62,35 @@ func getPubKeyFiles(t *testing.T) (string, []string, error) {
 	return tempDir, pubFiles, nil
 }
 
+func testPGPDecrypt(tb testing.TB, privKey, enc string) string {
+	tb.Helper()
+
+	privKeyBytes, err := base64.StdEncoding.DecodeString(privKey)
+	if err != nil {
+		tb.Fatal(err)
+	}
+
+	ptBuf := bytes.NewBuffer(nil)
+	entity, err := openpgp.ReadEntity(packet.NewReader(bytes.NewBuffer(privKeyBytes)))
+	if err != nil {
+		tb.Fatal(err)
+	}
+
+	var rootBytes []byte
+	rootBytes, err = base64.StdEncoding.DecodeString(enc)
+	if err != nil {
+		tb.Fatal(err)
+	}
+
+	entityList := &openpgp.EntityList{entity}
+	md, err := openpgp.ReadMessage(bytes.NewBuffer(rootBytes), entityList, nil, nil)
+	if err != nil {
+		tb.Fatal(err)
+	}
+	ptBuf.ReadFrom(md.UnverifiedBody)
+	return ptBuf.String()
+}
+
 func parseDecryptAndTestUnsealKeys(t *testing.T,
 	input, rootToken string,
 	fingerprints bool,