Fix null error (#7412)

Azure · Dec 7, 2023 · 1c9239e · 1c9239e
1 parent 1d3f851
commit 1c9239e
Showing 1 changed file with 14 additions and 6 deletions.
diff --git a/src/dotnet/APIView/APIViewWeb/Managers/APIRevisionsManager.cs b/src/dotnet/APIView/APIViewWeb/Managers/APIRevisionsManager.cs
@@ -296,9 +296,13 @@ public async Task GetLineNumbersOfHeadingsOfSectionsWithDiff(string reviewId, AP
                         {
                             var RevisionARootNode = RevisionACodeFile.GetCodeLineSectionRoot((int)diffLine.Line.SectionKey);
                             var RevisionBRootNode = RevisionBCodeFile.GetCodeLineSectionRoot((int)diffLine.OtherLine.SectionKey);
-                            var diffSectionRoot = ComputeSectionDiff(before: RevisionARootNode, after: RevisionBRootNode, beforeFile: RevisionACodeFile, afterFile: RevisionBCodeFile);
-                            if (RevisionACodeFile.ChildNodeHasDiff(diffSectionRoot))
-                                lineNumbersForHeadingOfSectionWithDiff.Add((int)diffLine.Line.LineNumber);
+
+                            if (RevisionARootNode != null && RevisionBRootNode != null)
+                            {
+                                var diffSectionRoot = ComputeSectionDiff(before: RevisionARootNode, after: RevisionBRootNode, beforeFile: RevisionACodeFile, afterFile: RevisionBCodeFile);
+                                if (RevisionACodeFile.ChildNodeHasDiff(diffSectionRoot))
+                                    lineNumbersForHeadingOfSectionWithDiff.Add((int)diffLine.Line.LineNumber);
+                            }
                         }
                     }
 
@@ -321,9 +325,13 @@ public async Task GetLineNumbersOfHeadingsOfSectionsWithDiff(string reviewId, AP
                         {
                             var RevisionBRootNode = RevisionBCodeFile.GetCodeLineSectionRoot((int)diffLine.Line.SectionKey);
                             var RevisionARootNode = RevisionACodeFile.GetCodeLineSectionRoot((int)diffLine.OtherLine.SectionKey);
-                            var diffSectionRoot = ComputeSectionDiff(before: RevisionBRootNode, after: RevisionARootNode, beforeFile: RevisionBCodeFile, afterFile: RevisionACodeFile);
-                            if (RevisionACodeFile.ChildNodeHasDiff(diffSectionRoot))
-                                lineNumbersForHeadingOfSectionWithDiff.Add((int)diffLine.Line.LineNumber);
+
+                            if (RevisionARootNode != null && RevisionBRootNode != null)
+                            {
+                                var diffSectionRoot = ComputeSectionDiff(before: RevisionBRootNode, after: RevisionARootNode, beforeFile: RevisionBCodeFile, afterFile: RevisionACodeFile);
+                                if (RevisionACodeFile.ChildNodeHasDiff(diffSectionRoot))
+                                    lineNumbersForHeadingOfSectionWithDiff.Add((int)diffLine.Line.LineNumber);
+                            }
                         }
                     }