OwsMapContextTest.java

/**
 * OrbisGIS is a java GIS application dedicated to research in GIScience.
 * OrbisGIS is developed by the GIS group of the DECIDE team of the 
 * Lab-STICC CNRS laboratory, see <http://www.lab-sticc.fr/>.
 *
 * The GIS group of the DECIDE team is located at :
 *
 * Laboratoire Lab-STICC – CNRS UMR 6285
 * Equipe DECIDE
 * UNIVERSITÉ DE BRETAGNE-SUD
 * Institut Universitaire de Technologie de Vannes
 * 8, Rue Montaigne - BP 561 56017 Vannes Cedex
 * 
 * OrbisGIS is distributed under GPL 3 license.
 *
 * Copyright (C) 2007-2014 CNRS (IRSTV FR CNRS 2488)
 * Copyright (C) 2015-2017 CNRS (Lab-STICC UMR CNRS 6285)
 *
 * This file is part of OrbisGIS.
 *
 * OrbisGIS 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 3 of the License, or (at your option) any later
 * version.
 *
 * OrbisGIS 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
 * OrbisGIS. If not, see <http://www.gnu.org/licenses/>.
 *
 * For more information, please consult: <http://www.orbisgis.org/>
 * or contact directly:
 * info_at_ orbisgis.org
 */
package org.orbisgis.coremap;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.sql.Connection;
import java.sql.Statement;
import java.util.Locale;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;

import org.h2gis.functions.io.shp.SHPDriverFunction;
import org.h2gis.functions.factory.H2GISDBFactory;
import org.h2gis.functions.factory.H2GISFunctions;
import org.h2gis.api.EmptyProgressVisitor;
import org.h2gis.utilities.TableLocation;
import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Test;
import org.orbisgis.corejdbc.DataManager;
import org.orbisgis.corejdbc.internal.DataManagerImpl;
import org.orbisgis.coremap.layerModel.ILayer;
import org.orbisgis.coremap.layerModel.MapContext;
import org.orbisgis.coremap.layerModel.OwsMapContext;
import org.orbisgis.coremap.renderer.se.common.Description;
import org.orbisgis.commons.progress.NullProgressMonitor;

import javax.sql.DataSource;

/**
 *
 */
public class OwsMapContextTest  {
    private static Connection connection;
    private static DataManager dataManager;

    @BeforeClass
    public static void tearUpClass() throws Exception {
        DataSource dataSource = H2GISDBFactory.createDataSource(OwsMapContextTest.class.getSimpleName(), false);
        connection = dataSource.getConnection();
        H2GISFunctions.load(connection);
        dataManager = new DataManagerImpl(dataSource);
    }

    @AfterClass
    public static void tearDownClass() throws Exception {
        connection.close();
        dataManager.dispose();
    }

    private DataManager getDataManager() {
        return dataManager;
    }

    private Connection getConnection() {
        return connection;
    }

    private MapContext mc = new OwsMapContext(dataManager);

        
    @Test
	public void testRemoveSelectedLayer() throws Exception {
		MapContext mc = new OwsMapContext(getDataManager());
		mc.open(null);
		ILayer layer = mc.createLayer(getDataManager().registerDataSource(OwsMapContextTest.class.getResource("../../../data/bv_sap.shp").toURI()));
		mc.getLayerModel().addLayer(layer);
		mc.setSelectedLayers(new ILayer[] { layer });
		assertTrue(mc.getSelectedLayers().length == 1);
		assertTrue(mc.getSelectedLayers()[0] == layer);
		mc.getLayerModel().remove(layer);
		assertTrue(mc.getSelectedLayers().length == 0);
		mc.close(null);
	}

