Index: Riskeer/Integration/test/Riskeer.Integration.Plugin.Test/StateInfos/CalculationsStateInfoTest.cs
===================================================================
diff -u -rec69acb95bfbd2c7686df7ddfbed9f373c1b3db4 -rfde8b80a8991006efc590fa47ffa3ddf2626c07d
--- Riskeer/Integration/test/Riskeer.Integration.Plugin.Test/StateInfos/CalculationsStateInfoTest.cs	(.../CalculationsStateInfoTest.cs)	(revision ec69acb95bfbd2c7686df7ddfbed9f373c1b3db4)
+++ Riskeer/Integration/test/Riskeer.Integration.Plugin.Test/StateInfos/CalculationsStateInfoTest.cs	(.../CalculationsStateInfoTest.cs)	(revision fde8b80a8991006efc590fa47ffa3ddf2626c07d)
@@ -24,8 +24,18 @@
 using Core.Gui.Plugin;
 using NUnit.Framework;
 using Rhino.Mocks;
+using Riskeer.ClosingStructures.Forms.PresentationObjects;
 using Riskeer.Common.Data.AssessmentSection;
+using Riskeer.DuneErosion.Forms.PresentationObjects;
+using Riskeer.GrassCoverErosionInwards.Forms.PresentationObjects;
+using Riskeer.GrassCoverErosionOutwards.Forms.PresentationObjects;
+using Riskeer.HeightStructures.Forms.PresentationObjects;
 using Riskeer.Integration.Data;
+using Riskeer.MacroStabilityInwards.Forms.PresentationObjects;
+using Riskeer.Piping.Forms.PresentationObjects;
+using Riskeer.StabilityPointStructures.Forms.PresentationObjects;
+using Riskeer.StabilityStoneCover.Forms.PresentationObjects;
+using Riskeer.WaveImpactAsphaltCover.Forms.PresentationObjects;
 
 namespace Riskeer.Integration.Plugin.Test.StateInfos
 {
@@ -75,6 +85,16 @@
             var rootDataCollection = rootData as object[];
             Assert.IsNotNull(rootDataCollection);
             Assert.AreEqual(10, rootDataCollection.Length);
+            Assert.AreEqual(new PipingFailureMechanismContext(assessmentSection.Piping, assessmentSection), rootDataCollection[0]);
+            Assert.AreEqual(new GrassCoverErosionInwardsFailureMechanismContext(assessmentSection.GrassCoverErosionInwards, assessmentSection), rootDataCollection[1]);
+            Assert.AreEqual(new MacroStabilityInwardsFailureMechanismContext(assessmentSection.MacroStabilityInwards, assessmentSection), rootDataCollection[2]);
+            Assert.AreEqual(new StabilityStoneCoverFailureMechanismContext(assessmentSection.StabilityStoneCover, assessmentSection), rootDataCollection[3]);
+            Assert.AreEqual(new WaveImpactAsphaltCoverFailureMechanismContext(assessmentSection.WaveImpactAsphaltCover, assessmentSection), rootDataCollection[4]);
+            Assert.AreEqual(new GrassCoverErosionOutwardsFailureMechanismContext(assessmentSection.GrassCoverErosionOutwards, assessmentSection), rootDataCollection[5]);
+            Assert.AreEqual(new HeightStructuresFailureMechanismContext(assessmentSection.HeightStructures, assessmentSection), rootDataCollection[6]);
+            Assert.AreEqual(new ClosingStructuresFailureMechanismContext(assessmentSection.ClosingStructures, assessmentSection), rootDataCollection[7]);
+            Assert.AreEqual(new StabilityPointStructuresFailureMechanismContext(assessmentSection.StabilityPointStructures, assessmentSection), rootDataCollection[8]);
+            Assert.AreEqual(new DuneErosionFailureMechanismContext(assessmentSection.DuneErosion, assessmentSection), rootDataCollection[9]);
         }
 
         [Test]