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.dam;

import com.day.cq.dam.api.Asset;
import com.day.cq.dam.api.Rendition;

import org.junit.Before;
import org.junit.Test;

import java.util.ArrayList;
import java.util.List;

import static org.junit.Assert.assertEquals;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

public class RenditionPatternPickerTest {

    Rendition originalRendition;
    Rendition webRendition;
    Rendition largeRendition;
    Rendition smallRendition;
    Rendition customRendition;
    List<Rendition> renditions;
    Asset asset;

    public RenditionPatternPickerTest() {
    }

    @Before
    public void setUp() {
        originalRendition = mock(Rendition.class, "original");
        when(originalRendition.getName()).thenReturn("original");

        largeRendition = mock(Rendition.class, "large");
        when(largeRendition.getName()).thenReturn("cq5dam.thumbnail.1000.1000");

        smallRendition = mock(Rendition.class, "small");
        when(smallRendition.getName()).thenReturn("cq5dam.thumbnail.100.100");

        webRendition = mock(Rendition.class, "web");
        when(webRendition.getName()).thenReturn("cq5dam.web.1280.1280");

        customRendition = mock(Rendition.class, "custom");
        when(customRendition.getName()).thenReturn("custom");

        renditions = new ArrayList<>();
        renditions.add(originalRendition);
        renditions.add(webRendition);
        renditions.add(largeRendition);
        renditions.add(smallRendition);
        renditions.add(customRendition);

        asset = mock(Asset.class);
        when(asset.getRenditions()).thenReturn(renditions);
    }

    /**
     * Test of getRendition method, of class RenditionPatternPicker.
     */
    @Test
    public void testGetRendition_MatchingRegex() {
        RenditionPatternPicker instance = new RenditionPatternPicker("^cust.*");
        Rendition expResult = customRendition;
        Rendition result = instance.getRendition(asset);
        assertEquals(expResult, result);
    }

    @Test
    public void testGetRendition_MultiMatchingRegex() {
        RenditionPatternPicker instance = new RenditionPatternPicker("^cq5dam\\.thumb*");
        Rendition expResult = largeRendition;
        Rendition result = instance.getRendition(asset);
        assertEquals(expResult, result);
    }

    @Test
    public void testGetRendition_NonMatchingRegex() {
        RenditionPatternPicker instance = new RenditionPatternPicker("nothinghere");
        Rendition expResult = originalRendition;
        Rendition result = instance.getRendition(asset);
        assertEquals(expResult, result);
    }
}