1
0
Fork 0
mars-rover-kata/src/test/java/cat/hack3/codingtests/marsrover/MarsRoverTest.java

84 lines
2.7 KiB
Java

package cat.hack3.codingtests.marsrover;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;
import java.util.function.IntConsumer;
import java.util.stream.IntStream;
import static cat.hack3.codingtests.marsrover.Direction.*;
import static org.testng.Assert.assertEquals;
public class MarsRoverTest {
private MarsRover rover;
@BeforeMethod
public void setUp() {
int mapWidth = 10;
int mapHeight = 10;
int latitudeStartingPoint = 2;
int longitudeStartingPoint = 3;
var startingCoordinates = Coordinates.of(latitudeStartingPoint, longitudeStartingPoint);
var marsMap = new MarsMap(mapWidth, mapHeight, startingCoordinates);
rover = new MarsRover(marsMap, startingCoordinates, SOUTH);
}
@Test
public void stepForward() {
Direction originalDirection = rover.getCurrentDirection();
Coordinates oneStepForward = rover.moveForward();
assertEquals(oneStepForward, Coordinates.of(3, 3));
assertEquals(rover.getCurrentDirection(), originalDirection);
}
@Test
public void loopTheWorldInLatitudeMovingForward() {
Coordinates originalCoordinates = rover.getCurrentCoordinates();
Direction originalDirection = rover.getCurrentDirection();
repeatAction(10, i -> rover.moveForward());
assertEquals(rover.getCurrentCoordinates(), originalCoordinates);
assertEquals(rover.getCurrentDirection(), originalDirection);
}
@Test
public void stepBackwards() {
Direction originalDirection = rover.getCurrentDirection();
Coordinates currentPosition = rover.moveBackwards();
assertEquals(currentPosition, Coordinates.of(1, 3));
assertEquals(rover.getCurrentDirection(), originalDirection);
}
@Test
public void loopTheWorldInLatitudeMovingBackwards() {
Coordinates originalCoordinates = rover.getCurrentCoordinates();
Direction originalDirection = rover.getCurrentDirection();
repeatAction(10, i -> rover.moveBackwards());
assertEquals(rover.getCurrentCoordinates(), originalCoordinates);
assertEquals(rover.getCurrentDirection(), originalDirection);
}
@Test
public void turnLef() {
Direction currentDirection = rover.turnLeft();
assertEquals(currentDirection, EAST);
assertEquals(rover.turnLeft(), NORTH);
assertEquals(rover.turnLeft(), WEST);
repeatAction(2, i -> rover.turnLeft());
assertEquals(currentDirection, EAST);
}
private void repeatAction(int times, IntConsumer actionToRepeat) {
IntStream.rangeClosed(1, times)
.forEach(actionToRepeat);
}
}