Add explicit test for the status line object feedback

Signed-off-by: Taylor Smock <taylor.smock@kaart.com>

src/main/java/org/openstreetmap/josm/plugins/mapwithai/backend/MapWithAIObject.java

@@ -74,6 +74,7 @@ public class MapWithAIObject implements CommandQueueListener, Destroyable {
     @Override
     public void destroy() {
         statusLines.forEach(mapStatus -> mapStatus.remove(mapWithAIObjects));
+        statusLines.clear();
         UndoRedoHandler.getInstance().removeCommandQueueListener(this);
     }
 }

test/unit/org/openstreetmap/josm/plugins/mapwithai/backend/MapWithAIObjectTest.java

@@ -0,0 +1,116 @@
+// License: GPL. For details, see LICENSE file.
+package org.openstreetmap.josm.plugins.mapwithai.backend;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertNotNull;
+import static org.junit.jupiter.api.Assertions.assertTrue;
+
+import java.awt.Component;
+import java.util.Arrays;
+import java.util.Collections;
+import java.util.List;
+import java.util.stream.Stream;
+
+import javax.swing.text.JTextComponent;
+
+import org.junit.Before;
+import org.junit.Rule;
+import org.junit.Test;
+import org.openstreetmap.josm.JOSMFixture;
+import org.openstreetmap.josm.data.UndoRedoHandler;
+import org.openstreetmap.josm.data.coor.LatLon;
+import org.openstreetmap.josm.data.osm.DataSet;
+import org.openstreetmap.josm.data.osm.Node;
+import org.openstreetmap.josm.gui.MainApplication;
+import org.openstreetmap.josm.gui.MapStatus;
+import org.openstreetmap.josm.gui.layer.OsmDataLayer;
+import org.openstreetmap.josm.plugins.mapwithai.commands.MapWithAIAddCommand;
+import org.openstreetmap.josm.testutils.JOSMTestRules;
+
+/**
+ * @author Taylor Smock
+ *
+ */
+public class MapWithAIObjectTest {
+    @Rule
+    public JOSMTestRules rule = new JOSMTestRules().main().projection();
+
+    private MapWithAIObject mapWithAIObject;
+    private MapStatus statusLine;
+    private OsmDataLayer osmData;
+
+    @Before
+    public void setUp() {
+        JOSMFixture.initToolbar();
+        mapWithAIObject = new MapWithAIObject();
+        osmData = new OsmDataLayer(new DataSet(), "", null);
+        // Required to have a non-null MainApplication.getMap()
+        MainApplication.getLayerManager().addLayer(osmData);
+        statusLine = MainApplication.getMap().statusLine;
+    }
+
+    /**
+     * Test method for
+     * {@link org.openstreetmap.josm.plugins.mapwithai.backend.MapWithAIObject#addMapStatus(org.openstreetmap.josm.gui.MapStatus)}.
+     */
+    @Test
+    public void testAddMapStatus() {
+        int initialComponents = statusLine.getComponentCount();
+        for (int i = 0; i < 10; i++) {
+            mapWithAIObject.addMapStatus(statusLine);
+            assertEquals(initialComponents + 1, statusLine.getComponentCount());
+        }
+    }
+
+    /**
+     * Test method for
+     * {@link org.openstreetmap.josm.plugins.mapwithai.backend.MapWithAIObject#removeMapStatus(org.openstreetmap.josm.gui.MapStatus)}.
+     */
+    @Test
+    public void testRemoveMapStatus() {
+        testAddMapStatus();
+        int addedComponents = statusLine.getComponentCount();
+        for (int i = 0; i < 10; i++) {
+            mapWithAIObject.removeMapStatus(statusLine);
+            assertEquals(addedComponents - 1, statusLine.getComponentCount());
+        }
+    }
+
+    /**
+     * Test method for
+     * {@link org.openstreetmap.josm.plugins.mapwithai.backend.MapWithAIObject#commandChanged(int, int)}.
+     */
+    @Test
+    public void testCommandChanged() {
+        List<Component> initialComponents = Arrays.asList(statusLine.getComponents());
+        testAddMapStatus();
+        JTextComponent text = Stream.of(statusLine.getComponents()).filter(JTextComponent.class::isInstance)
+                .map(JTextComponent.class::cast).filter(i -> !initialComponents.contains(i)).findFirst().orElse(null);
+        assertNotNull(text);
+        assertTrue(text.getText().contains("0"));
+        UndoRedoHandler.getInstance().addCommandQueueListener(mapWithAIObject);
+        UndoRedoHandler.getInstance().add(
+                new MapWithAIAddCommand(MapWithAIDataUtils.getLayer(true), osmData, Collections.emptySet()), false);
+        assertTrue(text.getText().contains("0"));
+        Node node1 = new Node(LatLon.ZERO);
+        MapWithAIDataUtils.getLayer(true).getDataSet().addPrimitive(node1);
+        UndoRedoHandler.getInstance().add(
+                new MapWithAIAddCommand(MapWithAIDataUtils.getLayer(false), osmData, Collections.singleton(node1)),
+                false);
+        assertTrue(text.getText().contains("1"));
+        UndoRedoHandler.getInstance().removeCommandQueueListener(mapWithAIObject);
+    }
+
+    /**
+     * Test method for
+     * {@link org.openstreetmap.josm.plugins.mapwithai.backend.MapWithAIObject#destroy()}.
+     */
+    @Test
+    public void testDestroy() {
+        testAddMapStatus();
+        int addedComponents = statusLine.getComponentCount();
+        mapWithAIObject.destroy();
+        assertEquals(addedComponents - 1, statusLine.getComponentCount());
+    }
+
+}