Index: Ringtoets/Piping/test/Ringtoets.Piping.Forms.Test/NodePresenters/PipingSurfaceLineNodePresenterTest.cs =================================================================== diff -u -rd6e6eadf4a2521df75b6d371bacbb181a43058a3 -rc1ee92fcbf17c303c6c4a6e126b102190bcffd39 --- Ringtoets/Piping/test/Ringtoets.Piping.Forms.Test/NodePresenters/PipingSurfaceLineNodePresenterTest.cs (.../PipingSurfaceLineNodePresenterTest.cs) (revision d6e6eadf4a2521df75b6d371bacbb181a43058a3) +++ Ringtoets/Piping/test/Ringtoets.Piping.Forms.Test/NodePresenters/PipingSurfaceLineNodePresenterTest.cs (.../PipingSurfaceLineNodePresenterTest.cs) (revision c1ee92fcbf17c303c6c4a6e126b102190bcffd39) @@ -1,9 +1,8 @@ using System; using System.Drawing; - -using Core.Common.Controls; using Core.Common.Controls.Swf.TreeViewControls; using Core.Common.Gui; +using Core.Common.Gui.ContextMenu; using Core.Common.TestUtil; using NUnit.Framework; @@ -82,5 +81,30 @@ TestHelper.AssertImagesAreEqual(PipingFormsResources.PipingSurfaceLineIcon, dataNodeMock.Image); mockRepository.VerifyAll(); } + + [Test] + public void GetContextMenu_Always_CallsContextMenuBuilderMethods() + { + // Setup + var contextMenuBuilderProviderMock = mockRepository.StrictMock(); + var assessmentSection = mockRepository.Stub(); + var menuBuilderMock = mockRepository.StrictMock(); + var nodeMock = mockRepository.StrictMock(); + + menuBuilderMock.Expect(mb => mb.AddPropertiesItem()).Return(menuBuilderMock); + menuBuilderMock.Expect(mb => mb.Build()).Return(null); + + contextMenuBuilderProviderMock.Expect(cmp => cmp.Get(nodeMock)).Return(menuBuilderMock); + + mockRepository.ReplayAll(); + + var nodePresenter = new PipingSurfaceLineNodePresenter(contextMenuBuilderProviderMock); + + // Call + nodePresenter.GetContextMenu(nodeMock, assessmentSection); + + // Assert + mockRepository.VerifyAll(); + } } } \ No newline at end of file