Index: Application/Ringtoets/test/Application.Ringtoets.Storage.Test/Create/Piping/PipingFailureMechanismCreateExtensionsTest.cs =================================================================== diff -u -r6b60edb4d7f04f83c74089d70a72c4f79ed413dd -re5a3455795622291e3347d1f3a7f900d5bf62da7 --- Application/Ringtoets/test/Application.Ringtoets.Storage.Test/Create/Piping/PipingFailureMechanismCreateExtensionsTest.cs (.../PipingFailureMechanismCreateExtensionsTest.cs) (revision 6b60edb4d7f04f83c74089d70a72c4f79ed413dd) +++ Application/Ringtoets/test/Application.Ringtoets.Storage.Test/Create/Piping/PipingFailureMechanismCreateExtensionsTest.cs (.../PipingFailureMechanismCreateExtensionsTest.cs) (revision e5a3455795622291e3347d1f3a7f900d5bf62da7) @@ -221,14 +221,15 @@ public void Create_WithCalculationGroup_ReturnFailureMechanismEntityWithCalculationGroupEntities() { // Setup + var calculationGroup = new CalculationGroup("A", true); + var calculation = new PipingCalculationScenario(new GeneralPipingInput()); + var failureMechanism = new PipingFailureMechanism(); - failureMechanism.CalculationsGroup.Children.Add(new CalculationGroup("A", true)); - failureMechanism.CalculationsGroup.Children.Add(new CalculationGroup("B", true)); + failureMechanism.CalculationsGroup.Children.Add(calculationGroup); + failureMechanism.CalculationsGroup.Children.Add(calculation); - var registry = new PersistenceRegistry(); - // Call - FailureMechanismEntity entity = failureMechanism.Create(registry); + FailureMechanismEntity entity = failureMechanism.Create(new PersistenceRegistry()); // Assert Assert.IsNotNull(entity); @@ -238,11 +239,18 @@ CalculationGroupEntity[] childGroupEntities = entity.CalculationGroupEntity.CalculationGroupEntity1 .OrderBy(cge => cge.Order) .ToArray(); - Assert.AreEqual(2, childGroupEntities.Length); - Assert.AreEqual("A", childGroupEntities[0].Name); - Assert.AreEqual(0, childGroupEntities[0].Order); - Assert.AreEqual("B", childGroupEntities[1].Name); - Assert.AreEqual(1, childGroupEntities[1].Order); + Assert.AreEqual(1, childGroupEntities.Length); + CalculationGroupEntity childGroupEntity = childGroupEntities[0]; + Assert.AreEqual(calculationGroup.Name, childGroupEntity.Name); + Assert.AreEqual(0, childGroupEntity.Order); + + PipingCalculationEntity[] calculationEntities = entity.CalculationGroupEntity.PipingCalculationEntities + .OrderBy(ce => ce.Order) + .ToArray(); + Assert.AreEqual(1, calculationEntities.Length); + PipingCalculationEntity calculationEntity = calculationEntities[0]; + Assert.AreEqual(calculation.Name, calculationEntity.Name); + Assert.AreEqual(1, calculationEntity.Order); } private static void AssertSurfaceLine(PipingSurfaceLine surfaceLine, SurfaceLineEntity entity)