@Test public void buildChildren_WithNoDatabaseDataContainingChildren_DoesntAddDummyChildToTreeNode() { UUID parentId = randomUUID(); ArrayList<TreeNode> children = new ArrayList<TreeNode>(); ProductCategoryDAO prodCatDAO = mock(ProductCategoryDAO.class); Mockito.when(prodCatDAO.getChildCategories(parentId)) .thenReturn(asList(new ProductCategoryAggregate(null, "One", 0L))); TreeNodeBuilderImpl builder = new TreeNodeBuilderImpl(prodCatDAO); builder.buildChildren(children, new TreeModel("ProdCat1", parentId)); assertThat(children.size(), is(0)); }
@Test public void buildChildren_WithExistingDummyChild_RemovesDummyChildInTreeNode() { UUID parentId = randomUUID(); List<TreeNode> children = new ArrayList<TreeNode>(); children.add(new DefaultTreeNode(-1)); ProductCategoryDAO prodCatDAO = mock(ProductCategoryDAO.class); Mockito.when(prodCatDAO.getChildCategories(parentId)) .thenReturn(asList(new ProductCategoryAggregate(null, "One", 0L))); TreeNodeBuilderImpl builder = new TreeNodeBuilderImpl(prodCatDAO); builder.buildChildren(children, new TreeModel("ProdCat1", parentId)); assertThat(children.size(), is(1)); assertThat(children.get(0).getChildCount(), is(0)); }