/*
* Copyright 2017 Adobe.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.adobe.acs.commons.util.visitors;
import java.util.ArrayList;
import java.util.Arrays;
import org.apache.sling.api.resource.Resource;
import org.apache.sling.commons.testing.sling.MockResource;
import org.apache.sling.commons.testing.sling.MockResourceResolver;
import org.junit.Before;
import org.junit.Test;
import static org.junit.Assert.*;
public class TreeFilteringResourceVisitorTest {
MockResourceResolver rr = new MockResourceResolver();
MockResource root = new MockResource(rr, "/", "sling:folder");
Resource node1a = new MockResource(rr, "/1a", "nt:folder");
Resource node1b = new MockResource(rr, "/1b", "sling:folder");
Resource node1a1 = new MockResource(rr, "/1a/1", "cq:page");
Resource node1a2 = new MockResource(rr, "/1a/2", "nt:unstructured");
Resource node1b1 = new MockResource(rr, "/1b/1", "nt:unstructured");
Resource node1b2 = new MockResource(rr, "/1b/2", "nt:unstructured");
@Before
public void setUp() throws Exception {
rr.addChildren(root, Arrays.asList(node1a, node1b));
rr.addChildren(node1a, Arrays.asList(node1a1, node1a2));
rr.addChildren(node1b, Arrays.asList(node1b1, node1b2));
}
@Test
public void accept_leafTest_standardFolderTypes() throws Exception {
TreeFilteringResourceVisitor visitor = new TreeFilteringResourceVisitor();
visitor.setBreadthFirstMode();
ArrayList<Resource> nodes = new ArrayList<>();
ArrayList<Resource> leaves = new ArrayList<>();
visitor.setResourceVisitor((res,lvl)->nodes.add(res));
visitor.setLeafVisitor((res,lvl)->leaves.add(res));
visitor.accept(root);
assertArrayEquals(new Resource[] {root, node1a, node1b}, nodes.toArray());
assertArrayEquals(new Resource[] {node1a1, node1a2, node1b1, node1b2}, leaves.toArray());
}
@Test
public void accept_leafTest_customFolderTypes() throws Exception {
TreeFilteringResourceVisitor visitor = new TreeFilteringResourceVisitor("nt:folder","sling:folder","cq:page");
visitor.setBreadthFirstMode();
ArrayList<Resource> nodes = new ArrayList<>();
ArrayList<Resource> leaves = new ArrayList<>();
visitor.setResourceVisitor((res,lvl)->nodes.add(res));
visitor.setLeafVisitor((res,lvl)->leaves.add(res));
visitor.accept(root);
assertArrayEquals(new Resource[] {root, node1a, node1b, node1a1}, nodes.toArray());
assertArrayEquals(new Resource[] {node1a2, node1b1, node1b2}, leaves.toArray());
}
}