From 4122ca344f0b63c6627f18e7d784288b0b83c0dd Mon Sep 17 00:00:00 2001
From: Spencer O'HEGARTY <Spencer.OHEGARTY@EDUCATION.GOV.UK>
Date: Wed, 10 Jul 2024 10:16:16 +0100
Subject: [PATCH] Fixed possible DocumentObjectModel null reference issue in
 DocumentObjectModelExtractor.

---
 .../Web/Tests/Integration/PageObjectModel/HomePage.cs           | 1 +
 .../PageObjectModel/Setup/DocumentObjectModelExtractor.cs       | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/Dfe.Data.SearchPrototype/Web/Tests/Integration/PageObjectModel/HomePage.cs b/Dfe.Data.SearchPrototype/Web/Tests/Integration/PageObjectModel/HomePage.cs
index fb806a7..a4ce33d 100644
--- a/Dfe.Data.SearchPrototype/Web/Tests/Integration/PageObjectModel/HomePage.cs
+++ b/Dfe.Data.SearchPrototype/Web/Tests/Integration/PageObjectModel/HomePage.cs
@@ -18,6 +18,7 @@ public sealed class HomePage : DocumentObjectModelExtractor
         public HomePage(WebApplicationFactory<Program> webApplicationFactory) :
             base(webApplicationFactory, PageName)
         {
+            ArgumentNullException.ThrowIfNull(DocumentObjectModel);
             _pageHeader = PageHeader.Create(DocumentObjectModel);
             _pageBody = PageBody.Create(DocumentObjectModel);
         }
diff --git a/Dfe.Data.SearchPrototype/Web/Tests/Integration/PageObjectModel/Setup/DocumentObjectModelExtractor.cs b/Dfe.Data.SearchPrototype/Web/Tests/Integration/PageObjectModel/Setup/DocumentObjectModelExtractor.cs
index 95f47b5..efead69 100644
--- a/Dfe.Data.SearchPrototype/Web/Tests/Integration/PageObjectModel/Setup/DocumentObjectModelExtractor.cs
+++ b/Dfe.Data.SearchPrototype/Web/Tests/Integration/PageObjectModel/Setup/DocumentObjectModelExtractor.cs
@@ -6,7 +6,7 @@ namespace DfE.Data.SearchPrototype.Web.Tests.Integration.PageObjectModel.Setup;
 
 public abstract class DocumentObjectModelExtractor : WebApplicationBootstrapper
 {
-    protected IDocument DocumentObjectModel { get; private set; }
+    protected IDocument? DocumentObjectModel { get; private set; }
 
     protected DocumentObjectModelExtractor(
         WebApplicationFactory<Program> webApplicationFactory, string? pageName)