2026-09-23 09:40:59 +08:00
|
|
|
package auth
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"context"
|
|
|
|
|
"log/slog"
|
|
|
|
|
"os"
|
|
|
|
|
"path/filepath"
|
|
|
|
|
"testing"
|
|
|
|
|
|
2026-10-01 16:54:35 +08:00
|
|
|
"github.com/havline/havline/internal/db"
|
2026-09-23 09:40:59 +08:00
|
|
|
)
|
|
|
|
|
|
|
|
|
|
func TestEnsureDefaultAdminCreatesOnce(t *testing.T) {
|
2026-10-01 16:54:35 +08:00
|
|
|
conn, err := db.Open(filepath.Join(t.TempDir(), "havline.db"))
|
2026-09-23 09:40:59 +08:00
|
|
|
if err != nil {
|
|
|
|
|
t.Fatalf("open db: %v", err)
|
|
|
|
|
}
|
|
|
|
|
defer conn.Close()
|
|
|
|
|
|
|
|
|
|
if err := db.Migrate(context.Background(), conn, filepath.Join("..", "..", "migrations")); err != nil {
|
|
|
|
|
t.Fatalf("migrate: %v", err)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
svc := New(conn)
|
|
|
|
|
logger := slog.New(slog.NewTextHandler(os.Stderr, nil))
|
|
|
|
|
|
|
|
|
|
if err := svc.EnsureDefaultAdmin(context.Background(), logger); err != nil {
|
|
|
|
|
t.Fatalf("first bootstrap: %v", err)
|
|
|
|
|
}
|
|
|
|
|
initialized, err := svc.IsInitialized(context.Background())
|
|
|
|
|
if err != nil || !initialized {
|
|
|
|
|
t.Fatalf("expected initialized admin, got initialized=%v err=%v", initialized, err)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if err := svc.EnsureDefaultAdmin(context.Background(), logger); err != nil {
|
|
|
|
|
t.Fatalf("second bootstrap: %v", err)
|
|
|
|
|
}
|
|
|
|
|
if _, err := svc.Login(context.Background(), "admin", "wrong-password"); err != ErrUnauthorized {
|
|
|
|
|
t.Fatalf("expected unauthorized for wrong password, got %v", err)
|
|
|
|
|
}
|
|
|
|
|
}
|