MBStyleTestUtils.java

/*
 *    GeoTools - The Open Source Java GIS Toolkit
 *    http://geotools.org
 *
 *    (C) 2017, Open Source Geospatial Foundation (OSGeo)
 *
 *    This library is free software; you can redistribute it and/or
 *    modify it under the terms of the GNU Lesser General Public
 *    License as published by the Free Software Foundation;
 *    version 2.1 of the License.
 *
 *    This library is distributed in the hope that it will be useful,
 *    but WITHOUT ANY WARRANTY; without even the implied warranty of
 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 *    Lesser General Public License for more details.
 */

package org.geotools.mbstyle.parse;

import java.util.Set;

import org.geotools.mbstyle.parse.MBFunction.FunctionCategory;
import org.opengis.filter.expression.Expression;

import org.hamcrest.BaseMatcher;
import org.hamcrest.Description;
import org.hamcrest.Matcher;
import org.hamcrest.Matchers;

public class MBStyleTestUtils {
    public static Matcher<MBFunction> categories(Matcher<? super Set<FunctionCategory>> matcher) {
        return new BaseMatcher<MBFunction>() {
            
            @Override
            public boolean matches(Object item) {
                if(item instanceof MBFunction) {
                    return matcher.matches(((MBFunction) item).category());
                } else {
                    return false;
                }
            }
            
            @Override
            public void describeTo(Description description) {
                description
                    .appendText("MBFunction with categories ")
                    .appendDescriptionOf(matcher);
            }
            
            @Override
            public void describeMismatch(Object item, Description description) {
                if(item instanceof MBFunction) {
                    description.appendText("categories were ");
                    description.appendValueList("[", ", ", "]", ((MBFunction) item).category());
                } else {
                    description.appendText("was a ").appendValue(item.getClass().getName());
                }
            }
            
        };
    }
    
    public static Matcher<Expression> evaluatesTo(Object object, Matcher<?> resultMatcher) {
        return new BaseMatcher<Expression>() {
            
            @Override
            public boolean matches(Object item) {
                if(item instanceof Expression) {
                    return resultMatcher.matches(((Expression) item).evaluate(object));
                } else {
                    return false;
                }
            }
            
            @Override
            public void describeTo(Description description) {
                description
                    .appendText("expression evaluating ")
                    .appendValue(object)
                    .appendText(" to ")
                    .appendDescriptionOf(resultMatcher);
            }
            
            @Override
            public void describeMismatch(Object item, Description description) {
                if(item instanceof Expression) {
                    Object result = ((Expression) item).evaluate(object);
                    description
                        .appendValue(item)
                        .appendText(" evaluates to ")
                        .appendValue(result);
                } else {
                    description.appendText("was a ").appendValue(item.getClass().getName());
                }
            }
            
            
        };
    }
    
    public static Matcher<Number> asInt(Matcher<Integer> matcher) {
        return new BaseMatcher<Number>() {
            
            @Override
            public boolean matches(Object item) {
                if(item instanceof Number) {
                    return matcher.matches(((Long) item).intValue());
                } else {
                    return false;
                }
            }
            
            @Override
            public void describeTo(Description description) {
                description.appendDescriptionOf(matcher);
            }
            
            @Override
            public void describeMismatch(Object item, Description description) {
                matcher.describeMismatch(item, description);
            }
            
        };
    }
    
    public static Matcher<Number> equalInt(Integer value) {
        return asInt(Matchers.equalTo(value));
    }
    
    public static <T> Matcher<Expression> evaluatesTo(Object object, Class<T> context, Matcher<? super T> resultMatcher) {
        return new BaseMatcher<Expression>() {
            
            @Override
            public boolean matches(Object item) {
                if(item instanceof Expression) {
                    return resultMatcher.matches(((Expression) item).evaluate(object, context));
                } else {
                    return false;
                }
            }
            
            @Override
            public void describeTo(Description description) {
                description
                    .appendText("expression evaluating ")
                    .appendValue(object)
                    .appendText(" to ")
                    .appendDescriptionOf(resultMatcher);
            }
            
            @Override
            public void describeMismatch(Object item, Description description) {
                if(item instanceof Expression) {
                    Object result = ((Expression) item).evaluate(object);
                    description
                        .appendValue(item)
                        .appendText(" evaluates to ")
                        .appendValue(result);
                } else {
                    description.appendText("was a ").appendValue(item.getClass().getName());
                }
            }
            
            
        };
    }
}