From 347510a8bb21acf4e8b511d5028377b65b396cdf Mon Sep 17 00:00:00 2001
From: Sam Yuan <yy19902439@126.com>
Date: Sun, 3 Jan 2021 15:14:33 +0800
Subject: [PATCH] nit fix for #20 (#22)

Signed-off-by: SamYuan1990 <yy19902439@126.com>
---
 test/gmssl_test.go | 36 ++++++++++++++++++++++++++++++++++++
 test/smoke_test.go |  6 +++---
 2 files changed, 39 insertions(+), 3 deletions(-)

diff --git a/test/gmssl_test.go b/test/gmssl_test.go
index 18f85c236..27aeea52a 100644
--- a/test/gmssl_test.go
+++ b/test/gmssl_test.go
@@ -9,6 +9,9 @@
 package main
 
 import (
+	"crypto/ecdsa"
+	"crypto/elliptic"
+	"crypto/rand"
 	"fmt"
 	"testing"
 
@@ -382,3 +385,36 @@ func BenchmarkSM2Verify(b *testing.B) {
 		err = sm2pk.Verify("sm2sign", digest, signature, nil)
 	}
 }
+
+func BenchmarkEcdsaSign(t *testing.B) {
+	t.ReportAllocs()
+	msg := []byte("test")
+	priv, err := ecdsa.GenerateKey(elliptic.P256(), rand.Reader)
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.ResetTimer()
+	for i := 0; i < t.N; i++ {
+		_, _, err := ecdsa.Sign(rand.Reader, priv, msg)
+		if err != nil {
+			t.Fatal(err)
+		}
+	}
+}
+
+func BenchmarkEcdsaVerify(t *testing.B) {
+	t.ReportAllocs()
+	msg := []byte("test")
+	priv, err := ecdsa.GenerateKey(elliptic.P256(), rand.Reader)
+	if err != nil {
+		t.Fatal(err)
+	}
+	r, s, err := ecdsa.Sign(rand.Reader, priv, msg)
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.ResetTimer()
+	for i := 0; i < t.N; i++ {
+		ecdsa.Verify(&priv.PublicKey, msg, r, s)
+	}
+}
diff --git a/test/smoke_test.go b/test/smoke_test.go
index 51ca4a542..15b7acfde 100644
--- a/test/smoke_test.go
+++ b/test/smoke_test.go
@@ -58,11 +58,11 @@ func TestPublicKeyAlgorithms(t *testing.T) {
 	}
 }
 
-/*func TestEngines(t *testing.T) {
-	/* Engines
+func TestEngines(t *testing.T) {
+	/* Engines */
 	fmt.Print("Engines:")
 	engines := gmssl.GetEngineNames()
 	for _, engine := range engines {
 		fmt.Print(" " + engine)
 	}
-}*/
+}