Issue 3650: Fixed Restore Unit Action Test

MegaMek · Jan 8, 2025 · 39993a1 · 39993a1
1 parent 62b8d0e
commit 39993a1
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/MekHQ/unittests/mekhq/campaign/unit/actions/RestoreUnitActionTest.java b/MekHQ/unittests/mekhq/campaign/unit/actions/RestoreUnitActionTest.java
@@ -59,8 +59,6 @@ public void restoreUnitSwitchesOutEntityAndParts() {
         Unit unit = mock(Unit.class);
         when(unit.getId()).thenReturn(id);
         when(unit.getEntity()).thenReturn(mockEntity);
-        doNothing().when(unit).initializeShipTransportSpace();
-        doNothing().when(unit).initializeTacticalTransportSpace();
 
         Entity mockNewEntity = mock(Entity.class);
         doAnswer(inv -> {
@@ -86,6 +84,7 @@ public void restoreUnitSwitchesOutEntityAndParts() {
         verify(unit, times(1)).setEntity(eq(mockNewEntity));
         verify(unit, times(1)).removeParts();
         verify(unit, times(1)).initializeShipTransportSpace();
+        verify(unit, times(1)).initializeTacticalTransportSpace();
         verify(unit, times(1)).initializeParts(eq(true));
         verify(unit, times(1)).runDiagnostic(eq(false));
         verify(unit, times(1)).setSalvage(eq(false));