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)
	}
}