diff --git a/src/test/java/logic/BoardTest.java b/src/test/java/logic/BoardTest.java index 5de92e1..e820c7c 100644 --- a/src/test/java/logic/BoardTest.java +++ b/src/test/java/logic/BoardTest.java @@ -46,6 +46,9 @@ public void equals() { board2.playStone(getCoords("M17"), StoneColour.WHITE); assertThat(board1, is(board2)); + + assertThat(board1, is(board1)); + assertThat(board1, is(not(getCoords("D4")))); } @Test diff --git a/src/test/java/logic/LocalGameHandlerTest.java b/src/test/java/logic/LocalGameHandlerTest.java index 79c3fe6..4561c02 100644 --- a/src/test/java/logic/LocalGameHandlerTest.java +++ b/src/test/java/logic/LocalGameHandlerTest.java @@ -20,6 +20,7 @@ import org.junit.Test; import util.StoneColour; +import static org.hamcrest.CoreMatchers.hasItems; import static org.hamcrest.CoreMatchers.is; import static org.junit.Assert.assertThat; import static util.Coords.getCoords; @@ -28,6 +29,17 @@ public class LocalGameHandlerTest { private static final String[] koBoard = {"C4", "D4", "D3", "E3", "D5", "E5", "K4", "F4", "E4"}; + @Test + public void gameTracking() { + LocalGameHandler handler = new LocalGameHandler(); + + handler.playStone(getCoords("D4"), StoneColour.BLACK); + handler.playStone(getCoords("E5"), StoneColour.WHITE); + + assertThat(handler.getStones(StoneColour.BLACK), hasItems(getCoords("D4"))); + assertThat(handler.getStones(StoneColour.WHITE), hasItems(getCoords("E5"))); + } + @Test public void koIsIllegal() { LocalGameHandler handler = buildTestHandler(koBoard);