Index: Riskeer/WaveImpactAsphaltCover/src/Riskeer.WaveImpactAsphaltCover.Plugin/WaveImpactAsphaltCoverPlugin.cs =================================================================== diff -u -r2f8a00af2f3ec9b31b745d2e11d151c6322a7d73 -ra149b857c8f62494ab59a123646e8ab56588a06e --- Riskeer/WaveImpactAsphaltCover/src/Riskeer.WaveImpactAsphaltCover.Plugin/WaveImpactAsphaltCoverPlugin.cs (.../WaveImpactAsphaltCoverPlugin.cs) (revision 2f8a00af2f3ec9b31b745d2e11d151c6322a7d73) +++ Riskeer/WaveImpactAsphaltCover/src/Riskeer.WaveImpactAsphaltCover.Plugin/WaveImpactAsphaltCoverPlugin.cs (.../WaveImpactAsphaltCoverPlugin.cs) (revision a149b857c8f62494ab59a123646e8ab56588a06e) @@ -281,7 +281,7 @@ private static bool CloseFailureMechanismResultViewForData(WaveImpactAsphaltCoverFailureMechanismResultView view, object dataToCloseFor) { - WaveImpactAsphaltCoverFailureMechanism failureMechanism = null; + var failureMechanism = dataToCloseFor as WaveImpactAsphaltCoverFailureMechanism; if (dataToCloseFor is IAssessmentSection assessmentSection) { Index: Riskeer/WaveImpactAsphaltCover/test/Riskeer.WaveImpactAsphaltCover.Plugin.Test/ViewInfos/WaveImpactAsphaltCoverFailureMechanismResultViewInfoTest.cs =================================================================== diff -u -raa4f44030799130241257aa7e2e10f7dd7017cc6 -ra149b857c8f62494ab59a123646e8ab56588a06e --- Riskeer/WaveImpactAsphaltCover/test/Riskeer.WaveImpactAsphaltCover.Plugin.Test/ViewInfos/WaveImpactAsphaltCoverFailureMechanismResultViewInfoTest.cs (.../WaveImpactAsphaltCoverFailureMechanismResultViewInfoTest.cs) (revision aa4f44030799130241257aa7e2e10f7dd7017cc6) +++ Riskeer/WaveImpactAsphaltCover/test/Riskeer.WaveImpactAsphaltCover.Plugin.Test/ViewInfos/WaveImpactAsphaltCoverFailureMechanismResultViewInfoTest.cs (.../WaveImpactAsphaltCoverFailureMechanismResultViewInfoTest.cs) (revision a149b857c8f62494ab59a123646e8ab56588a06e) @@ -183,6 +183,38 @@ } [Test] + public void CloseForData_ViewCorrespondingToRemovedFailureMechanism_ReturnsTrue() + { + // Setup + var failureMechanism = new WaveImpactAsphaltCoverFailureMechanism(); + + using (var view = new WaveImpactAsphaltCoverFailureMechanismResultView(failureMechanism.SectionResults, failureMechanism)) + { + // Call + bool closeForData = info.CloseForData(view, failureMechanism); + + // Assert + Assert.IsTrue(closeForData); + } + } + + [Test] + public void CloseForData_ViewNotCorrespondingToRemovedFailureMechanism_ReturnsFalse() + { + // Setup + var failureMechanism = new WaveImpactAsphaltCoverFailureMechanism(); + + using (var view = new WaveImpactAsphaltCoverFailureMechanismResultView(failureMechanism.SectionResults, failureMechanism)) + { + // Call + bool closeForData = info.CloseForData(view, new WaveImpactAsphaltCoverFailureMechanism()); + + // Assert + Assert.IsFalse(closeForData); + } + } + + [Test] public void CloseForData_ViewCorrespondingToRemovedFailurePathContext_ReturnsTrue() { // Setup