@Test
public void toCountryDtoMapperTest() {
City city = CityTestBuilder.aCity().build();
ReflectionTestUtils.setField(city, "id", 2l);
Country country = CountryTestBuilder.aCountry().build();
ReflectionTestUtils.setField(country, "id", 1l);
country.addCity(city);
CountryDto countryDto = dtoMapper.toCountryDto(country);
assertThat(countryDto).isNotNull();
assertThat(countryDto.getName()).isEqualTo(country.getName());
assertThat(countryDto.getDescription()).isEqualTo(country.getDescription());
assertThat(countryDto.getCities().size()).isEqualTo(1);
CityDto mappedcity = countryDto.getCities().iterator().next();
assertThat(mappedcity.getName()).isEqualTo(city.getName());
}
