Application: gvSIG desktop » gvSIG tools: Vector editing

/*

 * Copyright 2014 DiSiD Technologies S.L.L. All rights reserved.

 * 

 * Project  : DiSiD org.gvsig.vectorediting.lib.prov.polyline 

 * SVN Id   : $Id$

 */

package org.gvsig.vectorediting.lib.prov.polyline;

import java.io.File;

import java.net.URL;

import java.util.List;

import org.gvsig.fmap.crs.CRSFactory;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataStoreParameters;

import org.gvsig.fmap.dal.feature.FeatureStore;

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.fmap.geom.GeometryManager;

import org.gvsig.fmap.geom.aggregate.MultiCurve;

import org.gvsig.fmap.geom.primitive.Curve;

import org.gvsig.fmap.geom.primitive.Line;

import org.gvsig.fmap.geom.primitive.Point;

import org.gvsig.tools.junit.AbstractLibraryAutoInitTestCase;

import org.gvsig.vectorediting.lib.api.DrawingStatus;

import org.gvsig.vectorediting.lib.api.EditingService;

import org.gvsig.vectorediting.lib.api.EditingServiceParameter;

import org.gvsig.vectorediting.lib.api.exceptions.FinishServiceException;

import org.gvsig.vectorediting.lib.api.exceptions.InvalidEntryException;

import org.gvsig.vectorediting.lib.api.exceptions.StartServiceException;

import org.gvsig.vectorediting.lib.api.exceptions.StopServiceException;

import org.gvsig.vectorediting.lib.impl.DefaultEditingManager;

/**

 * Polyline provider test

 * 

 * This test uses src/test/resources/cartography

 * 

 * @author lmarques

 *

 */

public class PolylineTest extends AbstractLibraryAutoInitTestCase {

    DefaultEditingManager editingManager;

    GeometryManager geoManager;

    FeatureStore featureStoreTest;

    EditingService service;

    DataManager dataManager;

    @Override

    protected void doSetUp() throws Exception {

        editingManager = new DefaultEditingManager();

        geoManager = GeometryLocator.getGeometryManager();

        try {

            featureStoreTest = openTestFeatureStore();

        } catch (Exception e) {

            fail("Unexpected exception opening test feature store: "

                + e.getMessage());

        }

        service =

            editingManager.getEditingService(

                PolylineEditingProviderFactory.PROVIDER_NAME, featureStoreTest);

        service.start();

    }

    public void testFinish() {

        // Finish without values

        try {

            service.finish();

            fail("No exception has been thrown");

        } catch (NullPointerException e) {

            // Test OK, got expected exception

        } catch (Exception e) {

            fail("Unexpected exception finishing service: " + e.getMessage());

        }

        // Finish with one value

        Point point = null;

        try {

            point = geoManager.createPoint(0, 0, Geometry.SUBTYPES.GEOM2D);

            service.setValue(point);

            service.finish();

            fail("No exception has been thrown");

        } catch (NullPointerException e) {

            // Test ok, got expected exception

        } catch (Exception e) {

            fail("Unexpected exception has been thrown endig service. "

                + e.getMessage());

        }

        // Finish with two values mode line

        Point point2 = null;

        try {

            point2 = geoManager.createPoint(10, 10, Geometry.SUBTYPES.GEOM2D);

            service.setValue(point2);

            Geometry geom = service.finish();

            assertTrue(geom instanceof MultiCurve);

            MultiCurve multicurve = (MultiCurve) geom;

            Curve curve = multicurve.getCurveAt(0);

            assertTrue(curve.getVertex(0).getX() == point.getX());

            assertTrue(curve.getVertex(0).getY() == point.getY());

            assertTrue(curve.getVertex(1).getX() == point2.getX());

            assertTrue(curve.getVertex(1).getY() == point2.getY());

        } catch (Exception e) {

            fail("Unexpected exception has been thrown ending service. "

                + e.getMessage());

        }

        // Restart service

        restartService();

        // Finish with three aligned values mode arc

        Point point3;

        try {

            point3 = geoManager.createPoint(20, 20, Geometry.SUBTYPES.GEOM2D);

            service.setValue(point);

            service.setValue(point2);

            service.setValue("A");

            service.setValue(point3);

            service.finish();

        } catch (FinishServiceException e) {

            // Test ok, got expected exception

        } catch (Exception e) {

            fail("Unexpected exception has been thrown ending service. "

                + e.getMessage());

        }

        // Restart service

        restartService();

        // Finish with random values

        try {

            for (int i = 0; i < 200; i++) {

                point =

                    geoManager.createPoint(Math.random() * 10000,

                        Math.random() * 10000, Geometry.SUBTYPES.GEOM2D);

                service.setValue(point);

                if (i > 3) {

                    if ((int) (Math.random() * 2) > 0) {

                        service.setValue("A");

                    } else {

                        service.setValue("L");

                    }

                }

            }

            service.finish();

        } catch (Exception e) {

            fail("Unexpected exception has been thrown ending service. "

                + e.getMessage());

        }

    }

