优化结构
This commit is contained in:
@@ -1,37 +0,0 @@
|
||||
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)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user