diff --git a/src/main/java/de/tims/fleetstorm/ai/Logic.java b/src/main/java/de/tims/fleetstorm/ai/Logic.java index 8acdc8d..76e29f8 100644 --- a/src/main/java/de/tims/fleetstorm/ai/Logic.java +++ b/src/main/java/de/tims/fleetstorm/ai/Logic.java @@ -10,6 +10,7 @@ public class Logic { private Matchfield matchfield; private Coordinate lastShot; + public boolean foundShip = false; public Coordinate chooseField() { ArrayList possibleFields = new ArrayList(); @@ -57,4 +58,12 @@ public class Logic { return matchfield; } + public void findShip() { + if (lastShot.getState() == Coordinate.HIT) { + + foundShip = true; + } + System.out.println("Do this"); + } + } diff --git a/src/test/java/de/tims/fleetstorm/ai/LogicTest.java b/src/test/java/de/tims/fleetstorm/ai/LogicTest.java index 7257ccf..f266a76 100644 --- a/src/test/java/de/tims/fleetstorm/ai/LogicTest.java +++ b/src/test/java/de/tims/fleetstorm/ai/LogicTest.java @@ -102,4 +102,19 @@ class LogicTest { assertNotNull(result); } + @Test + void testFindShip() { + Logic logic = new Logic(); + Matchfield matchfield; + int size = 4; + matchfield = new Matchfield(size); + matchfield.createMatchfield(); + logic.setMatchfield(matchfield); + + logic.setLastShot(new Coordinate(2, 2)); + logic.getLastShot().setState(Coordinate.HIT); + logic.findShip(); + assertEquals(logic.foundShip, true); + } + }