Index: Riskeer/StabilityStoneCover/src/Riskeer.StabilityStoneCover.Plugin/StabilityStoneCoverPlugin.cs =================================================================== diff -u -r3fc5872154df91afbec31b23be55faee8ab87091 -r82fa24f1a3d7cbcbe2cea897b43b75d4509a60de --- Riskeer/StabilityStoneCover/src/Riskeer.StabilityStoneCover.Plugin/StabilityStoneCoverPlugin.cs (.../StabilityStoneCoverPlugin.cs) (revision 3fc5872154df91afbec31b23be55faee8ab87091) +++ Riskeer/StabilityStoneCover/src/Riskeer.StabilityStoneCover.Plugin/StabilityStoneCoverPlugin.cs (.../StabilityStoneCoverPlugin.cs) (revision 82fa24f1a3d7cbcbe2cea897b43b75d4509a60de) @@ -282,7 +282,7 @@ private static bool CloseFailureMechanismResultViewForData(StabilityStoneCoverResultView view, object dataToCloseFor) { - StabilityStoneCoverFailureMechanism failureMechanism = null; + var failureMechanism = dataToCloseFor as StabilityStoneCoverFailureMechanism; if (dataToCloseFor is IAssessmentSection assessmentSection) { Index: Riskeer/StabilityStoneCover/test/Riskeer.StabilityStoneCover.Plugin.Test/ViewInfos/StabilityStoneCoverResultViewInfoTest.cs =================================================================== diff -u -rf5503aa2bc6bbeafb29a77b70e4162fa09473364 -r82fa24f1a3d7cbcbe2cea897b43b75d4509a60de --- Riskeer/StabilityStoneCover/test/Riskeer.StabilityStoneCover.Plugin.Test/ViewInfos/StabilityStoneCoverResultViewInfoTest.cs (.../StabilityStoneCoverResultViewInfoTest.cs) (revision f5503aa2bc6bbeafb29a77b70e4162fa09473364) +++ Riskeer/StabilityStoneCover/test/Riskeer.StabilityStoneCover.Plugin.Test/ViewInfos/StabilityStoneCoverResultViewInfoTest.cs (.../StabilityStoneCoverResultViewInfoTest.cs) (revision 82fa24f1a3d7cbcbe2cea897b43b75d4509a60de) @@ -179,6 +179,38 @@ } [Test] + public void CloseForData_ViewCorrespondingToRemovedFailureMechanism_ReturnsTrue() + { + // Setup + var failureMechanism = new StabilityStoneCoverFailureMechanism(); + + using (var view = new StabilityStoneCoverResultView(failureMechanism.SectionResults, failureMechanism)) + { + // Call + bool closeForData = info.CloseForData(view, failureMechanism); + + // Assert + Assert.IsTrue(closeForData); + } + } + + [Test] + public void CloseForData_ViewNotCorrespondingToRemovedFailureMechanism_ReturnsFalse() + { + // Setup + var failureMechanism = new StabilityStoneCoverFailureMechanism(); + + using (var view = new StabilityStoneCoverResultView(failureMechanism.SectionResults, failureMechanism)) + { + // Call + bool closeForData = info.CloseForData(view, new StabilityStoneCoverFailureMechanism()); + + // Assert + Assert.IsFalse(closeForData); + } + } + + [Test] public void CloseForData_ViewCorrespondingToRemovedFailurePathContext_ReturnsTrue() { // Setup