Index: Ringtoets/GrassCoverErosionOutwards/test/Ringtoets.GrassCoverErosionOutwards.Forms.Test/TreeNodeInfos/GrassCoverErosionOutwardsWaveConditionsCalculationGroupContextTreeNodeInfoTest.cs =================================================================== diff -u -r92572f54bf61141e88d3557b42f686c87b616d9e -ree395d64328db8f999b871e80491399a0a65e844 --- Ringtoets/GrassCoverErosionOutwards/test/Ringtoets.GrassCoverErosionOutwards.Forms.Test/TreeNodeInfos/GrassCoverErosionOutwardsWaveConditionsCalculationGroupContextTreeNodeInfoTest.cs (.../GrassCoverErosionOutwardsWaveConditionsCalculationGroupContextTreeNodeInfoTest.cs) (revision 92572f54bf61141e88d3557b42f686c87b616d9e) +++ Ringtoets/GrassCoverErosionOutwards/test/Ringtoets.GrassCoverErosionOutwards.Forms.Test/TreeNodeInfos/GrassCoverErosionOutwardsWaveConditionsCalculationGroupContextTreeNodeInfoTest.cs (.../GrassCoverErosionOutwardsWaveConditionsCalculationGroupContextTreeNodeInfoTest.cs) (revision ee395d64328db8f999b871e80491399a0a65e844) @@ -93,7 +93,7 @@ } [TearDown] - public void TearDown() + public override void TearDown() { plugin.Dispose(); mocks.VerifyAll(); @@ -153,6 +153,55 @@ } [Test] + public void ChildNodeObjects_EmptyGroup_ReturnEmpty() + { + // Setup + var assessmentSection = mocks.Stub(); + mocks.ReplayAll(); + + var failureMechanism = new GrassCoverErosionOutwardsFailureMechanism(); + + var groupContext = new GrassCoverErosionOutwardsWaveConditionsCalculationGroupContext(failureMechanism.WaveConditionsCalculationGroup, + failureMechanism, + assessmentSection); + + // Call + var children = info.ChildNodeObjects(groupContext); + + // Assert + CollectionAssert.IsEmpty(children); + } + + [Test] + public void ChildNodeObjects_GroupWithChildren_ReturnChildren() + { + // Setup + var assessmentSection = mocks.Stub(); + var calculationItem = mocks.StrictMock(); + mocks.ReplayAll(); + + var childGroup = new CalculationGroup(); + + var failureMechanism = new GrassCoverErosionOutwardsFailureMechanism(); + failureMechanism.WaveConditionsCalculationGroup.Children.Add(calculationItem); + failureMechanism.WaveConditionsCalculationGroup.Children.Add(childGroup); + + var nodeData = new GrassCoverErosionOutwardsWaveConditionsCalculationGroupContext(failureMechanism.WaveConditionsCalculationGroup, + failureMechanism, + assessmentSection); + + // Call + var children = info.ChildNodeObjects(nodeData).ToArray(); + + // Assert + Assert.AreEqual(failureMechanism.WaveConditionsCalculationGroup.Children.Count, children.Length); + Assert.AreSame(calculationItem, children[0]); + var returnedCalculationGroupContext = (GrassCoverErosionOutwardsWaveConditionsCalculationGroupContext)children[1]; + Assert.AreSame(childGroup, returnedCalculationGroupContext.WrappedData); + Assert.AreSame(failureMechanism, returnedCalculationGroupContext.FailureMechanism); + } + + [Test] public void ContextMenuStrip_NestedCalculationGroup_ReturnContextMenuWithItems() { // Setup