test: update test case

Author: leeqvip

rsa_test.go

@@ -5,13 +5,47 @@ import (
 	"crypto"
 	"crypto/rsa"
 	"crypto/x509"
-	"encoding/pem"
 	"encoding/base64"
+	"encoding/pem"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
 )
 
+func TestRSAGenerateKey(t *testing.T) {
+	priBuf := bytes.NewBuffer(nil)
+	err := RSAGenerateKey(2048, priBuf)
+	assert.NoError(t, err)
+	t.Logf("private key: %s\n", priBuf.Bytes())
+
+	block, _ := pem.Decode(priBuf.Bytes())
+	assert.NotNil(t, block, "Failed to decode private key")
+	assert.Equal(t, "RSA PRIVATE KEY", block.Type, "Invalid key type")
+
+	_, err = x509.ParsePKCS1PrivateKey(block.Bytes)
+	assert.NoError(t, err, "Failed to parse private key")
+}
+
+func TestRSAGeneratePublicKey(t *testing.T) {
+	priBuf := bytes.NewBuffer(nil)
+	err := RSAGenerateKey(2048, priBuf)
+	assert.NoError(t, err)
+
+	pubBuf := bytes.NewBuffer(nil)
+	err = RSAGeneratePublicKey(priBuf.Bytes(), pubBuf)
+	assert.NoError(t, err)
+	t.Logf("public key: %s\n", pubBuf.Bytes())
+
+	block, _ := pem.Decode(pubBuf.Bytes())
+	assert.NotNil(t, block, "Failed to decode public key")
+	assert.Equal(t, "RSA PUBLIC KEY", block.Type, "Invalid key type")
+
+	pubKey, err := x509.ParsePKIXPublicKey(block.Bytes)
+	assert.NoError(t, err, "Failed to parse public key")
+	_, ok := pubKey.(*rsa.PublicKey)
+	assert.True(t, ok, "Key is not an RSA public key")
+}
+
 func TestRSAEncrypt(t *testing.T) {
 	priBuf := bytes.NewBuffer(nil)
 	err := RSAGenerateKey(2048, priBuf)
@@ -55,35 +89,3 @@ func TestRSASign(t *testing.T) {
 	err = RSAVerify(src, sign, pubBuf.Bytes(), crypto.SHA256)
 	assert.NoError(t, err)
 }
-
-func TestRSAGenerateKey(t *testing.T) {
-	priBuf := bytes.NewBuffer(nil)
-	err := RSAGenerateKey(2048, priBuf)
-	assert.NoError(t, err)
-
-	block, _ := pem.Decode(priBuf.Bytes())
-	assert.NotNil(t, block, "Failed to decode private key")
-	assert.Equal(t, "RSA PRIVATE KEY", block.Type, "Invalid key type")
-
-	_, err = x509.ParsePKCS1PrivateKey(block.Bytes)
-	assert.NoError(t, err, "Failed to parse private key")
-}
-
-func TestRSAGeneratePublicKey(t *testing.T) {
-	priBuf := bytes.NewBuffer(nil)
-	err := RSAGenerateKey(2048, priBuf)
-	assert.NoError(t, err)
-
-	pubBuf := bytes.NewBuffer(nil)
-	err = RSAGeneratePublicKey(priBuf.Bytes(), pubBuf)
-	assert.NoError(t, err)
-
-	block, _ := pem.Decode(pubBuf.Bytes())
-	assert.NotNil(t, block, "Failed to decode public key")
-	assert.Equal(t, "RSA PUBLIC KEY", block.Type, "Invalid key type")
-
-	pubKey, err := x509.ParsePKIXPublicKey(block.Bytes)
-	assert.NoError(t, err, "Failed to parse public key")
-	_, ok := pubKey.(*rsa.PublicKey)
-	assert.True(t, ok, "Key is not an RSA public key")
-}