@Test
void should_return_distance_when_entity_is_found() {

    DistanceKey distanceKey = new DistanceKey(from.id(), to.id());
    long distance = 10305;
    DistanceEntity distanceEntity = new DistanceEntity(distanceKey, distance);
    when(crudRepository.findById(distanceKey)).thenReturn(Optional.of(distanceEntity));
    assertThat(repository.getDistance(from, to)).isEqualTo(distance);
}