优化

2026-08-17 02:12:17 +08:00
parent 560ae46b82
commit 6a05803e08
96 changed files with 2888 additions and 685 deletions
@@ -0,0 +1,83 @@
+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)
+	}
+}