public List<Menus> getRoles() { final Set<Role> roles = identity.getRoles(); final List<Menus> result = new ArrayList<>(roles.size()); result.add(MenuFactory.newSimpleItem(constants.LogOut()).respondsWith(new LogoutCommand()).endMenu().build()); for (final Role role : roles) { if (!role.getName().equals("IS_REMEMBER_ME")) { result.add(MenuFactory.newSimpleItem(constants.Role() + ": " + role.getName()).endMenu().build()); } } return result; }
@Test public void addRolesMenuItemsTest() { menusHelper.addRolesMenuItems(); ArgumentCaptor<Menus> menusCaptor = ArgumentCaptor.forClass(Menus.class); verify(userMenu, times(3)).addMenus(menusCaptor.capture()); List<Menus> menusList = menusCaptor.getAllValues(); assertEquals(3, menusList.size()); assertEquals(1, menusList.get(0).getItems().size()); assertEquals(1, menusList.get(1).getItems().size()); assertEquals(1, menusList.get(2).getItems().size()); checkIfMenuContainsRole(menusList, menusHelper.constants.LogOut()); checkIfMenuContainsRole(menusList, "Role: role1"); checkIfMenuContainsRole(menusList, "Role: role2"); }
public void addUserMenuItems() { userMenu.clear(); final Menus userMenus = MenuFactory .newTopLevelMenu(constants.LogOut()) .respondsWith(new LogoutCommand()) .endMenu() .newTopLevelMenu("About") .respondsWith(aboutCommand) .endMenu() .build(); userMenu.addMenus(userMenus); }