Index: Ringtoets/Integration/test/Ringtoets.Integration.Plugin.Test/ViewInfos/CommentViewInfoTest.cs =================================================================== diff -u -r1e6d07e8cbd1c29a16b5addddb7205e717543729 -r149c1b4d30710a88480556b70ba2b49015b1dded --- Ringtoets/Integration/test/Ringtoets.Integration.Plugin.Test/ViewInfos/CommentViewInfoTest.cs (.../CommentViewInfoTest.cs) (revision 1e6d07e8cbd1c29a16b5addddb7205e717543729) +++ Ringtoets/Integration/test/Ringtoets.Integration.Plugin.Test/ViewInfos/CommentViewInfoTest.cs (.../CommentViewInfoTest.cs) (revision 149c1b4d30710a88480556b70ba2b49015b1dded) @@ -7,6 +7,9 @@ using Ringtoets.Common.Data.AssessmentSection; using Ringtoets.Common.Forms.PresentationObjects; using Ringtoets.Common.Forms.Views; +using Ringtoets.Piping.Data; +using Ringtoets.Piping.Forms.PresentationObjects; +using Ringtoets.Piping.Primitives; using RingtoetsCommonFormsResources = Ringtoets.Common.Forms.Properties.Resources; namespace Ringtoets.Integration.Plugin.Test.ViewInfos @@ -151,5 +154,54 @@ // Assert Assert.IsFalse(closeForData); } + + [Test] + public void CloseForData_ViewCorrespondingToRemovedCalculationItem_ReturnsTrue() + { + // Setup + var calculation = new PipingCalculation(new GeneralPipingInput(), new SemiProbabilisticPipingInput()); + + var pipingFailureMechanismMock = mocks.StrictMock(); + var assessmentSectionMock = mocks.StrictMock(); + + var viewMock = mocks.StrictMock(); + var calculationContextMock = mocks.StrictMock(calculation, Enumerable.Empty(), Enumerable.Empty(), pipingFailureMechanismMock, assessmentSectionMock); + + viewMock.Expect(vm => vm.Data).Return(calculationContextMock.WrappedData); + + mocks.ReplayAll(); + + // Call + var closeForData = info.CloseForData(viewMock, calculationContextMock); + + // Assert + Assert.IsTrue(closeForData); + } + + [Test] + public void CloseForData_ViewNotCorrespondingToRemovedCalculationItem_ReturnsFalse() + { + // Setup + var calculation = new PipingCalculation(new GeneralPipingInput(), new SemiProbabilisticPipingInput()); + var calculation2 = new PipingCalculation(new GeneralPipingInput(), new SemiProbabilisticPipingInput()); + + var pipingFailureMechanismMock = mocks.StrictMock(); + var assessmentSectionMock = mocks.StrictMock(); + + var viewMock = mocks.StrictMock(); + var calculationContextMock = mocks.StrictMock(calculation, Enumerable.Empty(), Enumerable.Empty(), pipingFailureMechanismMock, assessmentSectionMock); + var calculationContextMock2 = mocks.StrictMock(calculation2, Enumerable.Empty(), Enumerable.Empty(), pipingFailureMechanismMock, assessmentSectionMock); + + viewMock.Expect(vm => vm.Data).Return(calculationContextMock2.WrappedData); + viewMock.Expect(vm => vm.AssessmentSection).Return(assessmentSectionMock); + + mocks.ReplayAll(); + + // Call + var closeForData = info.CloseForData(viewMock, calculationContextMock); + + // Assert + Assert.IsFalse(closeForData); + } } } \ No newline at end of file