SOSSoapRequestTest.java

/*
 *    Constellation - An open source and standard compliant SDI
 *    http://www.constellation-sdi.org
 *
 * Copyright 2014 Geomatys.
 *
 * 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 org.constellation.ws.embedded;

import org.constellation.business.IServiceBusiness;
import org.constellation.configuration.ConfigDirectory;
import org.constellation.admin.SpringHelper;
import org.constellation.configuration.DataSourceType;
import org.constellation.configuration.SOSConfiguration;
import org.constellation.generic.database.Automatic;
import org.constellation.generic.database.BDD;
import org.constellation.sos.ws.soap.SOService;
import org.constellation.test.utils.Order;
import org.constellation.test.utils.SpringTestRunner;
import org.constellation.util.Util;
import org.constellation.test.utils.CstlDOMComparator;
import org.junit.AfterClass;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.slf4j.LoggerFactory;
import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.test.context.ContextConfiguration;

import javax.annotation.PostConstruct;
import javax.inject.Inject;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.InputStream;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLConnection;
import java.sql.Connection;
import java.util.HashMap;
import java.util.Map;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import org.apache.sis.internal.storage.IOUtilities;
import org.geotoolkit.internal.sql.DefaultDataSource;
import org.geotoolkit.util.FileUtilities;
import org.geotoolkit.util.sql.DerbySqlScriptRunner;

import static org.junit.Assume.assumeNoException;
import org.junit.BeforeClass;
import org.springframework.test.context.ActiveProfiles;

/**
 *
 * @author Guilhem Legal (Geomatys)
 */
@RunWith(SpringTestRunner.class)
@ContextConfiguration("classpath:/cstl/spring/test-context.xml")
@ActiveProfiles({"standard"})
public class SOSSoapRequestTest extends AbstractGrizzlyServer implements ApplicationContextAware {

    private static final org.slf4j.Logger LOGGER = LoggerFactory.getLogger("org.constellation.ws.embedded");

