MapWithAI/src/test/unit/org/openstreetmap/josm/plugins/mapwithai/backend/MapWithAIActionTest.java

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

import static org.awaitility.Awaitility.await;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertFalse;
import static org.junit.jupiter.api.Assertions.assertNotNull;
import static org.junit.jupiter.api.Assertions.assertNotSame;
import static org.junit.jupiter.api.Assertions.assertNull;
import static org.junit.jupiter.api.Assertions.assertSame;
import static org.junit.jupiter.api.Assertions.assertTrue;

import org.awaitility.Durations;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.Timeout;
import org.openstreetmap.josm.data.Bounds;
import org.openstreetmap.josm.data.DataSource;
import org.openstreetmap.josm.data.coor.LatLon;
import org.openstreetmap.josm.data.osm.DataSet;
import org.openstreetmap.josm.gui.MainApplication;
import org.openstreetmap.josm.gui.Notification;
import org.openstreetmap.josm.gui.layer.Layer;
import org.openstreetmap.josm.gui.layer.OsmDataLayer;
import org.openstreetmap.josm.plugins.mapwithai.testutils.annotations.MapWithAISources;
import org.openstreetmap.josm.plugins.mapwithai.testutils.annotations.NoExceptions;
import org.openstreetmap.josm.testutils.annotations.BasicPreferences;
import org.openstreetmap.josm.testutils.annotations.Main;
import org.openstreetmap.josm.testutils.annotations.Projection;
import org.openstreetmap.josm.tools.Territories;

/**
 * Test class for {@link MapWithAIAction}
 *
 * @author Taylor Smock
 */
@BasicPreferences
@Main
@MapWithAISources
@NoExceptions
@Projection
@org.openstreetmap.josm.testutils.annotations.Territories
@Timeout(10)
class MapWithAIActionTest {
    private MapWithAIAction action;

    @BeforeEach
    void setUp() {
        action = new MapWithAIAction();
        LatLon temp = new LatLon(40, -100);
        await().atMost(Durations.TEN_SECONDS).until(() -> Territories.isIso3166Code("US", temp));
    }

    @Test
    void testEnabled() {
        assertFalse(action.isEnabled());
        MainApplication.getLayerManager().addLayer(new OsmDataLayer(new DataSet(), "temporary", null));
        assertTrue(action.isEnabled());
    }

    @Test
    void testDownload() {
        assertTrue(MainApplication.getLayerManager().getLayers().isEmpty());
        action.actionPerformed(null);
        assertTrue(MainApplication.getLayerManager().getLayers().isEmpty());

        MainApplication.getLayerManager().addLayer(new OsmDataLayer(new DataSet(), "temporary", null));
        action.actionPerformed(null);
        await().atMost(Durations.TEN_SECONDS)
                .until(() -> 1 == MainApplication.getLayerManager().getLayersOfType(MapWithAILayer.class).size());
        assertEquals(1, MainApplication.getLayerManager().getLayersOfType(MapWithAILayer.class).size());

        assertSame(MapWithAIDataUtils.getLayer(false), MainApplication.getLayerManager().getActiveLayer());
        action.actionPerformed(null);
        assertNotSame(MapWithAIDataUtils.getLayer(false), MainApplication.getLayerManager().getActiveLayer());
        MainApplication.getLayerManager().removeLayer(MainApplication.getLayerManager().getActiveLayer());
        action.actionPerformed(null);
        assertSame(MapWithAIDataUtils.getLayer(false), MainApplication.getLayerManager().getActiveLayer());
    }

    @Test
    void testToggleLayer() {
        MapWithAIAction.toggleLayer(null);
        Layer layer = new OsmDataLayer(new DataSet(), "Test layer", null);
        assertNull(MainApplication.getLayerManager().getActiveLayer());
        MainApplication.getLayerManager().addLayer(layer);
        assertSame(layer, MainApplication.getLayerManager().getActiveLayer());
        MapWithAIAction.toggleLayer(layer);
        assertSame(layer, MainApplication.getLayerManager().getActiveLayer());
        MapWithAIDataUtils.getLayer(true);
        await().atMost(Durations.ONE_SECOND).until(() -> MapWithAIDataUtils.getLayer(false) != null);
        // Adding the MapWithAI layer switches to it
        assertSame(MapWithAIDataUtils.getLayer(false), MainApplication.getLayerManager().getActiveLayer());
        MapWithAIAction.toggleLayer(layer);
        assertSame(layer, MainApplication.getLayerManager().getActiveLayer());
        MapWithAIAction.toggleLayer(layer);
        assertSame(MapWithAIDataUtils.getLayer(false), MainApplication.getLayerManager().getActiveLayer());
        MapWithAIAction.toggleLayer(null);
        assertSame(MapWithAIDataUtils.getLayer(false), MainApplication.getLayerManager().getActiveLayer());
    }

    @Test
    void testCreateNotification() {
        Notification notification = MapWithAIAction.createMessageDialog();
        assertNull(notification);

        MapWithAILayer layer = MapWithAIDataUtils.getLayer(true);
        await().atMost(Durations.ONE_SECOND).until(() -> MapWithAIDataUtils.getLayer(false) != null);
        notification = MapWithAIAction.createMessageDialog();
        assertNotNull(notification);
        DataSource source = new DataSource(new Bounds(38.8876078, -77.012173, 38.892087, -77.0059234), "test area");
        layer.getDataSet().addDataSource(source);
        notification = MapWithAIAction.createMessageDialog();
        assertNotNull(notification);
        assertEquals(Notification.TIME_DEFAULT, notification.getDuration());
    }
}