@Test public void testSetIsInFields_GetByShape() { OpenStreetMapSimpleImporter openStreetMapSimpleImporter = new OpenStreetMapSimpleImporter(); Point location = GeolocHelper.createPoint(2F, 3F); String countryCode = "FR"; ICityDao cityDao = EasyMock.createMock(ICityDao.class); City cityByShape = new City(); cityByShape.addZipCode(new ZipCode("zip")); cityByShape.setName("name"); cityByShape.setPopulation(1000000); Adm adm = new Adm(2); adm.setName("admName"); cityByShape.setAdm(adm); cityByShape.setFeatureId(1L); cityByShape.setId(123L); EasyMock.expect( cityDao.getByShape( EasyMock.anyObject(Point.class), EasyMock.anyObject(String.class), EasyMock.eq(true))) .andReturn(cityByShape); EasyMock.replay(cityDao); openStreetMapSimpleImporter.setCityDao(cityDao); OpenStreetMap street = new OpenStreetMap(); street.setCountryCode(countryCode); street.setLocation(location); openStreetMapSimpleImporter.setIsInFields(street); Set<String> expectedZip = new HashSet<String>(); expectedZip.add("ZIP"); Assert.assertEquals(expectedZip, street.getIsInZip()); Assert.assertEquals(true, street.isCityConfident()); Assert.assertEquals("admName", street.getIsInAdm()); Assert.assertEquals("name", street.getIsIn()); Assert.assertEquals(123L, street.getCityId()); Assert.assertEquals(null, street.getIsInPlace()); EasyMock.verify(cityDao); }