Index: Application/Ringtoets/src/Application.Ringtoets.Storage/Create/GrassCoverErosionInwardsFailureMechanismCreateExtensions.cs =================================================================== diff -u -rbb92ba4d7297c7bf0d1c81ff3d9d6bcab640d55f -r838250fef0c200dbbffa9a562ca5935fdc500d19 --- Application/Ringtoets/src/Application.Ringtoets.Storage/Create/GrassCoverErosionInwardsFailureMechanismCreateExtensions.cs (.../GrassCoverErosionInwardsFailureMechanismCreateExtensions.cs) (revision bb92ba4d7297c7bf0d1c81ff3d9d6bcab640d55f) +++ Application/Ringtoets/src/Application.Ringtoets.Storage/Create/GrassCoverErosionInwardsFailureMechanismCreateExtensions.cs (.../GrassCoverErosionInwardsFailureMechanismCreateExtensions.cs) (revision 838250fef0c200dbbffa9a562ca5935fdc500d19) @@ -20,6 +20,7 @@ // All rights reserved. using System; +using System.Collections.Generic; using Application.Ringtoets.Storage.DbContext; using Ringtoets.GrassCoverErosionInwards.Data; @@ -39,28 +40,18 @@ /// Thrown when is null. internal static FailureMechanismEntity Create(this GrassCoverErosionInwardsFailureMechanism mechanism, PersistenceRegistry registry) { - if (registry == null) - { - throw new ArgumentNullException("registry"); - } + var entity = mechanism.Create(FailureMechanismType.GrassRevetmentTopErosionAndInwards, registry); + AddEntitiesForSectionResults(mechanism.SectionResults, registry); - var entity = new FailureMechanismEntity - { - FailureMechanismType = (short) FailureMechanismType.GrassRevetmentTopErosionAndInwards, - IsRelevant = Convert.ToByte(mechanism.IsRelevant), - Comments = mechanism.Comments - }; - - mechanism.AddEntitiesForFailureMechanismSections(registry, entity); - AddEntitiesForSectionResults(mechanism, registry); - registry.Register(entity, mechanism); return entity; } - private static void AddEntitiesForSectionResults(GrassCoverErosionInwardsFailureMechanism mechanism, PersistenceRegistry registry) + private static void AddEntitiesForSectionResults( + IEnumerable sectionResults, + PersistenceRegistry registry) { - foreach (var failureMechanismSectionResult in mechanism.SectionResults) + foreach (var failureMechanismSectionResult in sectionResults) { var sectionResultEntity = failureMechanismSectionResult.Create(registry); var section = registry.Get(failureMechanismSectionResult.Section);