Index: Riskeer/GrassCoverErosionInwards/src/Riskeer.GrassCoverErosionInwards.Plugin/GrassCoverErosionInwardsPlugin.cs =================================================================== diff -u -rc1242c336a97112343b2e37c6dfc644cedddfd69 -rfc81f53fa4eea5508bc1b59cb8606c0583783b72 --- Riskeer/GrassCoverErosionInwards/src/Riskeer.GrassCoverErosionInwards.Plugin/GrassCoverErosionInwardsPlugin.cs (.../GrassCoverErosionInwardsPlugin.cs) (revision c1242c336a97112343b2e37c6dfc644cedddfd69) +++ Riskeer/GrassCoverErosionInwards/src/Riskeer.GrassCoverErosionInwards.Plugin/GrassCoverErosionInwardsPlugin.cs (.../GrassCoverErosionInwardsPlugin.cs) (revision fc81f53fa4eea5508bc1b59cb8606c0583783b72) @@ -605,7 +605,7 @@ CreateClearCalculationOutputChangeHandler(calculations, inquiryHelper, viewCommands)) .AddClearIllustrationPointsOfCalculationsInFailureMechanismItem( () => GrassCoverErosionInwardsIllustrationPointsHelper.HasIllustrationPoints(calculations), - CreateIllustrationPointsChangeHandler(inquiryHelper, calculations)) + CreateIllustrationPointsChangeHandler(calculations, inquiryHelper, viewCommands)) .AddSeparator() .AddCollapseAllItem() .AddExpandAllItem() @@ -749,7 +749,7 @@ CreateClearCalculationOutputChangeHandler(calculations, inquiryHelper, viewCommands)) .AddClearIllustrationPointsOfCalculationsInGroupItem( () => GrassCoverErosionInwardsIllustrationPointsHelper.HasIllustrationPoints(calculations), - CreateIllustrationPointsChangeHandler(inquiryHelper, calculations)); + CreateIllustrationPointsChangeHandler(calculations, inquiryHelper, viewCommands)); if (isNestedGroup) { @@ -1050,9 +1050,9 @@ #endregion private static ClearIllustrationPointsOfGrassCoverErosionInwardsCalculationCollectionChangeHandler CreateIllustrationPointsChangeHandler( - IInquiryHelper inquiryHelper, IEnumerable calculations) + IEnumerable calculations, IInquiryHelper inquiryHelper, IViewCommands viewCommands) { - return new ClearIllustrationPointsOfGrassCoverErosionInwardsCalculationCollectionChangeHandler(calculations, inquiryHelper); + return new ClearIllustrationPointsOfGrassCoverErosionInwardsCalculationCollectionChangeHandler(calculations, inquiryHelper, viewCommands); } private static ClearGrassCoverErosionInwardsCalculationOutputChangeHandler CreateClearCalculationOutputChangeHandler( Index: Riskeer/GrassCoverErosionInwards/test/Riskeer.GrassCoverErosionInwards.Plugin.Test/TreeNodeInfos/GrassCoverErosionInwardsCalculationGroupContextTreeNodeInfoTest.cs =================================================================== diff -u -r93c4c67fe1c358bd1670c4f9da7fe828d02218ac -rfc81f53fa4eea5508bc1b59cb8606c0583783b72 --- Riskeer/GrassCoverErosionInwards/test/Riskeer.GrassCoverErosionInwards.Plugin.Test/TreeNodeInfos/GrassCoverErosionInwardsCalculationGroupContextTreeNodeInfoTest.cs (.../GrassCoverErosionInwardsCalculationGroupContextTreeNodeInfoTest.cs) (revision 93c4c67fe1c358bd1670c4f9da7fe828d02218ac) +++ Riskeer/GrassCoverErosionInwards/test/Riskeer.GrassCoverErosionInwards.Plugin.Test/TreeNodeInfos/GrassCoverErosionInwardsCalculationGroupContextTreeNodeInfoTest.cs (.../GrassCoverErosionInwardsCalculationGroupContextTreeNodeInfoTest.cs) (revision fc81f53fa4eea5508bc1b59cb8606c0583783b72) @@ -1622,7 +1622,7 @@ } [Test] - public void GivenCalculationsWithIllustrationPoints_WhenClearIllustrationPointsClickedAndContinued_ThenInquiryAndIllustrationPointsCleared() + public void GivenCalculationsWithIllustrationPoints_WhenClearIllustrationPointsClickedAndContinued_ThenInquiryAndViewsClosedAndIllustrationPointsCleared() { // Given var calculationWithIllustrationPoints = new GrassCoverErosionInwardsCalculationScenario @@ -1668,9 +1668,14 @@ using (var treeViewControl = new TreeViewControl()) { + var viewCommands = mocks.StrictMock(); + viewCommands.Expect(vc => vc.RemoveAllViewsForItem(calculationWithIllustrationPoints.Output.OvertoppingOutput.GeneralResult)); + viewCommands.Expect(vc => vc.RemoveAllViewsForItem(calculationWithIllustrationPoints.Output.DikeHeightOutput.GeneralResult)); + viewCommands.Expect(vc => vc.RemoveAllViewsForItem(calculationWithIllustrationPoints.Output.OvertoppingRateOutput.GeneralResult)); + gui.Stub(g => g.Get(nodeData, treeViewControl)).Return(new CustomItemsOnlyContextMenuBuilder()); 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(); using (ContextMenuStrip contextMenuStrip = info.ContextMenuStrip(nodeData, null, treeViewControl)) Index: Riskeer/GrassCoverErosionInwards/test/Riskeer.GrassCoverErosionInwards.Plugin.Test/TreeNodeInfos/GrassCoverErosionInwardsFailureMechanismContextTreeNodeInfoTest.cs =================================================================== diff -u -r3f9e3cc944221ad3d31d77d151324d44bc7214cc -rfc81f53fa4eea5508bc1b59cb8606c0583783b72 --- Riskeer/GrassCoverErosionInwards/test/Riskeer.GrassCoverErosionInwards.Plugin.Test/TreeNodeInfos/GrassCoverErosionInwardsFailureMechanismContextTreeNodeInfoTest.cs (.../GrassCoverErosionInwardsFailureMechanismContextTreeNodeInfoTest.cs) (revision 3f9e3cc944221ad3d31d77d151324d44bc7214cc) +++ Riskeer/GrassCoverErosionInwards/test/Riskeer.GrassCoverErosionInwards.Plugin.Test/TreeNodeInfos/GrassCoverErosionInwardsFailureMechanismContextTreeNodeInfoTest.cs (.../GrassCoverErosionInwardsFailureMechanismContextTreeNodeInfoTest.cs) (revision fc81f53fa4eea5508bc1b59cb8606c0583783b72) @@ -1123,7 +1123,7 @@ } [Test] - public void GivenCalculationsWithIllustrationPoints_WhenClearIllustrationPointsClickedAndContinued_ThenInquiryAndIllustrationPointsCleared() + public void GivenCalculationsWithIllustrationPoints_WhenClearIllustrationPointsClickedAndContinued_ThenInquiryAndViewsClosedAndIllustrationPointsCleared() { // Given var calculationWithIllustrationPoints = new GrassCoverErosionInwardsCalculationScenario @@ -1170,10 +1170,15 @@ using (var treeViewControl = new TreeViewControl()) { + var viewCommands = mocksRepository.StrictMock(); + viewCommands.Expect(vc => vc.RemoveAllViewsForItem(calculationWithIllustrationPoints.Output.OvertoppingOutput.GeneralResult)); + viewCommands.Expect(vc => vc.RemoveAllViewsForItem(calculationWithIllustrationPoints.Output.DikeHeightOutput.GeneralResult)); + viewCommands.Expect(vc => vc.RemoveAllViewsForItem(calculationWithIllustrationPoints.Output.OvertoppingRateOutput.GeneralResult)); + var gui = mocksRepository.Stub(); gui.Stub(g => g.Get(nodeData, treeViewControl)).Return(new CustomItemsOnlyContextMenuBuilder()); gui.Stub(g => g.MainWindow).Return(mocksRepository.Stub()); - gui.Stub(g => g.ViewCommands).Return(mocksRepository.Stub()); + gui.Stub(g => g.ViewCommands).Return(viewCommands); mocksRepository.ReplayAll(); plugin.Gui = gui;