修复未绑定邮箱用户登录会话失效并保留管理员权限边界

This commit is contained in:
xuans 2026-09-25 02:12:13 +08:00
parent a795d59920
commit 03e7a14aa3
2 changed files with 83 additions and 1 deletions

View File

@ -0,0 +1,81 @@
package httpapi
import (
"encoding/json"
"net/http"
"net/http/httptest"
"path/filepath"
"testing"
"git.pcux.cn/xuans/Scert/backend/internal/database"
"git.pcux.cn/xuans/Scert/backend/internal/repository"
)
func TestAuthSessionsSupportOptionalEmail(t *testing.T) {
for _, role := range []string{repository.UserRoleUser, repository.UserRoleAdmin} {
t.Run(role, func(t *testing.T) {
directory := t.TempDir()
cfg := readyTestConfig(directory)
cfg.Database.DSN = filepath.Join(directory, "optional-email.db")
db, err := database.Open(t.Context(), cfg.Database)
if err != nil {
t.Fatal(err)
}
defer db.Close()
if err := database.Migrate(t.Context(), db.DB, "sqlite"); err != nil {
t.Fatal(err)
}
users := repository.NewSQLUserRepository(db.DB, "sqlite")
user := repository.User{
ID: "optional-email", Username: "optional-email",
PasswordHash: httpUserPasswordHash(t, "Operator123"),
Role: role, Status: repository.UserStatusActive,
}
if err := users.CreateUser(t.Context(), user); err != nil {
t.Fatal(err)
}
server := New(Dependencies{Config: cfg, Repository: &stubDomainRepository{}, Users: users})
login := loginAuthTest(t, server, user.Username, "Operator123", false)
cookies := login.Result().Cookies()
if len(cookies) == 0 {
t.Fatal("missing session cookie")
}
loginAuthTest(t, server, user.Username, "Operator123", false)
response := authSessionRequest(server, cookies[0])
if response.Code != http.StatusOK {
t.Fatalf("optional-email session: status=%d body=%s", response.Code, response.Body.String())
}
var payload struct {
Authenticated bool `json:"authenticated"`
Email string `json:"email"`
Role string `json:"role"`
IsAdmin bool `json:"is_admin"`
}
if err := json.Unmarshal(response.Body.Bytes(), &payload); err != nil {
t.Fatal(err)
}
if !payload.Authenticated || payload.Email != "" || payload.Role != role || payload.IsAdmin != (role == repository.UserRoleAdmin) {
t.Fatalf("unexpected session identity: %+v", payload)
}
request := httptest.NewRequest(http.MethodGet, "/api/v1/domains", nil)
request.AddCookie(cookies[0])
access := httptest.NewRecorder()
server.Handler().ServeHTTP(access, request)
expectedStatus := http.StatusForbidden
if role == repository.UserRoleAdmin {
expectedStatus = http.StatusOK
}
if access.Code != expectedStatus {
t.Fatalf("role boundary: status=%d want=%d", access.Code, expectedStatus)
}
user.Status = repository.UserStatusDisabled
if err := users.UpdateUser(t.Context(), user); err != nil {
t.Fatal(err)
}
disabled := authSessionRequest(server, cookies[0])
if disabled.Code != http.StatusUnauthorized {
t.Fatalf("disabled session: status=%d", disabled.Code)
}
})
}
}

View File

@ -11,7 +11,8 @@ const authStateCapacity = 4096
const authUserSessionCapacity = 16
func (session authSession) complete() bool {
return strings.TrimSpace(session.UserID) != "" && strings.TrimSpace(session.Username) != "" && strings.TrimSpace(session.Email) != "" &&
return strings.TrimSpace(session.UserID) != "" && strings.TrimSpace(session.Username) != "" &&
(session.Role != repository.UserRoleSuperAdmin || strings.TrimSpace(session.Email) != "") &&
strings.TrimSpace(session.PasswordHash) != "" &&
(session.Role == repository.UserRoleUser || session.Role == repository.UserRoleAdmin || session.Role == repository.UserRoleSuperAdmin)
}