feat: add auth provider boundary

This commit is contained in:
QiuSW
2026-07-08 23:37:19 +08:00
parent 811a9c889f
commit 5aaac0b166
19 changed files with 788 additions and 40 deletions
+94
View File
@@ -2,6 +2,7 @@ package services
import (
"database/sql"
"errors"
"testing"
)
@@ -62,3 +63,96 @@ func TestAppConfigServiceLanguage(t *testing.T) {
t.Fatalf("updated language = %q, want en", got)
}
}
func TestAuthServiceLocalLoginLifecycle(t *testing.T) {
store := newTestStore(t)
service := NewAuthService(store)
session, err := service.Login("local", "admin", "admin123")
if err != nil {
t.Fatalf("login: %v", err)
}
if !session.Active {
t.Fatal("session is not active")
}
if session.Username != "admin" {
t.Fatalf("username = %q, want admin", session.Username)
}
if session.Provider != "local" {
t.Fatalf("provider = %q, want local", session.Provider)
}
if session.Token == "" {
t.Fatal("session token is empty")
}
current, err := service.GetCurrentSession()
if err != nil {
t.Fatalf("get current session: %v", err)
}
if current.ID != session.ID {
t.Fatalf("current session id = %d, want %d", current.ID, session.ID)
}
var loginAuditCount int
if err := store.DB.QueryRow("SELECT COUNT(*) FROM audit_logs WHERE action = 'auth.login.success' AND actor = 'admin'").Scan(&loginAuditCount); err != nil {
t.Fatalf("count login audit logs: %v", err)
}
if loginAuditCount != 1 {
t.Fatalf("login audit count = %d, want 1", loginAuditCount)
}
if err := service.Logout(); err != nil {
t.Fatalf("logout: %v", err)
}
current, err = service.GetCurrentSession()
if err != nil {
t.Fatalf("get current session after logout: %v", err)
}
if current.Active {
t.Fatal("current session is active after logout")
}
var logoutAuditCount int
if err := store.DB.QueryRow("SELECT COUNT(*) FROM audit_logs WHERE action = 'auth.logout' AND actor = 'admin'").Scan(&logoutAuditCount); err != nil {
t.Fatalf("count logout audit logs: %v", err)
}
if logoutAuditCount != 1 {
t.Fatalf("logout audit count = %d, want 1", logoutAuditCount)
}
}
func TestAuthServiceLocalLoginFailureWritesAuditLog(t *testing.T) {
store := newTestStore(t)
service := NewAuthService(store)
_, err := service.Login("local", "admin", "bad-password")
if !errors.Is(err, errInvalidCredentials) {
t.Fatalf("login error = %v, want invalid credentials", err)
}
var auditCount int
if err := store.DB.QueryRow("SELECT COUNT(*) FROM audit_logs WHERE action = 'auth.login.failure' AND actor = 'admin'").Scan(&auditCount); err != nil {
t.Fatalf("count audit logs: %v", err)
}
if auditCount != 1 {
t.Fatalf("audit count = %d, want 1", auditCount)
}
}
func TestAuthServiceRemoteProviderReserved(t *testing.T) {
store := newTestStore(t)
service := NewAuthService(store)
_, err := service.Login("remote", "admin", "admin123")
if !errors.Is(err, errRemoteProviderNotReady) {
t.Fatalf("login error = %v, want remote provider not ready", err)
}
var auditCount int
if err := store.DB.QueryRow("SELECT COUNT(*) FROM audit_logs WHERE action = 'auth.login.failure' AND target_id = 'remote'").Scan(&auditCount); err != nil {
t.Fatalf("count audit logs: %v", err)
}
if auditCount != 1 {
t.Fatalf("audit count = %d, want 1", auditCount)
}
}