1 files changed, 47 insertions, 0 deletions
diff --git a/prov/src/test/java/org/spongycastle/pqc/jcajce/provider/test/KeyPairGeneratorTest.java b/prov/src/test/java/org/spongycastle/pqc/jcajce/provider/test/KeyPairGeneratorTest.java
new file mode 100644
index 00000000..2d7807d5
--- /dev/null
+++ b/prov/src/test/java/org/spongycastle/pqc/jcajce/provider/test/KeyPairGeneratorTest.java
@@ -0,0 +1,47 @@
+package org.spongycastle.pqc.jcajce.provider.test;
+
+import java.security.KeyFactory;
+import java.security.KeyPair;
+import java.security.KeyPairGenerator;
+import java.security.PrivateKey;
+import java.security.PublicKey;
+import java.security.spec.PKCS8EncodedKeySpec;
+import java.security.spec.X509EncodedKeySpec;
+
+public abstract class KeyPairGeneratorTest
+    extends FlexiTest
+{
+
+    protected KeyPairGenerator kpg;
+
+    protected KeyFactory kf;
+
+    protected final void performKeyPairEncodingTest()
+    {
+        try
+        {
+            KeyPair keyPair = kpg.genKeyPair();
+            PublicKey pubKey = keyPair.getPublic();
+            PrivateKey privKey = keyPair.getPrivate();
+
+            byte[] encPubKey = pubKey.getEncoded();
+            byte[] encPrivKey = privKey.getEncoded();
+
+            X509EncodedKeySpec pubKeySpec = new X509EncodedKeySpec(encPubKey);
+            PKCS8EncodedKeySpec privKeySpec = new PKCS8EncodedKeySpec(
+                encPrivKey);
+
+            PublicKey decPubKey = kf.generatePublic(pubKeySpec);
+            PrivateKey decPrivKey = kf.generatePrivate(privKeySpec);
+
+            assertEquals(pubKey, decPubKey);
+            assertEquals(privKey, decPrivKey);
+        }
+        catch (Exception e)
+        {
+            e.printStackTrace();
+            fail(e);
+        }
+    }
+
+}