public void FacilitiesCanOnlyHaveOneSiteOfTreatment() { EntityHelper.SetEntityIds(facility1, facility2); facilityCollection.SetFacilityAsSiteOfTreatment(facility1.Id); var siteOfTreatmentCount = facilityCollection.Facilities.Count(p => p.IsActualSiteOfTreatment); Assert.Equal(1, siteOfTreatmentCount); }
public async Task CanRemoveFacilityOtherThanActualSiteOfTreatment() { var notification = NotificationApplicationFactory.Create(Guid.NewGuid(), NotificationType.Recovery, UKCompetentAuthority.England, 0); var address = ObjectFactory.CreateDefaultAddress(); var business = ObjectFactory.CreateEmptyBusiness(); var contact = ObjectFactory.CreateEmptyContact(); context.NotificationApplications.Add(notification); await context.SaveChangesAsync(); var facilityCollection = new FacilityCollection(notification.Id); context.Facilities.Add(facilityCollection); await context.SaveChangesAsync(); var facility = facilityCollection.AddFacility(business, address, contact); var anotherFacility = facilityCollection.AddFacility(business, address, contact); await context.SaveChangesAsync(); Assert.True(facilityCollection.Facilities.Count() == 2); facilityCollection.SetFacilityAsSiteOfTreatment(facility.Id); await context.SaveChangesAsync(); facilityCollection.RemoveFacility(anotherFacility.Id); await context.SaveChangesAsync(); Assert.True(facilityCollection.Facilities.Count() == 1); }
public async Task CanRemoveFacilityOtherThanActualSiteOfTreatment() { var notification = NotificationApplicationFactory.Create(Guid.NewGuid(), NotificationType.Recovery, UKCompetentAuthority.England, 0); var address = ObjectFactory.CreateDefaultAddress(); var business = ObjectFactory.CreateEmptyBusiness(); var contact = ObjectFactory.CreateEmptyContact(); context.NotificationApplications.Add(notification); await context.SaveChangesAsync(); var facilityCollection = new FacilityCollection(notification.Id); context.Facilities.Add(facilityCollection); await context.SaveChangesAsync(); var facility = facilityCollection.AddFacility(business, address, contact); var anotherFacility = facilityCollection.AddFacility(business, address, contact); await context.SaveChangesAsync(); Assert.True(facilityCollection.Facilities.Count() == 2); facilityCollection.SetFacilityAsSiteOfTreatment(facility.Id); await context.SaveChangesAsync(); facilityCollection.RemoveFacility(anotherFacility.Id); await context.SaveChangesAsync(); Assert.True(facilityCollection.Facilities.Count() == 1); context.DeleteOnCommit(facilityCollection); context.DeleteOnCommit(notification); await context.SaveChangesAsync(); }