Index: Ringtoets/Piping/test/Ringtoets.Piping.Forms.Test/NodePresenters/PipingSoilProfileNodePresenterTest.cs =================================================================== diff -u -rd6e6eadf4a2521df75b6d371bacbb181a43058a3 -rc1ee92fcbf17c303c6c4a6e126b102190bcffd39 --- Ringtoets/Piping/test/Ringtoets.Piping.Forms.Test/NodePresenters/PipingSoilProfileNodePresenterTest.cs (.../PipingSoilProfileNodePresenterTest.cs) (revision d6e6eadf4a2521df75b6d371bacbb181a43058a3) +++ Ringtoets/Piping/test/Ringtoets.Piping.Forms.Test/NodePresenters/PipingSoilProfileNodePresenterTest.cs (.../PipingSoilProfileNodePresenterTest.cs) (revision c1ee92fcbf17c303c6c4a6e126b102190bcffd39) @@ -1,10 +1,9 @@ using System; using System.Collections.Generic; 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; using Rhino.Mocks; @@ -90,5 +89,38 @@ TestHelper.AssertImagesAreEqual(PipingFormsResources.PipingSoilProfileIcon, dataNodeMock.Image); mockRepository.VerifyAll(); } + + [Test] + public void GetContextMenu_Always_CallsContextMenuBuilderMethods() + { + // Setup + IEnumerable layers = new[] + { + new PipingSoilLayer(1) + { + IsAquifer = true + } + }; + + var contextMenuBuilderProviderMock = mockRepository.StrictMock(); + var assessmentSection = mockRepository.Stub(string.Empty, 0, layers); + 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 PipingSoilProfileNodePresenter(contextMenuBuilderProviderMock); + + // Call + nodePresenter.GetContextMenu(nodeMock, assessmentSection); + + // Assert + mockRepository.VerifyAll(); + } } } \ No newline at end of file