Fixed possible DocumentObjectModel null reference issue in DocumentOb…

…jectModelExtractor.
DFE-Digital · Jul 10, 2024 · 4122ca3 · 4122ca3
1 parent baf3ae5
commit 4122ca3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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/...archPrototype/Web/Tests/Integration/PageObjectModel/Setup/DocumentObjectModelExtractor.cs b/...archPrototype/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)