Index: Ringtoets/ClosingStructures/test/Ringtoets.ClosingStructures.Forms.Test/TreeNodeInfos/ClosingStructureFailureMechanismContextTreeNodeInfoTest.cs =================================================================== diff -u -rbc9af05d3eb0d40e5620b96f537ed92f3ae52e94 -r7a165d807e1cce8ca51f7707ce6addb0951aa8b4 --- Ringtoets/ClosingStructures/test/Ringtoets.ClosingStructures.Forms.Test/TreeNodeInfos/ClosingStructureFailureMechanismContextTreeNodeInfoTest.cs (.../ClosingStructureFailureMechanismContextTreeNodeInfoTest.cs) (revision bc9af05d3eb0d40e5620b96f537ed92f3ae52e94) +++ Ringtoets/ClosingStructures/test/Ringtoets.ClosingStructures.Forms.Test/TreeNodeInfos/ClosingStructureFailureMechanismContextTreeNodeInfoTest.cs (.../ClosingStructureFailureMechanismContextTreeNodeInfoTest.cs) (revision 7a165d807e1cce8ca51f7707ce6addb0951aa8b4) @@ -146,8 +146,8 @@ Assert.AreSame(failureMechanism, closingStructuresContext.FailureMechanism); Assert.AreSame(assessmentSectionMock, closingStructuresContext.AssessmentSection); - var commentContext = (CommentContext) inputsFolder.Contents[3]; - Assert.AreSame(failureMechanism, commentContext.WrappedData); + var inputCommentContext = (CommentContext) inputsFolder.Contents[3]; + Assert.AreSame(failureMechanism.InputComments, inputCommentContext.WrappedData); var calculationsFolder = (ClosingStructuresCalculationGroupContext) children[1]; Assert.AreEqual("Berekeningen", calculationsFolder.WrappedData.Name); @@ -157,19 +157,22 @@ var outputsFolder = (CategoryTreeFolder) children[2]; Assert.AreEqual("Oordeel", outputsFolder.Name); Assert.AreEqual(TreeFolderCategory.Output, outputsFolder.Category); - Assert.AreEqual(2, outputsFolder.Contents.Count); + Assert.AreEqual(3, outputsFolder.Contents.Count); var scenariosContext = (ClosingStructuresScenariosContext) outputsFolder.Contents[0]; Assert.AreSame(failureMechanism.CalculationsGroup, scenariosContext.WrappedData); Assert.AreSame(failureMechanism, scenariosContext.ParentFailureMechanism); var failureMechanismResultsContext = (FailureMechanismSectionResultContext) outputsFolder.Contents[1]; Assert.AreSame(failureMechanism, failureMechanismResultsContext.FailureMechanism); Assert.AreSame(failureMechanism.SectionResults, failureMechanismResultsContext.WrappedData); + + var outputCommentContext = (CommentContext) outputsFolder.Contents[2]; + Assert.AreSame(failureMechanism.OutputComments, outputCommentContext.WrappedData); } [Test] - public void ChildNodeObjects_FailureMechanismIsNotRelevant_ReturnOnlyFailureMechanismComments() + public void ChildNodeObjects_FailureMechanismIsNotRelevant_ReturnOnlyFailureMechanismNotRelevantComments() { // Setup var assessmentSectionMock = mocksRepository.Stub(); @@ -187,7 +190,7 @@ // Assert Assert.AreEqual(1, children.Length); var commentContext = (CommentContext) children[0]; - Assert.AreSame(failureMechanism, commentContext.WrappedData); + Assert.AreSame(failureMechanism.NotRelevantComments, commentContext.WrappedData); } [Test] @@ -354,8 +357,8 @@ { // Setup var failureMechanism = new ClosingStructuresFailureMechanism(); - var assessmentSectionMock = mocksRepository.Stub(); - var failureMechanismContext = new ClosingStructuresFailureMechanismContext(failureMechanism, assessmentSectionMock); + var assessmentSectionStub = mocksRepository.Stub(); + var failureMechanismContext = new ClosingStructuresFailureMechanismContext(failureMechanism, assessmentSectionStub); var viewCommandsMock = mocksRepository.StrictMock(); var menuBuilder = new CustomItemsOnlyContextMenuBuilder(); var guiMock = mocksRepository.StrictMock(); @@ -382,18 +385,22 @@ } [Test] - public void ContextMenuStrip_FailureMechanismIsNotRelevantAndClickOnIsRelevantItem_MakeFailureMechanismRelevant() + public void ContextMenuStrip_FailureMechanismIsNotRelevantAndClickOnIsRelevantItem_OnChangeActionRemovesAllViewsForItem() { // Setup var failureMechanism = new ClosingStructuresFailureMechanism { IsRelevant = false }; - var assessmentSectionMock = mocksRepository.Stub(); - var failureMechanismContext = new ClosingStructuresFailureMechanismContext(failureMechanism, assessmentSectionMock); + var assessmentSectionStub = mocksRepository.Stub(); + var failureMechanismContext = new ClosingStructuresFailureMechanismContext(failureMechanism, assessmentSectionStub); + var viewCommandsMock = mocksRepository.StrictMock(); var menuBuilder = new CustomItemsOnlyContextMenuBuilder(); var guiMock = mocksRepository.StrictMock(); + viewCommandsMock.Expect(vs => vs.RemoveAllViewsForItem(failureMechanismContext)); + guiMock.Stub(g => g.ViewCommands).Return(viewCommandsMock); + using (var treeViewControl = new TreeViewControl()) { guiMock.Expect(g => g.Get(failureMechanismContext, treeViewControl)).Return(menuBuilder);