@Test public void shouldGetFacilityByMrsFacilityId() { String facilityId = "0987654"; String name = "name"; String province = "province"; String district = "district"; String region = "region"; String country = "country"; final MRSFacility mrsFacility = new MRSFacility(facilityId, name, country, region, district, province); when(mockMrsFacilityAdaptor.saveFacility(mrsFacility)).thenReturn(mrsFacility); allFacilities.add(facility("123460987", null, null, null, mrsFacility).motechId(facilityId)); when(mockMrsFacilityAdaptor.getFacility(facilityId)).thenReturn(mrsFacility); Facility facility; assertFacility( allFacilities.getFacility(facilityId), facilityId, name, country, region, district, province); when(mockMrsFacilityAdaptor.getFacility(facilityId)).thenReturn(null); facility = allFacilities.getFacility(facilityId); assertThat(facility, is(equalTo(null))); }