Index: Ringtoets/Integration/test/Ringtoets.Integration.Forms.Test/TreeNodeInfos/AssessmentSectionTreeNodeInfoTest.cs =================================================================== diff -u -rf01bfd8d4c0541a96d57071b397c39792a99da26 -r4e6c8ac0eb663ddfe4565745fb4829a8432019e2 --- Ringtoets/Integration/test/Ringtoets.Integration.Forms.Test/TreeNodeInfos/AssessmentSectionTreeNodeInfoTest.cs (.../AssessmentSectionTreeNodeInfoTest.cs) (revision f01bfd8d4c0541a96d57071b397c39792a99da26) +++ Ringtoets/Integration/test/Ringtoets.Integration.Forms.Test/TreeNodeInfos/AssessmentSectionTreeNodeInfoTest.cs (.../AssessmentSectionTreeNodeInfoTest.cs) (revision 4e6c8ac0eb663ddfe4565745fb4829a8432019e2) @@ -261,9 +261,7 @@ { // Setup var gui = mocks.StrictMultiMock(); - var treeViewControlMock = mocks.StrictMock(); var menuBuilderMock = mocks.StrictMock(); - gui.Expect(g => g.Get(null, treeViewControlMock)).Return(menuBuilderMock); gui.Stub(g => g.ProjectOpened += null).IgnoreArguments(); gui.Stub(g => g.ProjectOpened -= null).IgnoreArguments(); @@ -281,15 +279,19 @@ menuBuilderMock.Expect(mb => mb.AddPropertiesItem()).Return(menuBuilderMock); menuBuilderMock.Expect(mb => mb.Build()).Return(null); - mocks.ReplayAll(); - - using (var plugin = new RingtoetsGuiPlugin()) + using (var treeViewControl = new TreeViewControl()) { - var info = GetInfo(plugin); - plugin.Gui = gui; + gui.Expect(g => g.Get(null, treeViewControl)).Return(menuBuilderMock); + mocks.ReplayAll(); - // Call - info.ContextMenuStrip(null, null, treeViewControlMock); + using (var plugin = new RingtoetsGuiPlugin()) + { + var info = GetInfo(plugin); + plugin.Gui = gui; + + // Call + info.ContextMenuStrip(null, null, treeViewControl); + } } // Assert mocks.VerifyAll();