Index: Ringtoets/Common/test/Ringtoets.Common.Forms.Test/TreeNodeInfos/CalculationTreeNodeInfoFactoryTest.cs =================================================================== diff -u -r97517f6a66f66d512315618a08fa21a9999624a1 -r3bd79dc878fa0b81b1ebb5c35dd5612f2bfed356 --- Ringtoets/Common/test/Ringtoets.Common.Forms.Test/TreeNodeInfos/CalculationTreeNodeInfoFactoryTest.cs (.../CalculationTreeNodeInfoFactoryTest.cs) (revision 97517f6a66f66d512315618a08fa21a9999624a1) +++ Ringtoets/Common/test/Ringtoets.Common.Forms.Test/TreeNodeInfos/CalculationTreeNodeInfoFactoryTest.cs (.../CalculationTreeNodeInfoFactoryTest.cs) (revision 3bd79dc878fa0b81b1ebb5c35dd5612f2bfed356) @@ -428,6 +428,60 @@ Assert.AreEqual(1, counter); } + [Test] + public void AddPerformCalculationItem_Always_CreatesPerformItem() + { + // Setup + var mocks = new MockRepository(); + var applicationFeatureCommandHandler = mocks.Stub(); + var exportImportHandler = mocks.Stub(); + var viewCommandsHandler = mocks.StrictMock(); + var treeViewControl = mocks.StrictMock(); + var calculation = mocks.StrictMock(); + + mocks.ReplayAll(); + + var menubuilder = new ContextMenuBuilder(applicationFeatureCommandHandler, exportImportHandler, viewCommandsHandler, calculation, treeViewControl); + + // Call + CalculationTreeNodeInfoFactory.AddPerformCalculationItem(menubuilder, calculation, null); + + // Assert + TestHelper.AssertContextMenuStripContainsItem(menubuilder.Build(), 0, + RingtoetsFormsResources.Calculate, + RingtoetsFormsResources.Calculate_ToolTip, + RingtoetsFormsResources.CalculateIcon); + + mocks.VerifyAll(); + } + + [Test] + public void AddPerformCalculationItem_PerformClickOnCreatedItem_PerformCalculationMethod() + { + // Setup + var mocks = new MockRepository(); + var applicationFeatureCommandHandler = mocks.Stub(); + var exportImportHandler = mocks.Stub(); + var viewCommandsHandler = mocks.StrictMock(); + var treeViewControl = mocks.StrictMock(); + var calculation = mocks.StrictMock(); + + mocks.ReplayAll(); + + var counter = 0; + + var menuBuilder = new ContextMenuBuilder(applicationFeatureCommandHandler, exportImportHandler, viewCommandsHandler, calculation, treeViewControl); + + CalculationTreeNodeInfoFactory.AddPerformCalculationItem(menuBuilder, calculation, context => counter++); + var contextMenuItem = menuBuilder.Build().Items[0]; + + // Call + contextMenuItem.PerformClick(); + + // Assert + Assert.AreEqual(1, counter); + } + # region CreateCalculationGroupContextTreeNodeInfo [Test]