fix: handle textBodyDB entity being empty - possible fix

DFE-Digital · Jan 25, 2024 · 5ad8a11 · 5ad8a11
1 parent 34661b2
commit 5ad8a11
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 7 deletions.
diff --git a/src/Dfe.PlanTech.Domain/Content/Interfaces/IHasRichText.cs b/src/Dfe.PlanTech.Domain/Content/Interfaces/IHasRichText.cs
@@ -5,5 +5,5 @@ namespace Dfe.PlanTech.Domain.Content.Interfaces;
 public interface IHasRichText
 {
     public RichTextContentDbEntity RichText { get; set; }
-    public long RichTextId { get; }
+    public long? RichTextId { get; }
 }
diff --git a/src/Dfe.PlanTech.Domain/Content/Models/TextBodyDbEntity.cs b/src/Dfe.PlanTech.Domain/Content/Models/TextBodyDbEntity.cs
@@ -11,7 +11,7 @@ public class TextBodyDbEntity : ContentComponentDbEntity, ITextBody<RichTextCont
 {
     public RichTextContentDbEntity RichText { get; set; } = null!;
 
-    public long RichTextId { get; set; }
+    public long? RichTextId { get; set; }
 
     public List<WarningComponentDbEntity> Warnings { get; set; } = new();
 }
diff --git a/src/Dfe.PlanTech.Domain/Content/Models/WarningComponentDbEntity.cs b/src/Dfe.PlanTech.Domain/Content/Models/WarningComponentDbEntity.cs
@@ -9,13 +9,20 @@ public class WarningComponentDbEntity : ContentComponentDbEntity, IWarningCompon
     public string TextId { get; set; } = null!;
     public TextBodyDbEntity Text { get; set; } = null!;
 
-    [NotMapped]
+    [NotMapped][DontCopyValue]
     public RichTextContentDbEntity RichText
     {
-        get => Text.RichText;
-        set => Text.RichText = value;
+        get => Text?.RichText;
+        set
+        {
+            if (Text != null)
+            {
+                Text.RichText = value;
+            }
+        }
     }
 
-    [NotMapped]
-    public long RichTextId => Text.RichTextId;
+    [NotMapped][DontCopyValue]
+    public long? RichTextId => Text?.RichTextId;
+
 }