Index: Ringtoets/Piping/test/Ringtoets.Piping.Forms.Test/PresentationObjects/PipingCalculationsTreeFolderTest.cs =================================================================== diff -u -rfb32e3411873a7b4c567792b9f335eba462e2702 -red0b4ea830beb801bcc0e3eb5b28993001ed2981 --- Ringtoets/Piping/test/Ringtoets.Piping.Forms.Test/PresentationObjects/PipingCalculationsTreeFolderTest.cs (.../PipingCalculationsTreeFolderTest.cs) (revision fb32e3411873a7b4c567792b9f335eba462e2702) +++ Ringtoets/Piping/test/Ringtoets.Piping.Forms.Test/PresentationObjects/PipingCalculationsTreeFolderTest.cs (.../PipingCalculationsTreeFolderTest.cs) (revision ed0b4ea830beb801bcc0e3eb5b28993001ed2981) @@ -14,7 +14,7 @@ public class PipingCalculationsTreeFolderTest { [Test] - public void ParameteredConstructor_ExpectedValues() + public void ParameteredConstructor_FailureMechanismWithOneCalculation_ExpectedValues() { // Setup const string folderName = "Berekeningen"; @@ -29,9 +29,11 @@ Assert.IsInstanceOf(calculationsFolder); Assert.AreEqual(folderName, calculationsFolder.Name); Assert.AreSame(failureMechanism, calculationsFolder.ParentFailureMechanism); - var calculationPresentationObjects = calculationsFolder.Contents - .OfType() - .ToArray(); + + object[] contentsAsArray = calculationsFolder.Contents.OfType().ToArray(); + Assert.AreEqual(1, contentsAsArray.Length); + var calculationPresentationObjects = contentsAsArray.Cast() + .ToArray(); foreach (var pipingCalculationContext in calculationPresentationObjects) { CollectionAssert.AreEqual(failureMechanism.SurfaceLines, pipingCalculationContext.AvailablePipingSurfaceLines); @@ -43,6 +45,31 @@ Assert.AreEqual(TreeFolderCategory.General, calculationsFolder.Category); } + [Test] + public void ParameteredConstructor_FailureMechanismWithOneEmptyCalculationGroup_ExpectedValues() + { + // Setup + const string folderName = "Berekeningen"; + + var group = new PipingCalculationGroup(); + + var failureMechanism = new PipingFailureMechanism(); + failureMechanism.Calculations.Clear(); + failureMechanism.Calculations.Add(group); + + // Call + var calculationsFolder = new PipingCalculationsTreeFolder(folderName, failureMechanism); + + // Assert + Assert.IsInstanceOf(calculationsFolder); + Assert.AreEqual(folderName, calculationsFolder.Name); + Assert.AreSame(failureMechanism, calculationsFolder.ParentFailureMechanism); + + object[] contentsAsArray = calculationsFolder.Contents.OfType().ToArray(); + Assert.AreEqual(1, contentsAsArray.Length); + CollectionAssert.AreEqual(new[]{group}, contentsAsArray); + } + private void AddTestSurfaceLines(PipingFailureMechanism failureMechanism) { var collection = (ICollection)failureMechanism.SurfaceLines;