Index: Riskeer/GrassCoverErosionInwards/src/Riskeer.GrassCoverErosionInwards.Plugin/GrassCoverErosionInwardsPlugin.cs =================================================================== diff -u -rddad7117cf82255493a1a9f80b2c6562dc88bfd2 -r9aedaf69371866ed6f94fc6a53de20a803a7b546 --- Riskeer/GrassCoverErosionInwards/src/Riskeer.GrassCoverErosionInwards.Plugin/GrassCoverErosionInwardsPlugin.cs (.../GrassCoverErosionInwardsPlugin.cs) (revision ddad7117cf82255493a1a9f80b2c6562dc88bfd2) +++ Riskeer/GrassCoverErosionInwards/src/Riskeer.GrassCoverErosionInwards.Plugin/GrassCoverErosionInwardsPlugin.cs (.../GrassCoverErosionInwardsPlugin.cs) (revision 9aedaf69371866ed6f94fc6a53de20a803a7b546) @@ -898,7 +898,8 @@ }; } - private ContextMenuStrip CalculationContextContextMenuStrip(GrassCoverErosionInwardsCalculationScenarioContext context, object parentData, TreeViewControl treeViewControl) + private ContextMenuStrip CalculationContextContextMenuStrip(GrassCoverErosionInwardsCalculationScenarioContext context, + object parentData, TreeViewControl treeViewControl) { GrassCoverErosionInwardsCalculationScenario calculation = context.WrappedData; @@ -932,7 +933,8 @@ }, inquiryHelper, viewCommands)) .AddClearIllustrationPointsOfCalculationItem( () => GrassCoverErosionInwardsIllustrationPointsHelper.HasIllustrationPoints(calculation), - new ClearIllustrationPointsOfGrassCoverErosionInwardsCalculationChangeHandler(calculation, inquiryHelper)) + new ClearIllustrationPointsOfGrassCoverErosionInwardsCalculationChangeHandler( + calculation, inquiryHelper, viewCommands)) .AddDeleteItem() .AddSeparator() .AddCollapseAllItem() Index: Riskeer/GrassCoverErosionInwards/test/Riskeer.GrassCoverErosionInwards.Plugin.Test/TreeNodeInfos/GrassCoverErosionInwardsCalculationScenarioContextTreeNodeInfoTest.cs =================================================================== diff -u -r4231613a2fe8a95de3ad758ba284a8f9a3d5f55f -r9aedaf69371866ed6f94fc6a53de20a803a7b546 --- Riskeer/GrassCoverErosionInwards/test/Riskeer.GrassCoverErosionInwards.Plugin.Test/TreeNodeInfos/GrassCoverErosionInwardsCalculationScenarioContextTreeNodeInfoTest.cs (.../GrassCoverErosionInwardsCalculationScenarioContextTreeNodeInfoTest.cs) (revision 4231613a2fe8a95de3ad758ba284a8f9a3d5f55f) +++ Riskeer/GrassCoverErosionInwards/test/Riskeer.GrassCoverErosionInwards.Plugin.Test/TreeNodeInfos/GrassCoverErosionInwardsCalculationScenarioContextTreeNodeInfoTest.cs (.../GrassCoverErosionInwardsCalculationScenarioContextTreeNodeInfoTest.cs) (revision 9aedaf69371866ed6f94fc6a53de20a803a7b546) @@ -1229,7 +1229,7 @@ var gui = mocks.Stub(); gui.Stub(g => g.Get(nodeData, treeViewControl)).Return(menuBuilder); gui.Stub(g => g.MainWindow).Return(mocks.Stub()); - gui.Stub(g => g.ViewCommands).Return(mocks.Stub()); + gui.Stub(g => g.ViewCommands).Return(mocks.StrictMock()); mocks.ReplayAll(); plugin.Gui = gui; @@ -1251,7 +1251,7 @@ } [Test] - public void GivenCalculationWithIllustrationPoints_WhenClearIllustrationPointsClickedAndContinued_ThenInquiryAndIllustrationPointsCleared() + public void GivenCalculationWithIllustrationPoints_WhenClearIllustrationPointsClickedAndContinued_ThenInquiryAndOutputViewsClosedAndIllustrationPointsCleared() { // Given IAssessmentSection assessmentSection = AssessmentSectionTestHelper.CreateAssessmentSectionStub(mocks); @@ -1280,10 +1280,15 @@ using (var treeViewControl = new TreeViewControl()) { + var viewCommands = mocks.StrictMock(); + viewCommands.Expect(vc => vc.RemoveAllViewsForItem(calculation.Output.OvertoppingOutput.GeneralResult)); + viewCommands.Expect(vc => vc.RemoveAllViewsForItem(calculation.Output.DikeHeightOutput.GeneralResult)); + viewCommands.Expect(vc => vc.RemoveAllViewsForItem(calculation.Output.OvertoppingRateOutput.GeneralResult)); + var gui = mocks.Stub(); gui.Stub(g => g.Get(nodeData, treeViewControl)).Return(menuBuilder); gui.Stub(g => g.MainWindow).Return(mocks.Stub()); - gui.Stub(g => g.ViewCommands).Return(mocks.Stub()); + gui.Stub(g => g.ViewCommands).Return(viewCommands); mocks.ReplayAll(); plugin.Gui = gui;