### Question:
CountryAxisGenerator implements AxisGenerator { @Override public Stream<AxisPointDTO> getPoints(long projectId) { return countryRepository.findAllByProjectIdOrderByCode(projectId).stream() .map(country -> new AxisPointDTO(country.getCode(), country.getCode().toUpperCase(), country.getName())); }  @Override String getCode(); @Override String getName(); @Override Stream<AxisPointDTO> getPoints(long projectId); @Override String[] getValuePoints(Functionality functionality);  }### Answer:
@Test public void testGetPoints() { when(countryRepository.findAllByProjectIdOrderByCode(A_PROJECT_ID)).thenReturn(Arrays.asList( new Country().withCode("be").withName("Belgium"), new Country().withCode("cn").withName("China"))); assertThat(cut.getPoints(A_PROJECT_ID)).containsExactly( new AxisPointDTO("be", "BE", "Belgium"), new AxisPointDTO("cn", "CN", "China")); }