完成 TreeBuilder 单元测试。
parent
f6b2509b12
commit
424857df6a
|
|
@ -1,7 +1,7 @@
|
|||
[x] 无需测试 - 6
|
||||
[ ] 未开始测试 - 34
|
||||
[-] 测试未完成 - 15
|
||||
[Y] 测试完成 - 15
|
||||
[-] 测试未完成 - 14
|
||||
[Y] 测试完成 - 16
|
||||
|
||||
xyz.zhouxy.plusone.commons
|
||||
├───annotation
|
||||
|
|
@ -97,4 +97,4 @@ xyz.zhouxy.plusone.commons
|
|||
RegexTools.java [ ]
|
||||
SnowflakeIdGenerator.java [ ]
|
||||
StringTools.java [Y]
|
||||
TreeBuilder.java [-]
|
||||
TreeBuilder.java [Y]
|
||||
|
|
|
|||
|
|
@ -16,8 +16,14 @@
|
|||
|
||||
package xyz.zhouxy.plusone.commons.util;
|
||||
|
||||
import static org.junit.jupiter.api.Assertions.assertEquals;
|
||||
|
||||
import java.io.Serializable;
|
||||
import java.util.Arrays;
|
||||
import java.util.Comparator;
|
||||
import java.util.HashMap;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.Optional;
|
||||
import java.util.stream.Collectors;
|
||||
|
||||
|
|
@ -25,143 +31,203 @@ import org.junit.jupiter.api.Test;
|
|||
import org.slf4j.Logger;
|
||||
import org.slf4j.LoggerFactory;
|
||||
|
||||
import com.google.common.collect.ImmutableList;
|
||||
import com.google.common.collect.Lists;
|
||||
import com.google.gson.Gson;
|
||||
|
||||
import cn.hutool.core.util.ObjectUtil;
|
||||
import lombok.EqualsAndHashCode;
|
||||
import lombok.ToString;
|
||||
|
||||
class TreeBuilderTests {
|
||||
|
||||
private static final Logger log = LoggerFactory.getLogger(TreeBuilderTests.class);
|
||||
|
||||
private final MenuItem A = MenuItem.of("A", "首页", "/home", 1);
|
||||
private final MenuList B = MenuList.of("B", "系统管理", 3);
|
||||
private final MenuItem B001 = /**/MenuItem.of("B", "B001", "功能管理", "/sys/function-mgmt", 4);
|
||||
private final MenuItem B002 = /**/MenuItem.of("B", "B002", "角色管理", "/sys/role-mgmt", 3);
|
||||
private final MenuItem B003 = /**/MenuItem.of("B", "B003", "账号管理", "/sys/account-mgmt", 2);
|
||||
private final MenuItem B004 = /**/MenuItem.of("B", "B004", "系统参数管理", "/sys/param-mgmt", 1);
|
||||
private final MenuList C = MenuList.of("C", "一级菜单C", 2);
|
||||
private final MenuList C1 = /**/MenuList.of("C", "C1", "二级菜单C1", 3);
|
||||
private final MenuItem C1001 = /**//**/MenuItem.of("C1", "C1001", "三级菜单C1001", "/c/c1/c1001", 1);
|
||||
private final MenuItem C1002 = /**//**/MenuItem.of("C1", "C1002", "三级菜单C1002", "/c/c1/c1002", 2);
|
||||
private final MenuItem C2 = /**/MenuItem.of("C", "C2", "二级菜单C2", "/c/c2", 1);
|
||||
private final MenuItem C3 = /**/MenuItem.of("C", "C3", "二级菜单C3", "/c/c3", 2);
|
||||
|
||||
private final List<Menu> menus = ImmutableList.of(B, C1002, A, B004, C3, B001, B003, C1, C1001, B002, C, C2);
|
||||
|
||||
private final TreeBuilder<Menu, MenuList, String> treeBuilder = new TreeBuilder<>(
|
||||
Menu::getMenuCode,
|
||||
menu -> Optional.ofNullable(menu.parentMenuCode),
|
||||
MenuList::addChild,
|
||||
(a, b) -> Integer.compare(a.getOrderNum(), b.getOrderNum()));
|
||||
Menu.orderNumComparator);
|
||||
|
||||
@Test
|
||||
void testBuildTree() {
|
||||
List<Menu> menus = Lists.newArrayList(
|
||||
MenuList.of("B", "系统管理", 3),
|
||||
MenuItem.of("A", "首页", "/home", 1),
|
||||
/**/MenuItem.of("B", "B002", "角色管理", "/sys/role-mgmt", 3),
|
||||
/**/MenuItem.of("B", "B001", "功能管理", "/sys/function-mgmt", 4),
|
||||
/**/MenuItem.of("B", "B004", "系统参数管理", "/sys/param-mgmt", 1),
|
||||
/**/MenuItem.of("B", "B003", "账号管理", "/sys/account-mgmt", 2),
|
||||
MenuList.of("C", "一级菜单C", 2),
|
||||
/**/MenuItem.of("C", "C3", "二级菜单C3", "/c/c3", 2),
|
||||
/**/MenuList.of("C", "C1", "二级菜单C1", 2),
|
||||
/**//**/MenuItem.of("C1", "C1001", "三级菜单C1001", "/c/c1/c1001", 1),
|
||||
/**//**/MenuItem.of("C1", "C1002", "三级菜单C1002", "/c/c1/c1002", 2),
|
||||
/**/MenuItem.of("C", "C2", "二级菜单C2", "/c/c2", 1));
|
||||
|
||||
List<Menu> clonedMenus;
|
||||
|
||||
clonedMenus = menus.stream().map(m -> ObjectUtil.clone(m)).collect(Collectors.toList());
|
||||
void testBuildTreeAndSortedByOrderNum() {
|
||||
List<Menu> clonedMenus = menus.stream().map(ObjectUtil::clone).collect(Collectors.toList());
|
||||
List<Menu> menuTreeSortedByOrderNum = treeBuilder.buildTree(clonedMenus);
|
||||
log.info("menuTreeSortedByOrderNum: {}", new Gson().toJson(menuTreeSortedByOrderNum));
|
||||
|
||||
clonedMenus = menus.stream().map(m -> ObjectUtil.clone(m)).collect(Collectors.toList());
|
||||
assertEquals(clonedMenus.stream()
|
||||
.filter(menu -> menu.getParentMenuCode() == null)
|
||||
.sorted(Menu.orderNumComparator)
|
||||
.collect(Collectors.toList()),
|
||||
menuTreeSortedByOrderNum);
|
||||
|
||||
Map<String, Menu> menuMap = new HashMap<>();
|
||||
for (Menu element : clonedMenus) {
|
||||
menuMap.put(element.getMenuCode(), element);
|
||||
}
|
||||
|
||||
assertEquals(Arrays.stream(new Menu[] { B001, B002, B003, B004 })
|
||||
.sorted(Menu.orderNumComparator)
|
||||
.collect(Collectors.toList()),
|
||||
MenuList.class.cast(menuMap.get("B")).children);
|
||||
|
||||
assertEquals(Arrays.stream(new Menu[] { C1, C2, C3 })
|
||||
.sorted(Menu.orderNumComparator)
|
||||
.collect(Collectors.toList()),
|
||||
MenuList.class.cast(menuMap.get("C")).children);
|
||||
|
||||
assertEquals(Arrays.stream(new Menu[] { C1001, C1002 })
|
||||
.sorted(Menu.orderNumComparator)
|
||||
.collect(Collectors.toList()),
|
||||
MenuList.class.cast(menuMap.get("C1")).children);
|
||||
|
||||
}
|
||||
|
||||
@Test
|
||||
void testBuildTreeAndSortedByMenuCode() {
|
||||
List<Menu> clonedMenus;
|
||||
|
||||
clonedMenus = menus.stream().map(ObjectUtil::clone).collect(Collectors.toList());
|
||||
List<Menu> menuTreeSortedByMenuCode = treeBuilder.buildTree(
|
||||
clonedMenus,
|
||||
(a, b) -> a.getMenuCode().compareTo(b.getMenuCode())
|
||||
);
|
||||
(a, b) -> a.getMenuCode().compareTo(b.getMenuCode()));
|
||||
log.info("menuTreeSortedByMenuCode: {}", new Gson().toJson(menuTreeSortedByMenuCode));
|
||||
}
|
||||
}
|
||||
|
||||
@ToString
|
||||
abstract class Menu implements Serializable {
|
||||
protected final String parentMenuCode;
|
||||
protected final String menuCode;
|
||||
protected final String title;
|
||||
protected final int orderNum;
|
||||
assertEquals(clonedMenus.stream()
|
||||
.filter(menu -> menu.getParentMenuCode() == null)
|
||||
.sorted((a, b) -> a.getMenuCode().compareTo(b.getMenuCode()))
|
||||
.collect(Collectors.toList()),
|
||||
menuTreeSortedByMenuCode);
|
||||
|
||||
public Menu(String parentMenuCode, String menuCode, String title, int orderNum) {
|
||||
this.parentMenuCode = parentMenuCode;
|
||||
this.menuCode = menuCode;
|
||||
this.title = title;
|
||||
this.orderNum = orderNum;
|
||||
}
|
||||
|
||||
public String getMenuCode() {
|
||||
return menuCode;
|
||||
}
|
||||
|
||||
public String getParentMenuCode() {
|
||||
return parentMenuCode;
|
||||
}
|
||||
|
||||
public String getTitle() {
|
||||
return title;
|
||||
}
|
||||
|
||||
public int getOrderNum() {
|
||||
return orderNum;
|
||||
}
|
||||
|
||||
private static final long serialVersionUID = 20240917181424L;
|
||||
}
|
||||
|
||||
@ToString(callSuper = true)
|
||||
class MenuItem extends Menu {
|
||||
|
||||
private final String url;
|
||||
|
||||
private MenuItem(String parentMenuCode, String menuCode, String title, String url, int orderNum) {
|
||||
super(parentMenuCode, menuCode, title, orderNum);
|
||||
this.url = url;
|
||||
}
|
||||
|
||||
static MenuItem of(String parentMenuCode, String menuCode, String title, String url, int orderNum) {
|
||||
return new MenuItem(parentMenuCode, menuCode, title, url, orderNum);
|
||||
}
|
||||
|
||||
static MenuItem of(String menuCode, String title, String url, int orderNum) {
|
||||
return new MenuItem(null, menuCode, title, url, orderNum);
|
||||
}
|
||||
|
||||
public String getUrl() {
|
||||
return url;
|
||||
}
|
||||
|
||||
private static final long serialVersionUID = 20240917181910L;
|
||||
}
|
||||
|
||||
@ToString(callSuper = true)
|
||||
class MenuList extends Menu {
|
||||
|
||||
private List<Menu> children;
|
||||
|
||||
private MenuList(String parentMenuCode, String menuCode, String title, int orderNum) {
|
||||
super(parentMenuCode, menuCode, title, orderNum);
|
||||
}
|
||||
|
||||
static MenuList of(String parentMenuCode, String menuCode, String title, int orderNum) {
|
||||
return new MenuList(parentMenuCode, menuCode, title, orderNum);
|
||||
}
|
||||
|
||||
static MenuList of(String menuCode, String title, int orderNum) {
|
||||
return new MenuList(null, menuCode, title, orderNum);
|
||||
}
|
||||
|
||||
static MenuList of(String menuCode, String title, Iterable<Menu> children, int orderNum) {
|
||||
return of(null, menuCode, title, children, orderNum);
|
||||
}
|
||||
|
||||
static MenuList of(String parentMenuCode, String menuCode, String title, Iterable<Menu> children, int orderNum) {
|
||||
final MenuList instance = of(parentMenuCode, menuCode, title, orderNum);
|
||||
children.forEach(instance::addChild);
|
||||
return instance;
|
||||
}
|
||||
|
||||
public void addChild(Menu child) {
|
||||
if (this.children == null) {
|
||||
this.children = Lists.newArrayList();
|
||||
Map<String, Menu> menuMap = new HashMap<>();
|
||||
for (Menu element : clonedMenus) {
|
||||
menuMap.put(element.getMenuCode(), element);
|
||||
}
|
||||
this.children.add(child);
|
||||
|
||||
assertEquals(ImmutableList.of(B001, B002, B003, B004),
|
||||
MenuList.class.cast(menuMap.get("B")).children);
|
||||
|
||||
assertEquals(ImmutableList.of(C1, C2, C3),
|
||||
MenuList.class.cast(menuMap.get("C")).children);
|
||||
|
||||
assertEquals(ImmutableList.of(C1001, C1002),
|
||||
MenuList.class.cast(menuMap.get("C1")).children);
|
||||
}
|
||||
|
||||
private static final long serialVersionUID = 20240917181917L;
|
||||
@ToString
|
||||
@EqualsAndHashCode
|
||||
private static abstract class Menu implements Serializable { // NOSONAR
|
||||
protected final String parentMenuCode;
|
||||
protected final String menuCode;
|
||||
protected final String title;
|
||||
protected final int orderNum;
|
||||
|
||||
public Menu(String parentMenuCode, String menuCode, String title, int orderNum) {
|
||||
this.parentMenuCode = parentMenuCode;
|
||||
this.menuCode = menuCode;
|
||||
this.title = title;
|
||||
this.orderNum = orderNum;
|
||||
}
|
||||
|
||||
public String getMenuCode() {
|
||||
return menuCode;
|
||||
}
|
||||
|
||||
public String getParentMenuCode() {
|
||||
return parentMenuCode;
|
||||
}
|
||||
|
||||
public String getTitle() {
|
||||
return title;
|
||||
}
|
||||
|
||||
public int getOrderNum() {
|
||||
return orderNum;
|
||||
}
|
||||
|
||||
public static Comparator<Menu> orderNumComparator =
|
||||
(a, b) -> Integer.compare(a.getOrderNum(), b.getOrderNum());
|
||||
|
||||
private static final long serialVersionUID = 20240917181424L;
|
||||
}
|
||||
|
||||
@ToString(callSuper = true)
|
||||
@EqualsAndHashCode(callSuper = true)
|
||||
private static final class MenuItem extends Menu {
|
||||
|
||||
private final String url;
|
||||
|
||||
private MenuItem(String parentMenuCode, String menuCode, String title, String url, int orderNum) {
|
||||
super(parentMenuCode, menuCode, title, orderNum);
|
||||
this.url = url;
|
||||
}
|
||||
|
||||
static MenuItem of(String parentMenuCode, String menuCode, String title, String url, int orderNum) {
|
||||
return new MenuItem(parentMenuCode, menuCode, title, url, orderNum);
|
||||
}
|
||||
|
||||
static MenuItem of(String menuCode, String title, String url, int orderNum) {
|
||||
return new MenuItem(null, menuCode, title, url, orderNum);
|
||||
}
|
||||
|
||||
public String getUrl() {
|
||||
return url;
|
||||
}
|
||||
|
||||
private static final long serialVersionUID = 20240917181910L;
|
||||
}
|
||||
|
||||
@ToString(callSuper = true)
|
||||
private static final class MenuList extends Menu {
|
||||
|
||||
private List<Menu> children;
|
||||
|
||||
private MenuList(String parentMenuCode, String menuCode, String title, int orderNum) {
|
||||
super(parentMenuCode, menuCode, title, orderNum);
|
||||
}
|
||||
|
||||
static MenuList of(String parentMenuCode, String menuCode, String title, int orderNum) {
|
||||
return new MenuList(parentMenuCode, menuCode, title, orderNum);
|
||||
}
|
||||
|
||||
static MenuList of(String menuCode, String title, int orderNum) {
|
||||
return new MenuList(null, menuCode, title, orderNum);
|
||||
}
|
||||
|
||||
@SuppressWarnings("unused")
|
||||
static MenuList of(String menuCode, String title, Iterable<Menu> children, int orderNum) {
|
||||
return of(null, menuCode, title, children, orderNum);
|
||||
}
|
||||
|
||||
static MenuList of(String parentMenuCode, String menuCode, String title, Iterable<Menu> children,
|
||||
int orderNum) {
|
||||
final MenuList instance = of(parentMenuCode, menuCode, title, orderNum);
|
||||
children.forEach(instance::addChild);
|
||||
return instance;
|
||||
}
|
||||
|
||||
public void addChild(Menu child) {
|
||||
if (this.children == null) {
|
||||
this.children = Lists.newArrayList();
|
||||
}
|
||||
this.children.add(child);
|
||||
}
|
||||
|
||||
private static final long serialVersionUID = 20240917181917L;
|
||||
}
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue