org.fao.geonet.kernel.harvest.harvester.csw.CswHarvesterIntegrationTest.java Source code

Introduction

Here is the source code for org.fao.geonet.kernel.harvest.harvester.csw.CswHarvesterIntegrationTest.java

Source

/*
 * Copyright (C) 2001-2016 Food and Agriculture Organization of the
 * United Nations (FAO-UN), United Nations World Food Programme (WFP)
 * and United Nations Environment Programme (UNEP)
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or (at
 * your option) any later version.
 *
 * This program 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
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
 *
 * Contact: Jeroen Ticheler - FAO - Viale delle Terme di Caracalla 2,
 * Rome - Italy. email: geonetwork@osgeo.org
 */

package org.fao.geonet.kernel.harvest.harvester.csw;

import com.google.common.base.Predicate;
import com.google.common.collect.Maps;

import org.apache.http.client.methods.HttpPost;
import org.apache.http.client.methods.HttpRequestBase;
import org.fao.geonet.MockRequestFactoryGeonet;
import org.fao.geonet.csw.common.Csw;
import org.fao.geonet.kernel.harvest.AbstractHarvesterIntegrationTest;
import org.fao.geonet.utils.MockXmlRequest;
import org.fao.geonet.utils.Xml;
import org.jdom.Element;

import java.util.Map;

import javax.annotation.Nullable;

/**
 * Integration Test for the Csw Harvester class.
 *
 * User: Jesse Date: 10/18/13 Time: 4:01 PM
 */
public class CswHarvesterIntegrationTest extends AbstractHarvesterIntegrationTest {
    private static final String PROTOCOL = "http";
    private static final String HOST = "localhost";
    private static final int PORT = 8080;
    public static final String REQUEST = PROTOCOL + "://" + HOST + ":" + PORT + "/geonetwork/srv/eng/csw";
    public static final String CAPABILITIES_QUERY_STRING = "?service=CSW&request=GetCapabilities";
    public static final String CAPABILITIES_URL = REQUEST + CAPABILITIES_QUERY_STRING;
    public static final String OUTPUT_SCHEMA = "http://www.isotc211.org/2005/gmd";

    public CswHarvesterIntegrationTest() {
        super("csw");
    }

    public static void addCswSpecificParams(Element params, String outputSchema) {
        params.getChild("site").addContent(new Element("capabilitiesUrl").setText(CAPABILITIES_URL))
                .addContent(new Element("outputSchema").setText(outputSchema));
    }

    protected void mockHttpRequests(MockRequestFactoryGeonet bean) {
        final MockXmlRequest cswServerRequest = new MockXmlRequest(HOST, PORT, PROTOCOL);
        cswServerRequest.when(CAPABILITIES_URL).thenReturn(fileStream("capabilities.xml"));
        final String queryString = "?request=GetRecordById&service=CSW&version=2.0.2&outputSchema="
                + getOutputSchema() + "&elementSetName=full&id=";
        cswServerRequest.when(REQUEST + queryString + "7e926fbf-00fb-4ff5-a99e-c8576027c4e7")
                .thenReturn(fileStream("GetRecordById-7e926fbf-00fb-4ff5-a99e-c8576027c4e7.xml"));
        cswServerRequest.when(REQUEST + queryString + "da165110-88fd-11da-a88f-000d939bc5d8")
                .thenReturn(fileStream("GetRecordById-da165110-88fd-11da-a88f-000d939bc5d8.xml"));
        cswServerRequest.when(new Predicate<HttpRequestBase>() {
            @Override
            public boolean apply(@Nullable HttpRequestBase input) {

                final boolean isHttpPost = input instanceof HttpPost;
                final boolean correctPath = input.getURI().toString().startsWith(REQUEST);
                if (!correctPath) {
                    return false;
                }

                String request, typeNames, elementSetName;
                final boolean noQueryFilter;

                if (isHttpPost) {
                    final Element xml;
                    try {
                        xml = Xml.loadStream(((HttpPost) input).getEntity().getContent());
                    } catch (Throwable e) {
                        return false;
                    }
                    request = xml.getName();
                    final Element queryEl = xml.getChild("Query", Csw.NAMESPACE_CSW);
                    typeNames = queryEl.getAttributeValue("typeNames");
                    elementSetName = queryEl.getChild("ElementSetName", Csw.NAMESPACE_CSW).getText();
                    noQueryFilter = queryEl.getChildren().size() == 1;
                } else {
                    final String[] params = input.getURI().getQuery().split("\\&");
                    Map<String, String> paramMap = Maps.newHashMap();
                    for (String param : params) {
                        final String[] split = param.split("=");
                        String key = split[0].toLowerCase();
                        String value = "";
                        if (split.length > 1) {
                            value = split[1];
                        }
                        paramMap.put(key, value);
                    }

                    request = paramMap.get("request");
                    typeNames = paramMap.get("typenames");
                    elementSetName = paramMap.get("elementsetname");
                    noQueryFilter = paramMap.get("query") == null || paramMap.get("query").isEmpty();
                }

                final boolean isGetRecords = "GetRecords".equalsIgnoreCase(request);
                final boolean correctTypeNames = typeNames.contains("gmd:MD_Metadata")
                        || typeNames.contains("csw:Record");
                final boolean isSummary = elementSetName.equals("summary");
                return isGetRecords && correctTypeNames && isSummary && noQueryFilter;
            }
        }).thenReturn(fileStream("getRecords.xml"));

        bean.registerRequest(true, HOST, PORT, PROTOCOL, cswServerRequest);
        bean.registerRequest(true, null, -1, PROTOCOL, cswServerRequest);
    }

    protected void customizeParams(Element params) {
        addCswSpecificParams(params, getOutputSchema());
    }

    @Override
    protected int getExpectedAdded() {
        return 2;
    }

    @Override
    protected int getExpectedTotalFound() {
        return 2;
    }

    public String getOutputSchema() {
        return OUTPUT_SCHEMA;
    }
}