84 lines
2.6 KiB
Go
84 lines
2.6 KiB
Go
package data
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"kra/internal/conf"
|
|
)
|
|
|
|
func TestPermissionSelectedButtonsEmptyShape(t *testing.T) {
|
|
db, err := openWithDriver("sqlite", "file:"+t.Name()+"?mode=memory&cache=shared")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := db.AutoMigrate(&authorityButtonPO{}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
data := &Data{gormDB: newReloadableDB(db, nil), runtime: conf.NewRuntime(&conf.Data{Database: &conf.Data_Database{Driver: "sqlite"}}, &conf.AdminBackend{})}
|
|
t.Cleanup(func() { data.gormDB.close() })
|
|
|
|
ids, err := (&permissionRepo{data: data}).SelectedButtons(context.Background(), 1, 2)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if ids != nil {
|
|
t.Fatalf("selected buttons = %#v, want nil for an empty compatible response", ids)
|
|
}
|
|
}
|
|
|
|
func TestPermissionCanRemoveButtonUsesRawQueryID(t *testing.T) {
|
|
db, err := openWithDriver("sqlite", "file:"+t.Name()+"?mode=memory&cache=shared")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := db.AutoMigrate(&authorityButtonPO{}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
data := &Data{gormDB: newReloadableDB(db, nil), runtime: conf.NewRuntime(&conf.Data{Database: &conf.Data_Database{Driver: "sqlite"}}, &conf.AdminBackend{})}
|
|
t.Cleanup(func() { data.gormDB.close() })
|
|
repo := &permissionRepo{data: data}
|
|
if err := data.gormDB.WithContext(context.Background()).Create(&authorityButtonPO{AuthorityID: 888, MenuID: 10, ButtonID: 31}).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
allowed, err := repo.CanRemoveButton(context.Background(), "31")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if allowed {
|
|
t.Fatal("assigned button was reported removable")
|
|
}
|
|
allowed, err = repo.CanRemoveButton(context.Background(), "")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !allowed {
|
|
t.Fatal("empty query ID should not match an assigned button")
|
|
}
|
|
}
|
|
|
|
func TestPermissionSetSelectedButtonsPreservesDuplicates(t *testing.T) {
|
|
db, err := openWithDriver("sqlite", "file:"+t.Name()+"?mode=memory&cache=shared")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := db.AutoMigrate(&authorityButtonPO{}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
data := &Data{gormDB: newReloadableDB(db, nil), runtime: conf.NewRuntime(&conf.Data{Database: &conf.Data_Database{Driver: "sqlite"}}, &conf.AdminBackend{})}
|
|
t.Cleanup(func() { data.gormDB.close() })
|
|
repo := &permissionRepo{data: data}
|
|
|
|
if err := repo.SetSelectedButtons(context.Background(), 888, 10, []uint{31, 31}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
selected, err := repo.SelectedButtons(context.Background(), 888, 10)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(selected) != 2 || selected[0] != 31 || selected[1] != 31 {
|
|
t.Fatalf("selected = %#v, want duplicate button IDs preserved", selected)
|
|
}
|
|
}
|