de.ingrid.iplug.dsc.index.ScriptedWmsDocumentProducerTest.java Source code

Introduction

Here is the source code for de.ingrid.iplug.dsc.index.ScriptedWmsDocumentProducerTest.java

Source

/*
 * **************************************************-
 * ingrid-iplug-dsc-mapclient
 * ==================================================
 * Copyright (C) 2014 - 2016 wemove digital solutions GmbH
 * ==================================================
 * Licensed under the EUPL, Version 1.1 or  as soon they will be
 * approved by the European Commission - subsequent versions of the
 * EUPL (the "Licence");
 * 
 * You may not use this work except in compliance with the Licence.
 * You may obtain a copy of the Licence at:
 * 
 * http://ec.europa.eu/idabc/eupl5
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the Licence is distributed on an "AS IS" basis,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the Licence for the specific language governing permissions and
 * limitations under the Licence.
 * **************************************************#
 */
package de.ingrid.iplug.dsc.index;

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

import junit.framework.TestCase;

import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import org.springframework.core.io.FileSystemResource;

import de.ingrid.admin.elasticsearch.StatusProvider;
import de.ingrid.iplug.dsc.om.SourceRecord;
import de.ingrid.iplug.dscmapclient.index.mapper.IRecordMapper;
import de.ingrid.iplug.dscmapclient.index.mapper.ScriptedIdfDocumentMapper;
import de.ingrid.iplug.dscmapclient.index.mapper.ScriptedWmsDocumentMapper;
import de.ingrid.iplug.dscmapclient.index.producer.DscWmsDocumentProducer;
import de.ingrid.iplug.dscmapclient.index.producer.PlugDescriptionConfiguredWmsRecordSetProducer;
import de.ingrid.utils.ElasticDocument;
import de.ingrid.utils.PlugDescription;

public class ScriptedWmsDocumentProducerTest extends TestCase {

    @Mock
    StatusProvider statusProvider;

    public ScriptedWmsDocumentProducerTest() {
        super();
        MockitoAnnotations.initMocks(this);
    }

    public void testScriptedDatabaseDocumentProducer() throws Exception {

        PlugDescription pd = new PlugDescription();
        pd.put("WebmapXmlConfigFile", "src/test/resources/ingrid_webmap_client_config.xml");

        PlugDescriptionConfiguredWmsRecordSetProducer p = new PlugDescriptionConfiguredWmsRecordSetProducer();
        p.setStatusProvider(statusProvider);
        p.setIdTag("//capabilitiesUrl");
        p.configure(pd);

        ScriptedWmsDocumentMapper m = new ScriptedWmsDocumentMapper();
        m.setMappingScript(new FileSystemResource("src/main/resources/mapping/wms_to_lucene.js"));
        m.setCompile(false);
        ScriptedIdfDocumentMapper mapper = new ScriptedIdfDocumentMapper();
        mapper.setMappingScript(new FileSystemResource("src/main/resources/mapping/wms_to_idf.js"));
        mapper.setCompile(false);

        List<IRecordMapper> mList = new ArrayList<IRecordMapper>();
        mList.add(m);
        mList.add(mapper);
        DscWmsDocumentProducer dp = new DscWmsDocumentProducer();
        dp.setRecordSetProducer(p);
        dp.setRecordMapperList(mList);
        //TODO define proper directory
        if (dp.hasNext()) {
            while (dp.hasNext()) {
                ElasticDocument doc = dp.next();
                // NOTICE: may be null (e.g. if service not reachable) !
                if (doc == null) {
                    continue;
                }
                //writer.addDocument(doc);
            }
        } else {
            fail("No document produced");
        }

    }

    public void testMapper() throws Exception {

        PlugDescription pd = new PlugDescription();
        pd.put("WebmapXmlConfigFile", "src/test/resources/ingrid_webmap_client_config.xml");

        PlugDescriptionConfiguredWmsRecordSetProducer p = new PlugDescriptionConfiguredWmsRecordSetProducer();
        p.setStatusProvider(statusProvider);
        p.setIdTag("//capabilitiesUrl");
        p.configure(pd);

        ScriptedWmsDocumentMapper mapperLucene = new ScriptedWmsDocumentMapper();
        mapperLucene.setMappingScript(new FileSystemResource("src/main/resources/mapping/wms_to_lucene.js"));
        mapperLucene.setCompile(false);
        ScriptedIdfDocumentMapper mapperIdf = new ScriptedIdfDocumentMapper();
        mapperIdf.setMappingScript(new FileSystemResource("src/main/resources/mapping/wms_to_idf.js"));
        mapperIdf.setCompile(false);

        while (p.hasNext()) {
            ElasticDocument doc = new ElasticDocument();
            SourceRecord record = p.next();
            mapperLucene.map(record, doc);
            mapperIdf.map(record, doc);
        }

    }

}