source: josm/trunk/test/unit/org/openstreetmap/josm/actions/SelectByInternalPointActionTest.java@ 10378

// License: GPL. For details, see LICENSE file.
package org.openstreetmap.josm.actions;

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

import org.junit.BeforeClass;
import org.junit.Test;
import org.openstreetmap.josm.JOSMFixture;
import org.openstreetmap.josm.Main;
import org.openstreetmap.josm.data.coor.EastNorth;
import org.openstreetmap.josm.data.osm.DataSet;
import org.openstreetmap.josm.data.osm.Node;
import org.openstreetmap.josm.data.osm.Relation;
import org.openstreetmap.josm.data.osm.RelationMember;
import org.openstreetmap.josm.data.osm.Way;
import org.openstreetmap.josm.gui.layer.Layer;
import org.openstreetmap.josm.gui.layer.OsmDataLayer;

/**
 * Unit tests for class {@link SelectByInternalPointAction}.
 */
public final class SelectByInternalPointActionTest {

    /**
     * Setup test.
     */
    @BeforeClass
    public static void setUp() {
        JOSMFixture.createUnitTestFixture().init(true);
    }

    /**
     * Unit test - no dataset.
     */
    @Test
    public void testNoDataSet() {
        while (Main.main.hasEditLayer()) {
            Main.main.removeLayer(Main.main.getEditLayer());
        }
        assertNull(JosmAction.getCurrentDataSet());
        assertEquals(0, SelectByInternalPointAction.getSurroundingObjects(null).size());
        assertNull(SelectByInternalPointAction.getSmallestSurroundingObject(null));
        SelectByInternalPointAction.performSelection(null, false, false);
    }

    private static Layer initDataSet() {
        DataSet ds = new DataSet();
        Node n1 = new Node(new EastNorth(1, 1));
        Node n2 = new Node(new EastNorth(1, 2));
        Node n3 = new Node(new EastNorth(2, 2));
        Node n4 = new Node(new EastNorth(2, 1));
        ds.addPrimitive(n1);
        ds.addPrimitive(n2);
        ds.addPrimitive(n3);
        ds.addPrimitive(n4);
        Way w = new Way();
        w.addNode(n1);
        w.addNode(n2);
        w.addNode(n3);
        w.addNode(n4);
        w.addNode(n1);
        assertTrue(w.isClosed());
        ds.addPrimitive(w);
        Relation r = new Relation();
        r.addMember(new RelationMember("outer", w));
        ds.addPrimitive(r);
        OsmDataLayer layer = new OsmDataLayer(ds, "", null);
        Main.main.addLayer(layer);
        return layer;
    }

    /**
     * Unit test of {@link SelectByInternalPointAction#getSurroundingObjects} method.
     */
    @Test
    public void testGetSurroundingObjects() {
        Layer layer = initDataSet();
        try {
            assertEquals(0, SelectByInternalPointAction.getSurroundingObjects(null).size());
            assertEquals(0, SelectByInternalPointAction.getSurroundingObjects(new EastNorth(0, 0)).size());
            assertEquals(1, SelectByInternalPointAction.getSurroundingObjects(new EastNorth(1.5, 1.5)).size());
            assertEquals(0, SelectByInternalPointAction.getSurroundingObjects(new EastNorth(3, 3)).size());
        } finally {
            // Ensure we clean the place before leaving, even if test fails.
            Main.main.removeLayer(layer);
        }
    }

    /**
     * Unit test of {@link SelectByInternalPointAction#getSmallestSurroundingObject} method.
     */
    @Test
    public void testGetSmallestSurroundingObject() {
        Layer layer = initDataSet();
        try {
            assertNull(SelectByInternalPointAction.getSmallestSurroundingObject(null));
            assertNotNull(SelectByInternalPointAction.getSmallestSurroundingObject(new EastNorth(1.5, 1.5)));
        } finally {
            // Ensure we clean the place before leaving, even if test fails.
            Main.main.removeLayer(layer);
        }
    }

    /**
     * Unit test of {@link SelectByInternalPointAction#performSelection} method.
     */
    @Test
    public void testPerformSelection() {
        Layer layer = initDataSet();
        try {
            DataSet ds = JosmAction.getCurrentDataSet();

            assertEquals(0, ds.getSelected().size());
            SelectByInternalPointAction.performSelection(null, false, false);
            assertEquals(0, ds.getSelected().size());
            SelectByInternalPointAction.performSelection(new EastNorth(0, 0), false, false);
            assertEquals(0, ds.getSelected().size());
            SelectByInternalPointAction.performSelection(new EastNorth(1.5, 1.5), false, false);
            assertEquals(1, ds.getSelected().size());
            ds.clearSelection();
            ds.addSelected(ds.getNodes());
            assertEquals(4, ds.getSelected().size());
            SelectByInternalPointAction.performSelection(new EastNorth(1.5, 1.5), true, false);
            assertEquals(5, ds.getSelected().size());
            SelectByInternalPointAction.performSelection(new EastNorth(1.5, 1.5), false, true);
            assertEquals(4, ds.getSelected().size());
        } finally {
            // Ensure we clean the place before leaving, even if test fails.
            Main.main.removeLayer(layer);
        }
    }
}