45 lines
1.4 KiB
Go
45 lines
1.4 KiB
Go
package setup
|
|
|
|
import (
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"git.pcux.cn/xuans/Smail/config"
|
|
"git.pcux.cn/xuans/Smail/db"
|
|
"git.pcux.cn/xuans/Smail/models"
|
|
"git.pcux.cn/xuans/Smail/utils/context"
|
|
"git.pcux.cn/xuans/Smail/utils/password"
|
|
)
|
|
|
|
func TestSetAdminPasswordValidatesAndPreventsDuplicates(t *testing.T) {
|
|
oldConfig := config.Instance
|
|
config.Instance = &config.Config{DbType: config.DBTypeSQLite, DbDSN: filepath.Join(t.TempDir(), "setup.db")}
|
|
t.Cleanup(func() { config.Instance = oldConfig })
|
|
if err := db.Init(""); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Cleanup(func() { _ = db.Instance.Close() })
|
|
|
|
ctx := &context.Context{}
|
|
if err := SetAdminPassword(ctx, ".invalid", "StrongPass123"); err == nil {
|
|
t.Fatal("invalid administrator account was accepted")
|
|
}
|
|
if err := SetAdminPassword(ctx, "admin-user", "short"); err == nil {
|
|
t.Fatal("short administrator password was accepted")
|
|
}
|
|
if err := SetAdminPassword(ctx, "admin-user", "StrongPass123"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := SetAdminPassword(ctx, "second-admin", "StrongPass456"); err == nil {
|
|
t.Fatal("duplicate administrator initialization was accepted")
|
|
}
|
|
|
|
var user models.User
|
|
if _, err := db.Instance.Where("account = ?", "admin-user").Get(&user); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if user.ID == 0 || user.IsAdmin != 1 || !password.Verify(user.Password, "StrongPass123") {
|
|
t.Fatal("administrator was not initialized correctly")
|
|
}
|
|
}
|