Added unit test

MegaMek · Nov 2, 2024 · 1257fee · 1257fee
1 parent 9475fc9
commit 1257fee
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/megamek/src/megamek/client/bot/princess/AeroPathUtil.java b/megamek/src/megamek/client/bot/princess/AeroPathUtil.java
@@ -118,9 +118,9 @@ public static boolean willStall(MovePath movePath) {
      */
     public static boolean willCrash(MovePath movePath) {
         return movePath.getEntity().isAero() &&
+                movePath.isOnAtmosphericGroundMap() &&
                 (movePath.getFinalAltitude() < 1) &&
                 !movePath.contains(MoveStepType.VLAND) &&
-                movePath.isOnAtmosphericGroundMap() &&
                 !movePath.contains(MoveStepType.LAND);
     }
 

diff --git a/megamek/unittests/megamek/client/bot/princess/AeroPathUtilTest.java b/megamek/unittests/megamek/client/bot/princess/AeroPathUtilTest.java
@@ -84,4 +84,18 @@ void testAssertWillCrashWithLandOrVland() {
 		boolean result = AeroPathUtil.willCrash(mockPath);
 		assertFalse(result);
 	}
+
+	@Test
+	void testAssertWillCrashNoAtmosphere() {
+		final Entity mockEntity = MockGenerators.generateMockAerospace(0, 0);
+
+		final MovePath mockPath = MockGenerators.generateMockPath(16, 16, mockEntity);
+		when(mockPath.getFinalVelocity()).thenReturn(1);
+		when(mockPath.isOnAtmosphericGroundMap()).thenReturn(false);
+
+		boolean result = AeroPathUtil.willCrash(mockPath);
+		assertFalse(result);
+	}
+
+
 }