diff --git a/src/main/java/de/tims/viergewinnt/ai/Logic.java b/src/main/java/de/tims/viergewinnt/ai/Logic.java index 123202a..922f32f 100644 --- a/src/main/java/de/tims/viergewinnt/ai/Logic.java +++ b/src/main/java/de/tims/viergewinnt/ai/Logic.java @@ -66,12 +66,18 @@ public class Logic { int size = getBoardSize() - 1; while(getField(size, column) != 0) { size--; + if(size < 0) { + return -1; + } } return size; } public int playChip(int column) { setField(getCurrentPlayer(), getFreePlace(column), column); + if(getFreePlace(column) == -1) { + buttons[column].setEnabled(false); + } updateGui(); if(testForWin()) { gameover(); @@ -238,5 +244,9 @@ public class Logic { setPlayer2Score(calcScore()); } clearField(); + for(int i = 0; i < getBoardSize(); i++) { + buttons[i].setEnabled(true); + } + updateGui(); } } diff --git a/src/test/java/de/tims/viergewinnt/ai/LogicTest.java b/src/test/java/de/tims/viergewinnt/ai/LogicTest.java index 76ff279..3e27f5c 100644 --- a/src/test/java/de/tims/viergewinnt/ai/LogicTest.java +++ b/src/test/java/de/tims/viergewinnt/ai/LogicTest.java @@ -181,6 +181,7 @@ class LogicTest { @Test void gameoverTest() { testObj.setField(1, 0, 0); + testObj.create4gewinntGui(); testObj.gameover();