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