Index: Ringtoets/HeightStructures/test/Ringtoets.HeightStructures.Service.Test/HeightStructuresDataSynchronizationServiceTest.cs =================================================================== diff -u -r1051f838aebbd5977351542027ce6330583b0d43 -rd9e5e7ea1e028347a738fc8e3d9c158a9b53ce15 --- Ringtoets/HeightStructures/test/Ringtoets.HeightStructures.Service.Test/HeightStructuresDataSynchronizationServiceTest.cs (.../HeightStructuresDataSynchronizationServiceTest.cs) (revision 1051f838aebbd5977351542027ce6330583b0d43) +++ Ringtoets/HeightStructures/test/Ringtoets.HeightStructures.Service.Test/HeightStructuresDataSynchronizationServiceTest.cs (.../HeightStructuresDataSynchronizationServiceTest.cs) (revision d9e5e7ea1e028347a738fc8e3d9c158a9b53ce15) @@ -20,6 +20,7 @@ // All rights reserved. using System; +using System.Collections.Generic; using System.Linq; using NUnit.Framework; using Ringtoets.Common.Data.Probability; @@ -46,17 +47,21 @@ Output = new ProbabilityAssessmentOutput(0, 0, 0, 0, 0) }; + HeightStructuresCalculation calculation3 = new HeightStructuresCalculation(); + failureMechanism.CalculationsGroup.Children.Add(calculation1); failureMechanism.CalculationsGroup.Children.Add(calculation2); + failureMechanism.CalculationsGroup.Children.Add(calculation3); // Call - HeightStructuresDataSynchronizationService.ClearAllCalculationOutput(failureMechanism); + IEnumerable affectedItems = HeightStructuresDataSynchronizationService.ClearAllCalculationOutput(failureMechanism); // Assert foreach (HeightStructuresCalculation calculation in failureMechanism.CalculationsGroup.Children.Cast()) { Assert.IsNull(calculation.Output); } + CollectionAssert.AreEqual(new[] { calculation1, calculation2 }, affectedItems); } [Test] @@ -87,7 +92,7 @@ } [Test] - public void ClearHydraulicBoundaryLocations_WithHydraulicBoundaryLocation_ClearsHydraulicBoundaryLocation() + public void ClearHydraulicBoundaryLocations_WithHydraulicBoundaryLocation_ClearsHydraulicBoundaryLocationAndReturnsAffectedCalculations() { // Setup HeightStructuresFailureMechanism failureMechanism = new HeightStructuresFailureMechanism(); @@ -109,17 +114,25 @@ } }; + HeightStructuresCalculation calculation3 = new HeightStructuresCalculation(); + failureMechanism.CalculationsGroup.Children.Add(calculation1); failureMechanism.CalculationsGroup.Children.Add(calculation2); + failureMechanism.CalculationsGroup.Children.Add(calculation3); // Call - HeightStructuresDataSynchronizationService.ClearHydraulicBoundaryLocations(failureMechanism); + IEnumerable affectedItems = HeightStructuresDataSynchronizationService.ClearHydraulicBoundaryLocations(failureMechanism); // Assert foreach (HeightStructuresCalculation calculation in failureMechanism.CalculationsGroup.Children.Cast()) { Assert.IsNull(calculation.InputParameters.HydraulicBoundaryLocation); } + CollectionAssert.AreEqual(new[] + { + calculation1, + calculation2 + }, affectedItems); } } } \ No newline at end of file