    public void testfinishAndStore() {

        // Finish without values

        try {

            service.finishAndStore();

            fail("No exception has been thrown");

        } catch (NullPointerException e) {

            // Test OK, got expected exception

        } catch (Exception e) {

            fail("Unexpected exception finishing service: " + e.getMessage());

        }

        // Finish with one value

        Point point = null;

        try {

            point = geoManager.createPoint(0, 0, Geometry.SUBTYPES.GEOM2D);

            service.setValue(point);

            service.finishAndStore();

            fail("No exception has been thrown");

        } catch (NullPointerException e) {

            // Test ok, got expected exception

        } catch (Exception e) {

            fail("Unexpected exception has been thrown endig service. "

                + e.getMessage());

        }

        // Finish with two values mode line

        Point point2 = null;

        try {

            point2 = geoManager.createPoint(10, 10, Geometry.SUBTYPES.GEOM2D);

            service.setValue(point2);

            featureStoreTest.edit();

            service.finishAndStore();

            featureStoreTest.finishEditing();

        } catch (Exception e) {

            fail("Unexpected exception has been thrown ending service. "

                + e.getMessage());

        }

        // Restart service

        restartService();

        // Finish with three aligned values mode arc

        Point point3;

        try {

            point3 = geoManager.createPoint(20, 20, Geometry.SUBTYPES.GEOM2D);

            service.setValue(point);

            service.setValue(point2);

            service.setValue("A");

            service.setValue(point3);

            service.finishAndStore();

        } catch (FinishServiceException e) {

            // Test ok, got expected exception

        } catch (Exception e) {

            fail("Unexpected exception has been thrown ending service. "

                + e.getMessage());

        }

        // Restart service

        restartService();

        // Finish with random values

        try {

            for (int i = 0; i < 200; i++) {

                point =

                    geoManager.createPoint(Math.random() * 10000,

                        Math.random() * 10000, Geometry.SUBTYPES.GEOM2D);

                service.setValue(point);

                if (i > 3) {

                    if ((int) (Math.random() * 2) > 0) {

                        service.setValue("A");

                    } else {

                        service.setValue("L");

                    }

                }

            }

            featureStoreTest.edit();

            service.finishAndStore();

            featureStoreTest.finishEditing();

        } catch (Exception e) {

            fail("Unexpected exception has been thrown ending service. "

                + e.getMessage());

        }

    }