    protected ApplicationContext applicationContext;
    
    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        this.applicationContext = applicationContext;
    }

    private static final String SOS_DEFAULT = "http://localhost:9191/sos/default";

    @Inject
    private IServiceBusiness serviceBusiness;
    
    private static boolean initialized = false;
    
    private static DefaultDataSource ds = null;
    
    private static File configDirectory;
    
    @BeforeClass
    public static void initTestDir() {
        configDirectory = ConfigDirectory.setupTestEnvironement("SOSSoapRequestTest");
    }
    
    @PostConstruct
    public void initLayerList() {
        SpringHelper.setApplicationContext(applicationContext);
        if (!initialized) {
            try {
                try {
                    serviceBusiness.deleteAll();
                } catch (Exception ex) {
                    LOGGER.warn(ex.getMessage());
                }

                final File dataDirectory = new File(configDirectory, "dataSos");
                dataDirectory.mkdir();

                writeDataFile(dataDirectory, "urn-ogc-object-sensor-SunSpot-0014.4F01.0000.261A");
                writeDataFile(dataDirectory, "urn-ogc-object-sensor-SunSpot-0014.4F01.0000.2626");
                writeDataFile(dataDirectory, "urn-ogc-object-sensor-SunSpot-2");

                String url = "jdbc:derby:memory:OM2Test1;create=true";
                ds = new DefaultDataSource(url);

                Connection con = ds.getConnection();

                DerbySqlScriptRunner sr = new DerbySqlScriptRunner(con);
                sr.setEncoding("UTF-8");
                String sql = FileUtilities.getStringFromStream(Util.getResourceAsStream("org/constellation/om2/structure_observations.sql"));
                sql = sql.replace("$SCHEMA", "");
                sr.run(sql);
                sr.run(Util.getResourceAsStream("org/constellation/sql/sos-data-om2.sql"));
        
                final Automatic smlConfig = new Automatic(null, dataDirectory.getPath());
                final Automatic omCOnfig = new Automatic(null, new BDD("", url, "test", "test"));
                final SOSConfiguration sosconf = new SOSConfiguration(smlConfig, omCOnfig);
                sosconf.setObservationFilterType(DataSourceType.OM2);
                sosconf.setObservationReaderType(DataSourceType.OM2);
                sosconf.setObservationWriterType(DataSourceType.OM2);
                sosconf.setSMLType(DataSourceType.FILESYSTEM);
                sosconf.setProfile("transactional");
                sosconf.setObservationIdBase("urn:ogc:object:observation:SunSpot:");
                sosconf.setSensorIdBase("urn:ogc:object:sensor:SunSpot:");
                sosconf.setPhenomenonIdBase("urn:phenomenon:");
                sosconf.setObservationTemplateIdBase("urn:ogc:object:observationTemplate:SunSpot:");
                sosconf.setVerifySynchronization(false);

                serviceBusiness.create("sos", "default", sosconf, null);

                final Map<String, Object> map = new HashMap<>();
                map.put("sos", new SOService());
                initServer(null, map);
                initialized = true;
            } catch (Exception ex) {
               LOGGER.error(ex.getMessage(), ex);
            }
        }
    }

    @AfterClass
    public static void shutDown() {
        try {
            final IServiceBusiness service = SpringHelper.getBean(IServiceBusiness.class);
            if (service != null) {
                service.deleteAll();
            }
        } catch (Exception ex) {
            LOGGER.error(ex.getMessage(), ex);
        }
        ConfigDirectory.shutdownTestEnvironement("SOSSoapRequestTest");
        if (ds != null) {
            ds.shutdown();
        }
        finish();
    }

    /**
     */
    @Test
    @Order(order=1)
    public void testSOSGetCapabilities() throws Exception {

        waitForSoapStart("sos/default", 200);
        
        // Creates a valid GetCapabilities url.
        URL getCapsUrl;
        try {
            getCapsUrl = new URL(SOS_DEFAULT);
        } catch (MalformedURLException ex) {
            assumeNoException(ex);
            return;
        }

        URLConnection conec = getCapsUrl.openConnection();
        postRequestFile(conec, "org/constellation/xml/sos/GetCapabilitiesSOAP.xml", "application/soap+xml");

        String result    = getStringResponse(conec);
        String expResult = getStringFromFile("org/constellation/xml/sos/GetCapabilitiesResponseSOAP.xml");
        
        // try to fix an error with gml 
        String gmlPrefix = null;
        final Pattern p  = Pattern.compile("xmlns[^\"]+\"[^\"]+\"");
        final Matcher m  = p.matcher(result) ;  
        while (m.find()) {
            String s = m.group();
            String namespace = s.substring(s.indexOf('"') + 1, s.length() - 1);
            String prefix = s.substring(6, s.indexOf('='));
            if (namespace.equals("http://www.opengis.net/gml")) {
                gmlPrefix = prefix;
            }
        }

        if (gmlPrefix != null) {
            LOGGER.info("GML Prefix found:{0}", gmlPrefix);
            if (!gmlPrefix.equals("gml")) {
                result = result.replace(gmlPrefix + ':', "gml:");
                result = result.replace("xmlns:" + gmlPrefix + '=', "xmlns:gml=");
                result = result.replace("xmlns:gml=\"http://www.opengis.net/gml/3.2\"", "");
            }
        } else {
            LOGGER.info("No GML Prefix found.");
        }

        domCompare(result, expResult);
    }

    @Test
    @Order(order=2)
    public void testSOSGetFeatureOfInterest() throws Exception {
        // Creates a valid GetObservation url.
        final URL getCapsUrl = new URL(SOS_DEFAULT);

        // for a POST request
        URLConnection conec = getCapsUrl.openConnection();

        postRequestFile(conec, "org/constellation/xml/sos/GetFeatureOfInterestSOAP.xml", "application/soap+xml");

        String result    = getStringResponse(conec);
        String expResult = getStringFromFile("org/constellation/xml/sos/GetFeatureOfInterestResponseSOAP.xml");

        System.out.println("result:\n" + result);

        domCompare(expResult, result);

        conec = getCapsUrl.openConnection();

        postRequestFile(conec, "org/constellation/xml/sos/GetFeatureOfInterestSOAP2.xml", "application/soap+xml");

        result    = getStringResponse(conec);
        expResult = getStringFromFile("org/constellation/xml/sos/GetFeatureOfInterestResponseSOAP2.xml");


        // System.out.println("result:\n" + result);

        domCompare(result, expResult);
    }

    public static void writeDataFile(File dataDirectory, String resourceName) throws IOException {

        final File dataFile;
        if (System.getProperty("os.name", "").startsWith("Windows")) {
            final String windowsIdentifier = resourceName.replace(':', '-');
            dataFile = new File(dataDirectory, windowsIdentifier + ".xml");
        } else {
            dataFile = new File(dataDirectory, resourceName + ".xml");
        }
        FileWriter fw = new FileWriter(dataFile);
        InputStream in = Util.getResourceAsStream("org/constellation/embedded/test/" + resourceName + ".xml");

        byte[] buffer = new byte[1024];
        int size;

        while ((size = in.read(buffer, 0, 1024)) > 0) {
            fw.write(new String(buffer, 0, size));
        }
        in.close();
        fw.close();
    }
    
     public static void domCompare(final Object actual, final Object expected) throws Exception {

        final CstlDOMComparator comparator = new CstlDOMComparator(expected, actual);
        comparator.ignoredAttributes.add("http://www.w3.org/2000/xmlns:*");
        comparator.ignoredAttributes.add("updateSequence");
        comparator.ignoredAttributes.add("http://www.w3.org/2001/XMLSchema-instance:schemaLocation");
        comparator.compare();
    }
}