From b4c9115365ecbdac2127f7b68d8b46ca1975f1da Mon Sep 17 00:00:00 2001
From: Harsh Shandilya <me@msfjarvis.dev>
Date: Sat, 29 Oct 2022 04:36:51 +0530
Subject: [PATCH] feat(crypto-pgpainless): add a test for ASCII output

---
 .../crypto/PGPainlessCryptoHandlerTest.kt      | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/crypto-pgpainless/src/test/kotlin/app/passwordstore/crypto/PGPainlessCryptoHandlerTest.kt b/crypto-pgpainless/src/test/kotlin/app/passwordstore/crypto/PGPainlessCryptoHandlerTest.kt
index 386ca72b..8e65cc83 100644
--- a/crypto-pgpainless/src/test/kotlin/app/passwordstore/crypto/PGPainlessCryptoHandlerTest.kt
+++ b/crypto-pgpainless/src/test/kotlin/app/passwordstore/crypto/PGPainlessCryptoHandlerTest.kt
@@ -13,6 +13,7 @@ import com.google.testing.junit.testparameterinjector.TestParameter
 import com.google.testing.junit.testparameterinjector.TestParameterInjector
 import java.io.ByteArrayOutputStream
 import kotlin.test.Test
+import kotlin.test.assertContains
 import kotlin.test.assertEquals
 import kotlin.test.assertFalse
 import kotlin.test.assertIs
@@ -81,6 +82,23 @@ class PGPainlessCryptoHandlerTest {
     assertIs<IncorrectPassphraseException>(result.getError())
   }
 
+  @Test
+  fun encryptAsciiArmored() {
+    val ciphertextStream = ByteArrayOutputStream()
+    val encryptRes =
+      cryptoHandler.encrypt(
+        encryptionKey.keySet,
+        CryptoConstants.PLAIN_TEXT.byteInputStream(Charsets.UTF_8),
+        ciphertextStream,
+        PGPEncryptOptions.Builder().withAsciiArmor(true).build(),
+      )
+    assertIs<Ok<Unit>>(encryptRes)
+    val ciphertext = ciphertextStream.toString(Charsets.UTF_8)
+    assertContains(ciphertext, "Version: PGPainless")
+    assertContains(ciphertext, "-----BEGIN PGP MESSAGE-----")
+    assertContains(ciphertext, "-----END PGP MESSAGE-----")
+  }
+
   @Test
   fun canHandleFiltersFormats() {
     assertFalse { cryptoHandler.canHandle("example.com") }