    @Test
    public void testTitleAndDescriptionMapContext() throws Exception {
        String title = "Map of Ankh-Morpork";
        Locale locale = Locale.UK;
        String mapAbstract = "The principal city of the Sto Plains";
        // Define the map description
        Description mapDescription = new Description();
        mapDescription.addTitle(locale, title);
        mapDescription.addAbstract(locale, mapAbstract);
        MapContext mc = new OwsMapContext(getDataManager());
        mc.open(null);
        mc.setDescription(mapDescription);
        mc.close(null);
        ByteArrayOutputStream mapData = new ByteArrayOutputStream();
        mc.write(mapData);
        // Map data contain the serialisation
        // Read this data with another instance
        MapContext mc2 = new OwsMapContext(getDataManager());
        mc2.read(new ByteArrayInputStream(mapData.toByteArray()));
        mc2.open(null);
        // Test default title
        assertTrue(mc2.getTitle().equals(title));
        // Test the title with the provided locale
        assertTrue(mc2.getDescription().getTitle(locale).equals(title));
        // Test the abstract with the provided locale
        assertTrue(mc2.getDescription().getAbstract(locale).equals(mapAbstract));
        mc2.close(null);
    }

    @Test
    public void makeLayerUriFromTableFile() throws Exception {
        File dataFile = new File(OwsMapContextTest.class.getResource("../../../data/landcover2000.shp").getFile()).getCanonicalFile();
        File mapContextLocation = new File("landco_db.ows");
        String tableReference = getDataManager().registerDataSource(dataFile.toURI());
        MapContext mc = new OwsMapContext(getDataManager());
        mc.setLocation(mapContextLocation.toURI());
        mc.open(new NullProgressMonitor());
        ILayer layer = mc.createLayer(tableReference);
        mc.getLayerModel().addLayer(layer);
        mc.close(new NullProgressMonitor());
        try(FileOutputStream out = new FileOutputStream(mapContextLocation)) {
            mc.write(out);
        }
        // Open the map context
        MapContext newMapContext = new OwsMapContext(getDataManager());
        newMapContext.setLocation(mapContextLocation.toURI());
        try(FileInputStream in = new FileInputStream(mapContextLocation)) {
            newMapContext.read(in);
        }
        newMapContext.open(new NullProgressMonitor());
        assertEquals(1, newMapContext.getLayers().length);
        assertEquals(dataFile.toURI(), newMapContext.getLayers()[0].getDataUri());
    }

    @Test
    public void makeLayerUriFromNativeTable() throws Exception {
        File dataFile = new File(OwsMapContextTest.class.getResource("../../../data/landcover2000.shp").getFile());
        File mapContextLocation = new File("landco_db2.ows");
        SHPDriverFunction shpDriver = new SHPDriverFunction();
        try(Connection connection = getConnection();
            Statement st = connection.createStatement()) {
            st.execute("DROP TABLE IF EXISTS LANDCOVER2000");
            shpDriver.importFile(connection, "LANDCOVER2000", dataFile, new EmptyProgressVisitor());
        }
        MapContext mc = new OwsMapContext(getDataManager());
        mc.setLocation(mapContextLocation.toURI());
        mc.open(new NullProgressMonitor());
        ILayer layer = mc.createLayer("LANDCOVER2000");
        mc.getLayerModel().addLayer(layer);
        mc.close(new NullProgressMonitor());
        try(FileOutputStream out = new FileOutputStream(mapContextLocation)) {
            mc.write(out);
        }
        // Open the map context
        MapContext newMapContext = new OwsMapContext(getDataManager());
        newMapContext.setLocation(mapContextLocation.toURI());
        try(FileInputStream in = new FileInputStream(mapContextLocation)) {
            newMapContext.read(in);
        }
        newMapContext.open(new NullProgressMonitor());
        assertEquals(1, newMapContext.getLayers().length);
        assertTrue(newMapContext.getLayers()[0].getDataUri().toString().contains("table=LANDCOVER2000"));
        assertEquals(TableLocation.parse("LANDCOVER2000").getTable(), TableLocation.parse(newMapContext.getLayers()[0].getTableReference()).getTable());
    }
}