    public void testgetDrawingStatus() {

        // Without values and mouse point null

        try {

            service.getDrawingStatus(null);

        } catch (NullPointerException e) {

            // Testk ok, got expected exception

        } catch (Exception e) {

            fail("Unexpected exception has been thrown getting drawing status. "

                + e.getMessage());

        }

        // Without values and valid mouse point

        Point mousePoint = null;

        try {

            mousePoint =

                geoManager.createPoint(20, 50, Geometry.SUBTYPES.GEOM2D);

            DrawingStatus drawingStatus = service.getDrawingStatus(mousePoint);

            assertNull(drawingStatus);

        } catch (Exception e) {

            fail("Unexpected exception has been thrown getting drawing status. "

                + e.getMessage());

        }

        // One valid values and valid mouse position

        Point point1;

        try {

            point1 =

                geoManager.createPoint(20, 50, Geometry.SUBTYPES.GEOM2D);

            service.setValue(point1);

            DrawingStatus drawingStatus = service.getDrawingStatus(mousePoint);

            assertNotNull(drawingStatus);

            assertTrue(drawingStatus.getGeometries().size() == 1);

            Geometry geometry = drawingStatus.getGeometries().get(0);

            assertTrue(geometry instanceof Line);

            Line line = (Line) geometry;

            assertTrue(line.getVertex(0).getX() == point1.getX());

            assertTrue(line.getVertex(0).getY() == point1.getY());

            assertTrue(line.getVertex(1).getX() == point1.getX());

            assertTrue(line.getVertex(1).getY() == point1.getY());

        } catch (Exception e) {

            fail("Unexpected exception has been thrown getting drawing status. "

                + e.getMessage());

        }

    }

    public void testgetName() {

        assertEquals(PolylineEditingProviderFactory.PROVIDER_NAME,

            service.getName());

    }

    public void testgetParameters() {

        List<EditingServiceParameter> parameters = service.getParameters();

        assertTrue(parameters.size() == 1);

        assertTrue(parameters.get(0).getTypes()

            .contains(EditingServiceParameter.TYPE.LIST_POSITIONS));

    }

    public void testNext() {

        assertNotNull(service.next());

        List<EditingServiceParameter> parameters = service.getParameters();

        assertEquals(parameters.get(0), service.next());

        for (int i = 0; i < 10; i++) {

            try {

                Point point =

                    geoManager.createPoint(0, 0, Geometry.SUBTYPES.GEOM2D);

                service.setValue(point);

            } catch (Exception e) {

                fail("Unexpected exception creating point or setting value. "

                    + e.getMessage());

            }

        }

        assertEquals(parameters.get(0), service.next());

    }

    public void testsetValue() {

        // Valid point

        try {

            Point point =

                geoManager.createPoint(0, 0, Geometry.SUBTYPES.GEOM2D);

            service.setValue(point);

        } catch (Exception e) {

            fail("Unexpected exception setting value. " + e.getMessage());

        }

        // Invalid value

        try {

            Point point = null;

            service.setValue(point);

        } catch (InvalidEntryException e) {

            // Test ok, got expected value

        } catch (Exception e) {

            fail("Unexpected exception setting value. " + e.getMessage());

        }

    }

    public void testStart() {

        service =

            editingManager.getEditingService(

                PolylineEditingProviderFactory.PROVIDER_NAME, featureStoreTest);

        try {

            service.start();

        } catch (StartServiceException e) {

            fail("Unexpected exception starting service: " + e.getMessage());

        }

    }

    public void testStop() {

        service =

            editingManager.getEditingService(

                PolylineEditingProviderFactory.PROVIDER_NAME, featureStoreTest);

        try {

            service.start();

            service.stop();

        } catch (StartServiceException e) {

            fail("Unexpected exception starting service: " + e.getMessage());

        } catch (StopServiceException e) {

            fail("Unexpected exception stopping service: " + e.getMessage());

        }

    }

    private FeatureStore openTestFeatureStore() throws Exception {

        URL shapeURL =

            this.getClass().getResource("/cartography/hidro_andalucia.shp");

        if (shapeURL == null) {

            throw new IllegalStateException("Can't locate cartography resource");

        }

        File shapeFile = new File(shapeURL.getFile());

        if (!shapeFile.exists()) {

            throw new IllegalStateException("cartography resource not exists");

        }

        dataManager = DALLocator.getDataManager();

        DataStoreParameters params = dataManager.createStoreParameters("Shape");

        params.setDynValue("shpfile", shapeFile);

        params.setDynValue("CRS", CRSFactory.getCRS("EPSG:23030"));

        params.setDynValue("useNullGeometry", false);

        params.validate();

        return (FeatureStore) dataManager.openStore("Shape", params);

    }

    public void restartService() {

        try {

            service.stop();

            service.start();

        } catch (Exception e) {

            fail("Unexpected exception has been thrown restarting service. "

                + e.getMessage());

        }

    }

}