kra-new/internal/data/system/dictionary_department_parit...

188 lines
6.0 KiB
Go
Raw Permalink Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

package system
import (
"context"
"kra/internal/biz/system"
"testing"
)
func TestDictionaryListPreservesPreloadCollectionShapes(t *testing.T) {
data := newTransactionTestData(t)
ctx := context.Background()
repo := &dictionaryRepo{data: data}
if err := repo.CreateDictionary(ctx, &system.Dictionary{Name: "status", Type: "status"}); err != nil {
t.Fatal(err)
}
if err := repo.CreateDictionary(ctx, &system.Dictionary{Name: "kind", Type: "kind"}); err != nil {
t.Fatal(err)
}
plain, _, err := repo.ListDictionaries(ctx, 1, 1, "", "", false)
if err != nil {
t.Fatal(err)
}
if len(plain) != 2 || plain[0].Children == nil || len(plain[0].Children) != 0 {
t.Fatalf("ordinary dictionary list children = %#v, want an empty preloaded slice", plain)
}
if plain[0].Details != nil {
t.Fatalf("ordinary dictionary list details = %#v, want nil", plain[0].Details)
}
withDetails, _, err := repo.ListDictionaries(ctx, 1, 1, "", "", true)
if err != nil {
t.Fatal(err)
}
if len(withDetails) != 2 || withDetails[0].Details == nil || len(withDetails[0].Details) != 0 {
t.Fatalf("dictionary details preload = %#v, want an empty preloaded slice", withDetails)
}
if withDetails[0].Children != nil {
t.Fatalf("details-only dictionary list children = %#v, want nil", withDetails[0].Children)
}
}
func TestImportDictionaryRejectsDuplicateType(t *testing.T) {
data := newTransactionTestData(t)
ctx := context.Background()
repo := &dictionaryRepo{data: data}
if err := repo.CreateDictionary(ctx, &system.Dictionary{Name: "first", Type: "duplicate"}); err != nil {
t.Fatal(err)
}
if err := repo.ImportDictionary(ctx, &system.Dictionary{Name: "second", Type: "duplicate"}, nil); err == nil || err.Error() != "存在相同的type不允许导入" {
t.Fatalf("duplicate import error = %v", err)
}
var count int64
if err := data.gormDB.WithContext(ctx).Model(&dictionaryPO{}).Where("type = ?", "duplicate").Count(&count).Error; err != nil {
t.Fatal(err)
}
if count != 1 {
t.Fatalf("duplicate dictionary count = %d, want 1", count)
}
}
func TestDictionaryTreeAndParentQueriesPreserveLoadedChildrenShape(t *testing.T) {
data := newTransactionTestData(t)
ctx := context.Background()
repo := &dictionaryRepo{data: data}
dictionary := &system.Dictionary{Name: "tree", Type: "tree"}
if err := repo.CreateDictionary(ctx, dictionary); err != nil {
t.Fatal(err)
}
parent := &system.DictionaryDetail{Label: "parent", Value: "parent", DictionaryID: dictionary.ID, Sort: 2}
if err := repo.CreateDictionaryDetail(ctx, parent); err != nil {
t.Fatal(err)
}
parentID := parent.ID
child := &system.DictionaryDetail{Label: "child", Value: "child", DictionaryID: dictionary.ID, ParentID: &parentID, Sort: 1}
if err := repo.CreateDictionaryDetail(ctx, child); err != nil {
t.Fatal(err)
}
tree, err := repo.DictionaryDetailTree(ctx, dictionary.ID, "")
if err != nil {
t.Fatal(err)
}
if len(tree) != 1 || len(tree[0].Children) != 1 || tree[0].Children[0].Children == nil {
t.Fatalf("dictionary tree = %#v, want recursively loaded child slices", tree)
}
flat, err := repo.DictionaryDetailsByParent(ctx, dictionary.ID, &parentID, false)
if err != nil {
t.Fatal(err)
}
if len(flat) != 1 || flat[0].Children != nil {
t.Fatalf("flat parent query = %#v, want nil children", flat)
}
loaded, err := repo.DictionaryDetailsByParent(ctx, dictionary.ID, &parentID, true)
if err != nil {
t.Fatal(err)
}
if len(loaded) != 1 || loaded[0].Children == nil || len(loaded[0].Children) != 0 {
t.Fatalf("recursive parent query = %#v, want an empty loaded child slice", loaded)
}
}
func TestDictionaryDetailListAppliesZeroLimit(t *testing.T) {
data := newTransactionTestData(t)
ctx := context.Background()
repo := &dictionaryRepo{data: data}
dictionary := &system.Dictionary{Name: "limited", Type: "limited"}
if err := repo.CreateDictionary(ctx, dictionary); err != nil {
t.Fatal(err)
}
if err := repo.CreateDictionaryDetail(ctx, &system.DictionaryDetail{Label: "one", Value: "one", DictionaryID: dictionary.ID}); err != nil {
t.Fatal(err)
}
items, total, err := repo.ListDictionaryDetails(ctx, 0, 0, system.DictionaryDetailFilter{DictionaryID: dictionary.ID})
if err != nil {
t.Fatal(err)
}
if total != 1 || len(items) != 0 {
t.Fatalf("zero-limit detail list: total=%d items=%#v", total, items)
}
}
func TestUserListAppliesZeroLimitForNegativePageSize(t *testing.T) {
data := newTransactionTestData(t)
ctx := context.Background()
if err := data.gormDB.WithContext(ctx).Create(&userPO{Username: "limited", Password: "hash", AuthorityID: 888, Enable: 1}).Error; err != nil {
t.Fatal(err)
}
items, total, err := (&userRepo{data: data}).ListUsers(ctx, 1, -1, &system.UserListFilter{})
if err != nil {
t.Fatal(err)
}
if total != 1 || len(items) != 0 {
t.Fatalf("negative-size user list: total=%d items=%#v", total, items)
}
}
func TestDepartmentTreeAndFlatResultsPreserveChildrenShape(t *testing.T) {
data := newTransactionTestData(t)
ctx := context.Background()
repo := &departmentRepo{data: data}
root := &system.Department{Name: "root"}
if err := repo.CreateDepartment(ctx, root); err != nil {
t.Fatal(err)
}
var rootPO departmentPO
if err := data.gormDB.WithContext(ctx).Where("name = ?", "root").First(&rootPO).Error; err != nil {
t.Fatal(err)
}
if err := repo.CreateDepartment(ctx, &system.Department{Name: "child", ParentID: rootPO.ID}); err != nil {
t.Fatal(err)
}
tree, err := repo.ListDepartments(ctx, "")
if err != nil {
t.Fatal(err)
}
if len(tree) != 1 || len(tree[0].Children) != 1 || tree[0].Children[0].Children == nil || len(tree[0].Children[0].Children) != 0 {
t.Fatalf("department tree children = %#v, want every loaded node to expose a slice", tree)
}
flat, err := repo.ListDepartments(ctx, "child")
if err != nil {
t.Fatal(err)
}
if len(flat) != 1 || flat[0].Children != nil {
t.Fatalf("flat department search children = %#v, want nil", flat)
}
found, err := repo.FindDepartment(ctx, rootPO.ID)
if err != nil {
t.Fatal(err)
}
if found.Children != nil {
t.Fatalf("single department children = %#v, want nil", found.Children)
}
}