StyleGeneratorTest.java

/* (c) 2014 - 2015 Open Source Geospatial Foundation - all rights reserved
 * (c) 2001 - 2013 OpenPlans
 * This code is licensed under the GPL 2.0 license, available at the root
 * application directory.
 */
package org.geoserver.catalog;

import static org.easymock.EasyMock.anyObject;
import static org.easymock.EasyMock.expect;
import static org.easymock.EasyMock.expectLastCall;
import static org.easymock.EasyMock.getCurrentArguments;
import static org.easymock.classextension.EasyMock.createNiceMock;
import static org.easymock.classextension.EasyMock.replay;
import static org.hamcrest.Matchers.instanceOf;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertThat;

import java.io.ByteArrayInputStream;
import java.io.InputStream;
import java.util.List;

import org.apache.commons.io.IOUtils;
import org.easymock.IAnswer;
import org.geoserver.catalog.impl.CatalogFactoryImpl;
import org.geoserver.catalog.impl.CoverageInfoImpl;
import org.geotools.data.DataUtilities;
import org.geotools.factory.CommonFactoryFinder;
import org.geotools.styling.FeatureTypeStyle;
import org.geotools.styling.NamedLayer;
import org.geotools.styling.RasterSymbolizer;
import org.geotools.styling.SLDParser;
import org.geotools.styling.Style;
import org.geotools.styling.StyledLayerDescriptor;
import org.junit.Test;
import org.opengis.feature.simple.SimpleFeatureType;
import org.vfny.geoserver.util.SLDValidator;

public class StyleGeneratorTest {

    @Test
    public void testCreateInWorkspace() throws Exception {
        ResourcePool rp = createNiceMock(ResourcePool.class);

        Catalog cat = createNiceMock(Catalog.class);
        expect(cat.getFactory()).andReturn(new CatalogFactoryImpl(null)).anyTimes();
        expect(cat.getResourcePool()).andReturn(rp).anyTimes();

        WorkspaceInfo ws = createNiceMock(WorkspaceInfo.class);

        FeatureTypeInfo ft = createNiceMock(FeatureTypeInfo.class);
        expect(ft.getName()).andReturn("foo").anyTimes();

        replay(rp, ft, ws, cat);

        StyleGenerator gen = new StyleGenerator(cat);
        gen.setWorkspace(ws);

        SimpleFeatureType schema = DataUtilities.createType("foo", "geom:Point");
        StyleInfo style = gen.createStyle(new SLDHandler(), ft, schema);
        assertNotNull(style);
        assertNotNull(style.getWorkspace());
    }

    @Test
    public void testGenericStyle() throws Exception {
        ResourcePool rp = createNiceMock(ResourcePool.class);
        rp.writeStyle((StyleInfo) anyObject(), (InputStream) anyObject());
        expectLastCall().andAnswer(new IAnswer<Void>() {

            @Override
            public Void answer() throws Throwable {
                Object[] args = getCurrentArguments();
                InputStream is = (InputStream) args[1];
                byte[] input = IOUtils.toByteArray(is);
                
                SLDParser parser = new SLDParser(CommonFactoryFinder.getStyleFactory());
                parser.setInput(new ByteArrayInputStream(input));
                StyledLayerDescriptor sld = parser.parseSLD();
                
                NamedLayer nl = (NamedLayer) sld.getStyledLayers()[0];
                assertEquals("foo", nl.getName());
                Style style = nl.getStyles()[0];
                assertEquals("A orange generic style", style.getDescription().getTitle()
                        .toString());
                assertEquals(1, style.featureTypeStyles().size());
                FeatureTypeStyle fts = style.featureTypeStyles().get(0);
                assertEquals("first", fts.getOptions().get("ruleEvaluation"));
                assertEquals(4, fts.rules().size());
                assertEquals("raster", fts.rules().get(0).getDescription().getTitle().toString());
                assertEquals("orange polygon", fts.rules().get(1).getDescription().getTitle()
                        .toString());
                assertEquals("orange line", fts.rules().get(2).getDescription().getTitle()
                        .toString());
                assertEquals("orange point", fts.rules().get(3).getDescription().getTitle()
                        .toString());
                
                // make sure it's valid
                SLDValidator validator = new SLDValidator();
                List errors = validator.validateSLD(new ByteArrayInputStream(input));
                assertEquals(0, errors.size());
                
                return null;
            }
        });

        Catalog cat = createNiceMock(Catalog.class);
        expect(cat.getFactory()).andReturn(new CatalogFactoryImpl(null)).anyTimes();
        expect(cat.getResourcePool()).andReturn(rp).anyTimes();

        WorkspaceInfo ws = createNiceMock(WorkspaceInfo.class);

        FeatureTypeInfo ft = createNiceMock(FeatureTypeInfo.class);
        expect(ft.getName()).andReturn("foo").anyTimes();

        replay(rp, ft, ws, cat);

        StyleGenerator gen = new StyleGenerator(cat) {
            protected void randomizeRamp() {
                // do not randomize for this test
            };
        };
        gen.setWorkspace(ws);

        SimpleFeatureType schema = DataUtilities.createType("foo", "geom:Geometry");
        StyleInfo style = gen.createStyle(new SLDHandler(), ft, schema);
        assertNotNull(style);
        assertNotNull(style.getWorkspace());
    }
    
    @Test
    public void testRasterStyle() throws Exception {
        ResourcePool rp = createNiceMock(ResourcePool.class);
        rp.writeStyle((StyleInfo) anyObject(), (InputStream) anyObject());
        expectLastCall().andAnswer(new IAnswer<Void>() {

            @Override
            public Void answer() throws Throwable {
                Object[] args = getCurrentArguments();
                InputStream is = (InputStream) args[1];
                byte[] input = IOUtils.toByteArray(is);
                
                SLDParser parser = new SLDParser(CommonFactoryFinder.getStyleFactory());
                parser.setInput(new ByteArrayInputStream(input));
                StyledLayerDescriptor sld = parser.parseSLD();
                
                NamedLayer nl = (NamedLayer) sld.getStyledLayers()[0];
                assertEquals("foo", nl.getName());
                Style style = nl.getStyles()[0];
                assertEquals("A raster style", style.getDescription().getTitle()
                        .toString());
                assertEquals(1, style.featureTypeStyles().size());
                FeatureTypeStyle fts = style.featureTypeStyles().get(0);
                assertEquals(1, fts.rules().size());
                assertThat(fts.rules().get(0).symbolizers().get(0), instanceOf(RasterSymbolizer.class));
                
                // make sure it's valid
                SLDValidator validator = new SLDValidator();
                List errors = validator.validateSLD(new ByteArrayInputStream(input));
                assertEquals(0, errors.size());
                
                
                return null;
            }
        });

        Catalog cat = createNiceMock(Catalog.class);
        expect(cat.getFactory()).andReturn(new CatalogFactoryImpl(null)).anyTimes();
        expect(cat.getResourcePool()).andReturn(rp).anyTimes();

        WorkspaceInfo ws = createNiceMock(WorkspaceInfo.class);

        CoverageInfo ci = createNiceMock(CoverageInfo.class);
        expect(ci.getName()).andReturn("foo").anyTimes();

        replay(rp, ci, ws, cat);

        StyleGenerator gen = new StyleGenerator(cat) {
            protected void randomizeRamp() {
                // do not randomize for this test
            };
        };
        gen.setWorkspace(ws);

        StyleInfo style = gen.createStyle(new SLDHandler(), ci);
        assertNotNull(style);
        assertNotNull(style.getWorkspace());
    }
}