Index: Ringtoets/GrassCoverErosionInwards/src/Ringtoets.GrassCoverErosionInwards.Plugin/GrassCoverErosionInwardsGuiPlugin.cs =================================================================== diff -u -r0209ecd38a9cb1e7224ceabe8b13e0697346abb9 -r0b85daf67145cfa567ef273790729e1e76a3467d --- Ringtoets/GrassCoverErosionInwards/src/Ringtoets.GrassCoverErosionInwards.Plugin/GrassCoverErosionInwardsGuiPlugin.cs (.../GrassCoverErosionInwardsGuiPlugin.cs) (revision 0209ecd38a9cb1e7224ceabe8b13e0697346abb9) +++ Ringtoets/GrassCoverErosionInwards/src/Ringtoets.GrassCoverErosionInwards.Plugin/GrassCoverErosionInwardsGuiPlugin.cs (.../GrassCoverErosionInwardsGuiPlugin.cs) (revision 0b85daf67145cfa567ef273790729e1e76a3467d) @@ -100,6 +100,7 @@ ForeColor = context => context.WrappedData.Any() ? Color.FromKnownColor(KnownColor.ControlText) : Color.FromKnownColor(KnownColor.GrayText), + ChildNodeObjects = context => context.WrappedData.Cast().ToArray(), ContextMenuStrip = (nodeData, parentData, treeViewControl) => Gui.Get(nodeData, treeViewControl) .AddImportItem() .AddSeparator() Index: Ringtoets/GrassCoverErosionInwards/test/Ringtoets.GrassCoverErosionInwards.Forms.Test/TreeNodeInfos/DikeProfilesContextTreeNodeInfoTest.cs =================================================================== diff -u -r5546343f650bb277b3c1937f1abf3ee0b7a7dc89 -r0b85daf67145cfa567ef273790729e1e76a3467d --- Ringtoets/GrassCoverErosionInwards/test/Ringtoets.GrassCoverErosionInwards.Forms.Test/TreeNodeInfos/DikeProfilesContextTreeNodeInfoTest.cs (.../DikeProfilesContextTreeNodeInfoTest.cs) (revision 5546343f650bb277b3c1937f1abf3ee0b7a7dc89) +++ Ringtoets/GrassCoverErosionInwards/test/Ringtoets.GrassCoverErosionInwards.Forms.Test/TreeNodeInfos/DikeProfilesContextTreeNodeInfoTest.cs (.../DikeProfilesContextTreeNodeInfoTest.cs) (revision 0b85daf67145cfa567ef273790729e1e76a3467d) @@ -21,18 +21,14 @@ using System.Drawing; using System.Linq; - using Core.Common.Base; using Core.Common.Base.Geometry; using Core.Common.Controls.TreeView; using Core.Common.Gui; using Core.Common.Gui.ContextMenu; using Core.Common.TestUtil; - using NUnit.Framework; - using Rhino.Mocks; - using Ringtoets.Common.Data.AssessmentSection; using Ringtoets.GrassCoverErosionInwards.Data; using Ringtoets.GrassCoverErosionInwards.Forms.PresentationObjects; @@ -169,6 +165,37 @@ } [Test] + public void ChildNodeObjects_Always_ReturnDikeProfiles() + { + // Setup + var mocks = new MockRepository(); + var assessmentSection = mocks.Stub(); + mocks.ReplayAll(); + + var dikeProfile1 = new DikeProfile(new Point2D(0, 0)); + var dikeProfile2 = new DikeProfile(new Point2D(0, 0)); + var dikeProfiles = new ObservableList + { + dikeProfile1, + dikeProfile2 + }; + + var dikeProfilesContext = new DikeProfilesContext(dikeProfiles, assessmentSection); + + // Call + var children = info.ChildNodeObjects(dikeProfilesContext); + + // Assert + var expectedChildren = new[] + { + dikeProfile1, + dikeProfile2 + }; + CollectionAssert.AreEqual(expectedChildren, children); + mocks.ReplayAll(); + } + + [Test] public void ContextMenuStrip_Always_CallsBuilder() { // Setup