106 lines
2.2 KiB
Go
106 lines
2.2 KiB
Go
|
package crypto
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"testing"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func TestPasswordHashAndVerify(t *testing.T) {
|
|||
|
|
tests := []struct {
|
|||
|
|
name string
|
|||
|
|
password string
|
|||
|
|
cost []int
|
|||
|
|
}{
|
|||
|
|
{
|
|||
|
|
name: "普通密码-默认复杂度",
|
|||
|
|
password: "password123",
|
|||
|
|
cost: []int{},
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "普通密码-低复杂度",
|
|||
|
|
password: "password123",
|
|||
|
|
cost: []int{4},
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "空密码",
|
|||
|
|
password: "",
|
|||
|
|
cost: []int{},
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "中文密码",
|
|||
|
|
password: "密码123",
|
|||
|
|
cost: []int{},
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "特殊字符密码",
|
|||
|
|
password: "p@$$w0rd!#*&",
|
|||
|
|
cost: []int{},
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "我的密码",
|
|||
|
|
password: "aA2021.12.31",
|
|||
|
|
cost: []int{10},
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for _, tt := range tests {
|
|||
|
|
t.Run(tt.name, func(t *testing.T) {
|
|||
|
|
// 1. 测试加密
|
|||
|
|
hash, err := PasswordHash(tt.password, tt.cost...)
|
|||
|
|
if err != nil {
|
|||
|
|
t.Errorf("PasswordHash() error = %v", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
if hash == "" {
|
|||
|
|
t.Error("PasswordHash() 返回空哈希值")
|
|||
|
|
}
|
|||
|
|
if hash == tt.password {
|
|||
|
|
t.Error("PasswordHash() 返回的哈希值与明文密码相同")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 2. 测试验证 - 正确密码
|
|||
|
|
if !PasswordVerify(tt.password, hash) {
|
|||
|
|
t.Error("PasswordVerify() 无法验证正确的密码")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 3. 测试验证 - 错误密码
|
|||
|
|
if PasswordVerify(tt.password+"wrong", hash) {
|
|||
|
|
t.Error("PasswordVerify() 错误地验证了错误的密码")
|
|||
|
|
}
|
|||
|
|
t.Logf("用户名: %s, 密码: %s, 哈希值: %s", tt.name, tt.password, hash)
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func TestPasswordHashWithInvalidCost(t *testing.T) {
|
|||
|
|
// bcrypt库的cost范围是4-31，超出范围会返回错误
|
|||
|
|
password := "testpassword"
|
|||
|
|
|
|||
|
|
// 测试过低的cost
|
|||
|
|
_, err := PasswordHash(password, 2)
|
|||
|
|
if err == nil {
|
|||
|
|
t.Error("PasswordHash() 应该对过低的cost返回错误")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 测试过高的cost
|
|||
|
|
_, err = PasswordHash(password, 32)
|
|||
|
|
if err == nil {
|
|||
|
|
t.Error("PasswordHash() 应该对过高的cost返回错误")
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func BenchmarkPasswordHash(b *testing.B) {
|
|||
|
|
password := "benchmark-password"
|
|||
|
|
for i := 0; i < b.N; i++ {
|
|||
|
|
_, _ = PasswordHash(password)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func BenchmarkPasswordVerify(b *testing.B) {
|
|||
|
|
password := "benchmark-password"
|
|||
|
|
hash, _ := PasswordHash(password)
|
|||
|
|
b.ResetTimer()
|
|||
|
|
for i := 0; i < b.N; i++ {
|
|||
|
|
_ = PasswordVerify(password, hash)
|
|||
|
|
}
|
|||
|
|
}
|