mirror of
https://github.com/keven1024/015.git
synced 2026-05-26 07:08:02 +00:00
75 lines
1.5 KiB
Go
75 lines
1.5 KiB
Go
package utils
|
||
|
||
import (
|
||
"bytes"
|
||
"fmt"
|
||
"testing"
|
||
|
||
"backend/internal/utils"
|
||
u "pkg/utils"
|
||
|
||
"github.com/stretchr/testify/assert"
|
||
)
|
||
|
||
func TestGeneratePasswordHash(t *testing.T) {
|
||
|
||
tests := []struct {
|
||
name string
|
||
password string
|
||
salt string
|
||
expectError bool
|
||
err error
|
||
}{
|
||
{
|
||
name: "share.password_salt未配置",
|
||
password: "testpassword",
|
||
salt: "",
|
||
expectError: true,
|
||
err: utils.ErrPasswordSaltNotSet,
|
||
},
|
||
{
|
||
name: "正常生成哈希",
|
||
password: "testpassword123",
|
||
salt: "testsalt",
|
||
expectError: false,
|
||
},
|
||
}
|
||
|
||
for _, tt := range tests {
|
||
t.Run(tt.name, func(t *testing.T) {
|
||
// 设置环境变量
|
||
u.InitTestViper(u.EnvOption{
|
||
ConfigData: bytes.NewBuffer([]byte(fmt.Sprintf(`
|
||
share:
|
||
password_salt: %s
|
||
`, tt.salt))),
|
||
})
|
||
u.SetEnv("share.password_salt", tt.salt)
|
||
|
||
hash, err := utils.GeneratePasswordHash(tt.password)
|
||
|
||
if tt.expectError {
|
||
if err == nil {
|
||
t.Errorf("期望错误，但得到了 nil")
|
||
return
|
||
}
|
||
if err != tt.err {
|
||
t.Errorf("期望错误信息 '%s'，但得到了 '%s'", tt.err.Error(), err.Error())
|
||
}
|
||
return
|
||
}
|
||
|
||
if err != nil {
|
||
t.Errorf("不期望错误，但得到了: %v", err)
|
||
return
|
||
}
|
||
|
||
// 验证哈希格式：argon2 32字节 = 64个十六进制字符
|
||
if len(hash) != 64 {
|
||
t.Errorf("期望哈希长度为64，但得到了 %d", len(hash))
|
||
}
|
||
assert.Equal(t, hash, "537275f995fdd46eb2e5455b8a29adccb60c5637689d29646676a5f1bffb63f3")
|
||
})
|
||
}
|
||
}
|