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

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

diff --git a/pgpainless-core/src/test/kotlin/org/pgpainless/bouncycastle/extensions/PGPPublicKeyExtensionsTest.kt b/pgpainless-core/src/test/kotlin/org/pgpainless/bouncycastle/extensions/PGPPublicKeyExtensionsTest.kt
new file mode 100644
index 00000000..aab2bf91
--- /dev/null
+++ b/pgpainless-core/src/test/kotlin/org/pgpainless/bouncycastle/extensions/PGPPublicKeyExtensionsTest.kt
@@ -0,0 +1,60 @@
+// 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.Test
+import org.junit.jupiter.api.assertThrows
+import org.pgpainless.PGPainless
+import org.pgpainless.algorithm.PublicKeyAlgorithm
+import org.pgpainless.key.TestKeys
+import org.pgpainless.key.generation.KeySpec
+import org.pgpainless.key.generation.type.KeyType
+import org.pgpainless.key.generation.type.ecc.EllipticCurve
+import org.pgpainless.key.generation.type.eddsa.EdDSACurve
+
+class PGPPublicKeyExtensionsTest {
+
+    @Test
+    fun `test getCurveName for all ECDSA curves`() {
+        for (curve in EllipticCurve.values()) {
+            val key = PGPainless.buildKeyRing()
+                .setPrimaryKey(KeySpec.getBuilder(KeyType.ECDSA(curve)))
+                .build()
+                .publicKey
+
+            assertEquals(curve.curveName, key.getCurveName())
+        }
+    }
+
+    @Test
+    fun `test getCurveName for legacy EdDSA curves`() {
+        for (curve in EdDSACurve.values()) {
+            val key = PGPainless.buildKeyRing()
+                .setPrimaryKey(KeySpec.getBuilder(KeyType.EDDSA(curve)))
+                .build()
+                .publicKey
+
+            assertEquals(curve.curveName, key.getCurveName())
+        }
+    }
+
+    @Test
+    fun `test getCurveName fails for non-curve keys`() {
+        // RSA
+        val key = TestKeys.getJulietPublicKeyRing()
+        assertEquals(PublicKeyAlgorithm.RSA_GENERAL, key.publicKey.publicKeyAlgorithm)
+
+        assertThrows<IllegalArgumentException> {
+            key.publicKey.getCurveName()
+        }
+    }
+
+    @Test
+    fun `openPgpFingerprint returns fitting fingerprint`() {
+        val key = TestKeys.getEmilSecretKeyRing()
+        assertEquals(TestKeys.EMIL_FINGERPRINT, key.publicKey.openPgpFingerprint)
+    }
+}