Set territoriesToCheck with attack options (#3271)

triplea-game · Mar 12, 2018 · 505fd66 · 505fd66
1 parent cd64c1a
commit 505fd66
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/game-core/src/main/java/games/strategy/triplea/ai/pro/ProCombatMoveAi.java b/game-core/src/main/java/games/strategy/triplea/ai/pro/ProCombatMoveAi.java
@@ -87,7 +87,7 @@ Map<Territory, ProTerritory> doCombatMove(final IMoveDelegate moveDel) {
     for (final ProTerritory patd : attackOptions) {
       clearedTerritories.add(patd.getTerritory());
     }
-    territoryManager.populateEnemyAttackOptions(clearedTerritories, new ArrayList<>());
+    territoryManager.populateEnemyAttackOptions(clearedTerritories, clearedTerritories);
     Set<Territory> territoriesToCheck = new HashSet<>(clearedTerritories);
     territoriesToCheck.addAll(ProData.myUnitTerritories);
     Map<Territory, Double> territoryValueMap =
@@ -109,6 +109,7 @@ Map<Territory, ProTerritory> doCombatMove(final IMoveDelegate moveDel) {
             .addAll(data.getMap().getNeighbors(patd.getTerritory(), Matches.territoryIsWater()));
       }
     }
+    possibleTransportTerritories.addAll(clearedTerritories);
     territoryManager.populateEnemyAttackOptions(clearedTerritories, new ArrayList<>(possibleTransportTerritories));
     territoriesToCheck = new HashSet<>(clearedTerritories);
     territoriesToCheck.addAll(ProData.myUnitTerritories);