From 042c4861c4fe1abdd58d80d17cc654ca63eb5901 Mon Sep 17 00:00:00 2001
From: John Jiang <johnsjiang@tencent.com>
Date: Tue, 31 Oct 2023 10:24:19 +0800
Subject: [PATCH] TKSS-524: Add test for ECKeyPairGenerator on SM2 curve

---
 .../provider/ECKeyPairGeneratorTest.java      | 71 +++++++++++++++++++
 1 file changed, 71 insertions(+)
 create mode 100644 kona-crypto/src/test/java/com/tencent/kona/crypto/provider/ECKeyPairGeneratorTest.java

diff --git a/kona-crypto/src/test/java/com/tencent/kona/crypto/provider/ECKeyPairGeneratorTest.java b/kona-crypto/src/test/java/com/tencent/kona/crypto/provider/ECKeyPairGeneratorTest.java
new file mode 100644
index 00000000..25233438
--- /dev/null
+++ b/kona-crypto/src/test/java/com/tencent/kona/crypto/provider/ECKeyPairGeneratorTest.java
@@ -0,0 +1,71 @@
+/*
+ * Copyright (C) 2023, THL A29 Limited, a Tencent company. All rights reserved.
+ * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
+ *
+ * This code is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License version 2 only, as
+ * published by the Free Software Foundation.
+ *
+ * This code is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
+ * version 2 for more details (a copy is included in the LICENSE file that
+ * accompanied this code).
+ *
+ * You should have received a copy of the GNU General Public License version
+ * 2 along with this work; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+
+package com.tencent.kona.crypto.provider;
+
+import com.tencent.kona.crypto.TestUtils;
+import com.tencent.kona.crypto.spec.SM2ParameterSpec;
+import com.tencent.kona.sun.security.ec.ECOperator;
+import org.junit.jupiter.api.Assertions;
+import org.junit.jupiter.api.BeforeAll;
+import org.junit.jupiter.api.Test;
+
+import java.security.KeyPair;
+import java.security.KeyPairGenerator;
+import java.security.interfaces.ECPrivateKey;
+import java.security.interfaces.ECPublicKey;
+import java.security.spec.AlgorithmParameterSpec;
+import java.security.spec.ECGenParameterSpec;
+import java.security.spec.ECPoint;
+
+import static com.tencent.kona.crypto.TestUtils.PROVIDER;
+
+/**
+ * The test for EC key pair generation on SM2 curve.
+ */
+public class ECKeyPairGeneratorTest {
+
+    @BeforeAll
+    public static void setup() {
+        TestUtils.addProviders();
+    }
+
+    @Test
+    public void testKeyPairGenWithParams() throws Exception {
+        testKeyPairGen(SM2ParameterSpec.instance());
+    }
+
+    @Test
+    public void testKeyPairGenWithName() throws Exception {
+        testKeyPairGen(new ECGenParameterSpec("curveSM2"));
+    }
+
+    private void testKeyPairGen(AlgorithmParameterSpec spec) throws Exception {
+        KeyPairGenerator keyPairGen
+                = KeyPairGenerator.getInstance("EC", PROVIDER);
+        keyPairGen.initialize(spec);
+        KeyPair keyPair = keyPairGen.generateKeyPair();
+        ECPublicKey pubKey = (ECPublicKey) keyPair.getPublic();
+        ECPrivateKey priKey = (ECPrivateKey) keyPair.getPrivate();
+
+        ECPoint pubPoint = ECOperator.SM2.multiply(
+                SM2ParameterSpec.GENERATOR, priKey.getS());
+        Assertions.assertEquals(pubKey.getW(), pubPoint);
+    }
+}