优化

internal/biz/user_test.go

@@ -0,0 +1,37 @@
+package biz
+
+import (
+	"context"
+	"errors"
+	"testing"
+)
+
+type defaultRouterUserRepo struct {
+	UserRepo
+	hasMenu bool
+	err     error
+}
+
+func (r *defaultRouterUserRepo) HasAuthorityMenu(context.Context, uint, string) (bool, error) {
+	return r.hasMenu, r.err
+}
+
+func TestFallbackDefaultRouterMatchesMenuLookupOutcome(t *testing.T) {
+	user := &User{AuthorityID: 888, Authority: Authority{DefaultRouter: "dashboard"}}
+	NewUserUsecase(&defaultRouterUserRepo{hasMenu: false}).fallbackDefaultRouter(context.Background(), user)
+	if user.Authority.DefaultRouter != "404" {
+		t.Fatalf("missing default route = %q", user.Authority.DefaultRouter)
+	}
+
+	user.Authority.DefaultRouter = "dashboard"
+	NewUserUsecase(&defaultRouterUserRepo{err: errors.New("database unavailable")}).fallbackDefaultRouter(context.Background(), user)
+	if user.Authority.DefaultRouter != "dashboard" {
+		t.Fatalf("lookup error changed default route to %q", user.Authority.DefaultRouter)
+	}
+
+	user.Authority.DefaultRouter = ""
+	NewUserUsecase(&defaultRouterUserRepo{}).fallbackDefaultRouter(context.Background(), user)
+	if user.Authority.DefaultRouter != "404" {
+		t.Fatalf("empty missing default route = %q", user.Authority.DefaultRouter)
+	}
+}