From e561d58562be08b942a5fb9b1eaccb2245a36208 Mon Sep 17 00:00:00 2001
From: Paul Schaub <vanitasvitae@fsfe.org>
Date: Tue, 5 Mar 2024 21:05:34 +0100
Subject: [PATCH] Add tests for PGPSecretKeyExtensions

---
 .../extensions/PGPSecretKeyExtensionsTest.kt  | 75 +++++++++++++++++++
 1 file changed, 75 insertions(+)
 create mode 100644 pgpainless-core/src/test/kotlin/org/pgpainless/bouncycastle/extensions/PGPSecretKeyExtensionsTest.kt

diff --git a/pgpainless-core/src/test/kotlin/org/pgpainless/bouncycastle/extensions/PGPSecretKeyExtensionsTest.kt b/pgpainless-core/src/test/kotlin/org/pgpainless/bouncycastle/extensions/PGPSecretKeyExtensionsTest.kt
new file mode 100644
index 00000000..6edf3fa4
--- /dev/null
+++ b/pgpainless-core/src/test/kotlin/org/pgpainless/bouncycastle/extensions/PGPSecretKeyExtensionsTest.kt
@@ -0,0 +1,75 @@
+// SPDX-FileCopyrightText: 2024 Paul Schaub <vanitasvitae@fsfe.org>
+//
+// SPDX-License-Identifier: Apache-2.0
+
+package org.pgpainless.bouncycastle.extensions
+
+import org.junit.jupiter.api.Assertions.assertEquals
+import org.junit.jupiter.api.Assertions.assertNotNull
+import org.junit.jupiter.api.Test
+import org.junit.jupiter.api.assertDoesNotThrow
+import org.junit.jupiter.api.assertThrows
+import org.pgpainless.algorithm.PublicKeyAlgorithm
+import org.pgpainless.exception.WrongPassphraseException
+import org.pgpainless.key.TestKeys
+import org.pgpainless.util.Passphrase
+
+class PGPSecretKeyExtensionsTest {
+
+    @Test
+    fun `can successfully unlock protected secret key`() {
+        val key = TestKeys.getCryptieSecretKeyRing()
+        val secKey = key.secretKey
+
+        val privKey = assertDoesNotThrow {
+            secKey.unlock(TestKeys.CRYPTIE_PASSPHRASE)
+        }
+        assertNotNull(privKey)
+    }
+
+    @Test
+    fun `cannot unlock protected secret key using empty passphrase`() {
+        val key = TestKeys.getCryptieSecretKeyRing()
+        val secKey = key.secretKey
+
+        assertThrows<WrongPassphraseException> {
+            secKey.unlock(Passphrase.emptyPassphrase())
+        }
+    }
+
+    @Test
+    fun `can successfully unlock unprotected secret key with unprotectedKeys protector`() {
+        val key = TestKeys.getEmilSecretKeyRing()
+        val secKey = key.secretKey
+
+        val privKey = assertDoesNotThrow {
+            secKey.unlock()
+        }
+        assertNotNull(privKey)
+    }
+
+    @Test
+    fun `can successfully unlock unprotected secret key with empty passphrase`() {
+        val key = TestKeys.getEmilSecretKeyRing()
+        val secKey = key.secretKey
+
+        val privKey = assertDoesNotThrow {
+            secKey.unlock(Passphrase.emptyPassphrase())
+        }
+        assertNotNull(privKey)
+    }
+
+    @Test
+    fun `openPgpFingerprint returns fitting fingerprint`() {
+        val key = TestKeys.getEmilSecretKeyRing()
+
+        assertEquals(TestKeys.EMIL_FINGERPRINT, key.openPgpFingerprint)
+        assertEquals(TestKeys.EMIL_FINGERPRINT, key.secretKey.openPgpFingerprint)
+    }
+
+    @Test
+    fun `publicKeyAlgorithm returns fitting algorithm`() {
+        val key = TestKeys.getEmilSecretKeyRing()
+        assertEquals(PublicKeyAlgorithm.ECDSA, key.secretKey.publicKeyAlgorithm)
+    }
+}