Index: Ringtoets/Integration/test/Ringtoets.Integration.Forms.Test/Dialogs/AssessmentSectionProviderStubTest.cs
===================================================================
diff -u -r3c4b3ed1591216f9b3b40534e35c6df156495514 -r19363f8d54d6b2a71613fa87e3c1d3c86c9c0af2
--- Ringtoets/Integration/test/Ringtoets.Integration.Forms.Test/Dialogs/AssessmentSectionProviderStubTest.cs	(.../AssessmentSectionProviderStubTest.cs)	(revision 3c4b3ed1591216f9b3b40534e35c6df156495514)
+++ Ringtoets/Integration/test/Ringtoets.Integration.Forms.Test/Dialogs/AssessmentSectionProviderStubTest.cs	(.../AssessmentSectionProviderStubTest.cs)	(revision 19363f8d54d6b2a71613fa87e3c1d3c86c9c0af2)
@@ -20,6 +20,7 @@
 // All rights reserved.
 
 using System.Collections.Generic;
+using System.Linq;
 using System.Windows.Forms;
 using Core.Common.Controls.Dialogs;
 using NUnit.Extensions.Forms;
@@ -56,11 +57,14 @@
         public void GetAssessmentSections_Always_ShowsDialog()
         {
             // Setup
+            Button cancelButton = null;
+
             DialogBoxHandler = (name, wnd) =>
             {
                 using (new FormTester(name))
                 {
                     var button = new ButtonTester("cancelButton", name);
+                    cancelButton = (Button) button.TheObject;
                     button.Click();
                 }
             };
@@ -72,20 +76,20 @@
                 provider.GetAssessmentSections(null);
 
                 // Assert
-                var invalidProjectButtonSelect = (Button)new ButtonTester("invalidProjectButton", provider).TheObject;
+                var invalidProjectButtonSelect = (Button) new ButtonTester("invalidProjectButton", provider).TheObject;
                 Assert.AreEqual("Selecteer fout project", invalidProjectButtonSelect.Text);
                 Assert.IsTrue(invalidProjectButtonSelect.Enabled);
 
-                var noMatchButtonSelect = (Button)new ButtonTester("noMatchButton", provider).TheObject;
+                var noMatchButtonSelect = (Button) new ButtonTester("noMatchButton", provider).TheObject;
                 Assert.AreEqual("Selecteer project zonder overeenkomende trajecten", noMatchButtonSelect.Text);
                 Assert.IsTrue(noMatchButtonSelect.Enabled);
 
-                var matchButtonSelect = (Button)new ButtonTester("matchButton", provider).TheObject;
+                var matchButtonSelect = (Button) new ButtonTester("matchButton", provider).TheObject;
                 Assert.AreEqual("Selecteer project met overeenkomend traject", matchButtonSelect.Text);
                 Assert.IsTrue(matchButtonSelect.Enabled);
 
-                var buttonCancel = (Button)new ButtonTester("cancelButton", provider).TheObject;
-                Assert.AreEqual("Annuleren", buttonCancel.Text);
+                Assert.AreEqual("Annuleren", cancelButton.Text);
+                Assert.AreEqual(cancelButton, provider.CancelButton);
 
                 Assert.AreEqual(1, provider.MinimumSize.Width);
                 Assert.AreEqual(1, provider.MinimumSize.Height);
@@ -96,14 +100,11 @@
         public void GivenDialog_WhenCancelPressed_ThenReturnNull()
         {
             // Setup
-            Button cancelButton = null;
-
             DialogBoxHandler = (name, wnd) =>
             {
                 using (new FormTester(name))
                 {
                     var button = new ButtonTester("cancelButton", name);
-                    cancelButton = (Button) button.TheObject;
                     button.Click();
                 }
             };
@@ -116,8 +117,79 @@
 
                 // Assert
                 Assert.IsNull(assessmentSections);
-                Assert.AreEqual(provider.CancelButton, cancelButton);
             }
         }
+
+        [Test]
+        public void GivenDialog_WhenInvalidProjectButtonPressed_ThenReturnNull()
+        {
+            // Setup
+            DialogBoxHandler = (name, wnd) =>
+            {
+                using (new FormTester(name))
+                {
+                    var button = new ButtonTester("invalidProjectButton", name);
+                    button.Click();
+                }
+            };
+
+            using (var dialogParent = new Form())
+            using (var provider = new AssessmentSectionProviderStub(dialogParent))
+            {
+                // Call
+                IEnumerable<AssessmentSection> assessmentSections = provider.GetAssessmentSections(null);
+
+                // Assert
+                Assert.IsNull(assessmentSections);
+            }
+        }
+
+        [Test]
+        public void GivenDialog_WhenNoMatchButtonPressed_ThenReturnEmptyCollection()
+        {
+            // Setup
+            DialogBoxHandler = (name, wnd) =>
+            {
+                using (new FormTester(name))
+                {
+                    var button = new ButtonTester("noMatchButton", name);
+                    button.Click();
+                }
+            };
+
+            using (var dialogParent = new Form())
+            using (var provider = new AssessmentSectionProviderStub(dialogParent))
+            {
+                // Call
+                IEnumerable<AssessmentSection> assessmentSections = provider.GetAssessmentSections(null);
+
+                // Assert
+                CollectionAssert.IsEmpty(assessmentSections);
+            }
+        }
+
+        [Test]
+        public void GivenDialog_WhenMatchButtonPressed_ThenReturnCollectionWithOneAssessmentSection()
+        {
+            // Setup
+            DialogBoxHandler = (name, wnd) =>
+            {
+                using (new FormTester(name))
+                {
+                    var button = new ButtonTester("matchButton", name);
+                    button.Click();
+                }
+            };
+
+            using (var dialogParent = new Form())
+            using (var provider = new AssessmentSectionProviderStub(dialogParent))
+            {
+                // Call
+                IEnumerable<AssessmentSection> assessmentSections = provider.GetAssessmentSections(null);
+
+                // Assert
+                Assert.AreEqual(1, assessmentSections.Count());
+            }
+        }
     }
 }
\ No newline at end of file