diff --git a/src/test/java/SnakeTest.java b/src/test/java/SnakeTest.java index 1658151..9242420 100644 --- a/src/test/java/SnakeTest.java +++ b/src/test/java/SnakeTest.java @@ -37,6 +37,12 @@ class SnakeTest { assertEquals(new Point(gridSize / 2, gridSize / 2 - 1), bodySegments.get(0)); assertEquals(new Point(gridSize / 2, gridSize / 2), bodySegments.get(1)); + snake.setDirection(Snake.SnakeDirection.RIGHT); + snake.update(); + bodySegments = snake.getBodySegments(); + assertEquals(new Point(gridSize / 2 + 1, gridSize / 2 - 1), bodySegments.get(0)); + assertEquals(new Point(gridSize / 2, gridSize / 2 - 1), bodySegments.get(1)); + }