Update mod and optimize rsa test code

Author: quixote-liu

go.mod

@@ -1,4 +1,4 @@
-module codealg
+module github.com/quixote-liu/codealg
 
 go 1.17
 

rsa.go

@@ -95,6 +95,7 @@ func marshalPrivateKey(key *rsa.PrivateKey) ([]byte, error) {
 	buf := bytes.NewBuffer(nil)
 	stream := x509.MarshalPKCS1PrivateKey(key)
 	err := pem.Encode(buf, &pem.Block{
+		Type:  "RSA PRIVATE KEY",
 		Bytes: stream,
 	})
 	return buf.Bytes(), err
@@ -112,6 +113,7 @@ func marshalPublicKey(key *rsa.PublicKey) ([]byte, error) {
 	buf := bytes.NewBuffer(nil)
 	stream := x509.MarshalPKCS1PublicKey(key)
 	err := pem.Encode(buf, &pem.Block{
+		Type:  "RSA PUBLIC KEY",
 		Bytes: stream,
 	})
 	return buf.Bytes(), err

rsa_test.go

@@ -217,8 +217,10 @@ func TestDecryptPKCS1v15(t *testing.T) {
 }
 
 func TestGenerateKey(t *testing.T) {
-	privateKey, publicKey, err := GenRSAKey(1024)
+	privateKey, publicKey, err := GenRSAKey(2048)
 	assert.Nil(t, err)
 	assert.NotEmpty(t, privateKey)
 	assert.NotEmpty(t, publicKey)
+	// fmt.Println(string(privateKey))
+	// fmt.Println(string(publicKey))
 }