Index: Ringtoets/Integration/test/Ringtoets.Integration.Forms.Test/PresentationObjects/DesignWaterLevelLocationContextTest.cs
===================================================================
diff -u -r7594f72cabeb4802ffab5715e2edb2712fc1f0df -re649822ccb64af259adda08b8aa096bf08bf1a87
--- Ringtoets/Integration/test/Ringtoets.Integration.Forms.Test/PresentationObjects/DesignWaterLevelLocationContextTest.cs	(.../DesignWaterLevelLocationContextTest.cs)	(revision 7594f72cabeb4802ffab5715e2edb2712fc1f0df)
+++ Ringtoets/Integration/test/Ringtoets.Integration.Forms.Test/PresentationObjects/DesignWaterLevelLocationContextTest.cs	(.../DesignWaterLevelLocationContextTest.cs)	(revision e649822ccb64af259adda08b8aa096bf08bf1a87)
@@ -30,48 +30,19 @@
     public class DesignWaterLevelLocationContextTest
     {
         [Test]
-        public void Constructor_NullHydraulicBoundaryLocation_ThrowsArgumentNullException()
-        {
-            // Setup
-            var hydraulicBoundaryDatabase = new HydraulicBoundaryDatabase();
-
-            // Call
-            TestDelegate test = () => new DesignWaterLevelLocationContext(null, hydraulicBoundaryDatabase);
-
-            // Assert
-            string paramName = Assert.Throws<ArgumentNullException>(test).ParamName;
-            Assert.AreEqual("wrappedData", paramName);
-        }
-
-        [Test]
-        public void Constructor_NullHydraulicBoundariesDatabase_ThrowsArgumentNullException()
-        {
-            // Setup
-            var hydraulicBoundaryLocation = new HydraulicBoundaryLocation(1, "name", 2.0, 3.0);
-
-            // Call
-            TestDelegate test = () => new DesignWaterLevelLocationContext(hydraulicBoundaryLocation, null);
-
-            // Assert
-            string paramName = Assert.Throws<ArgumentNullException>(test).ParamName;
-            Assert.AreEqual("hydraulicBoundaryDatabase", paramName);
-        }
-
-        [Test]
         public void Constructor_ValidParameters_ExpectedValues()
         {
             // Setup
             var hydraulicBoundaryLocation = new HydraulicBoundaryLocation(1, "name", 2.0, 3.0);
-            var hydraulicBoundaryDatabase = new HydraulicBoundaryDatabase();
-            hydraulicBoundaryDatabase.Locations.Add(hydraulicBoundaryLocation);
+            Func<HydraulicBoundaryLocation, HydraulicBoundaryLocationCalculation> getCalculationFunc = hbl => null;
 
             // Call
-            var presentationObject = new DesignWaterLevelLocationContext(hydraulicBoundaryLocation, hydraulicBoundaryDatabase);
+            var presentationObject = new DesignWaterLevelLocationContext(hydraulicBoundaryLocation, getCalculationFunc);
 
             // Assert
             Assert.IsInstanceOf<HydraulicBoundaryLocationContext>(presentationObject);
-            Assert.AreSame(hydraulicBoundaryDatabase, presentationObject.HydraulicBoundaryDatabase);
             Assert.AreSame(hydraulicBoundaryLocation, presentationObject.WrappedData);
+            Assert.AreSame(getCalculationFunc, presentationObject.GetCalculationFunc);
         }
     }
 }
\ No newline at end of file