64 lines
1.3 KiB
Go
64 lines
1.3 KiB
Go
package password
|
||
|
||
import (
|
||
"testing"
|
||
)
|
||
|
||
func TestRSACrypto_EncryptAndDecrypt(t *testing.T) {
|
||
// 创建 RSACrypto 实例
|
||
crypto, err := NewRSACrypto(2048)
|
||
if err != nil {
|
||
t.Fatalf("创建 RSACrypto 实例失败: %v", err)
|
||
}
|
||
|
||
// 测试加密
|
||
originalData := "securedata"
|
||
encryptedData, err := crypto.Encrypt(originalData)
|
||
if err != nil {
|
||
t.Fatalf("加密失败: %v", err)
|
||
}
|
||
|
||
if encryptedData == "" {
|
||
t.Fatal("加密结果为空")
|
||
}
|
||
t.Log(encryptedData)
|
||
|
||
// 测试解密
|
||
decryptedData, err := crypto.Decrypt(encryptedData)
|
||
if err != nil {
|
||
t.Fatalf("解密失败: %v", err)
|
||
}
|
||
|
||
if decryptedData != originalData {
|
||
t.Fatalf("解密结果不匹配，期望: %s，实际: %s", originalData, decryptedData)
|
||
}
|
||
}
|
||
|
||
func TestRSACrypto_ExportKeys(t *testing.T) {
|
||
// 创建 RSACrypto 实例
|
||
crypto, err := NewRSACrypto(2048)
|
||
if err != nil {
|
||
t.Fatalf("创建 RSACrypto 实例失败: %v", err)
|
||
}
|
||
|
||
// 测试导出私钥
|
||
privateKey, err := crypto.ExportPrivateKey()
|
||
if err != nil {
|
||
t.Fatalf("导出私钥失败: %v", err)
|
||
}
|
||
|
||
if privateKey == "" {
|
||
t.Fatal("导出的私钥为空")
|
||
}
|
||
|
||
// 测试导出公钥
|
||
publicKey, err := crypto.ExportPublicKey()
|
||
if err != nil {
|
||
t.Fatalf("导出公钥失败: %v", err)
|
||
}
|
||
|
||
if publicKey == "" {
|
||
t.Fatal("导出的公钥为空")
|
||
}
|
||
}
|