Adobe-Consulting-Services/acs-aem-commons

/*
 * ACS AEM Commons
 *
 * Copyright (C) 2013 - 2023 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());        
    }
}