Extend coverage of last sections until end of document

Author: lionel-

crates/ark/src/lsp/folding_range.rs

@@ -127,7 +127,7 @@ fn parse_ts_node(
         end_node_handler(
             document,
             folding_ranges,
-            end.row,
+            end.row + 1,
             &mut child_comment_stack,
             &mut child_region_marker,
             &mut child_cell_marker,
@@ -479,7 +479,18 @@ f
 #+ Chunk section (knitr-style)
 g
 # + This is not a chunk
-h"
+h
+"
+        ));
+    }
+
+    #[test]
+    fn test_folding_section_comments_no_trailing_empty_line() {
+        insta::assert_debug_snapshot!(test_folding_range(
+            "
+# First section ----
+a
+b"
         ));
     }
 

crates/ark/src/lsp/snapshots/ark__lsp__folding_range__tests__folding_complex_nested.snap

@@ -6,7 +6,7 @@ expression: "test_folding_range(\"\n# Complex example ----\nfunction(a, b, c) {\
     FoldingRange {
         start_line: 1,
         start_character: None,
-        end_line: 18,
+        end_line: 19,
         end_character: None,
         kind: Some(
             Region,
@@ -62,7 +62,7 @@ expression: "test_folding_range(\"\n# Complex example ----\nfunction(a, b, c) {\
     FoldingRange {
         start_line: 18,
         start_character: None,
-        end_line: 18,
+        end_line: 19,
         end_character: None,
         kind: Some(
             Region,

crates/ark/src/lsp/snapshots/ark__lsp__folding_range__tests__folding_empty_sections.snap

@@ -26,7 +26,7 @@ expression: "test_folding_range(\"\n# Empty section ----\n\n# Another empty sect
     FoldingRange {
         start_line: 5,
         start_character: None,
-        end_line: 5,
+        end_line: 6,
         end_character: None,
         kind: Some(
             Region,

crates/ark/src/lsp/snapshots/ark__lsp__folding_range__tests__folding_mixed.snap

@@ -58,7 +58,7 @@ expression: "test_folding_range(\"\n# First section ----\nfunction() {\n  # #reg
     FoldingRange {
         start_line: 12,
         start_character: None,
-        end_line: 12,
+        end_line: 13,
         end_character: None,
         kind: Some(
             Region,

crates/ark/src/lsp/snapshots/ark__lsp__folding_range__tests__folding_nested_section_comments.snap

@@ -46,7 +46,7 @@ expression: "test_folding_range(\"\n# Level 1 ----\na\n\n## Level 2 ----\nb\n\n#
     FoldingRange {
         start_line: 13,
         start_character: None,
-        end_line: 19,
+        end_line: 20,
         end_character: None,
         kind: Some(
             Region,

crates/ark/src/lsp/snapshots/ark__lsp__folding_range__tests__folding_section_comments_basic.snap

@@ -16,7 +16,7 @@ expression: "test_folding_range(\"\n# First section ----\na\nb\n\n# Second secti
     FoldingRange {
         start_line: 5,
         start_character: None,
-        end_line: 15,
+        end_line: 16,
         end_character: None,
         kind: Some(
             Region,

crates/ark/src/lsp/snapshots/ark__lsp__folding_range__tests__folding_section_comments_no_trailing_empty_line.snap

@@ -0,0 +1,16 @@
+---
+source: crates/ark/src/lsp/folding_range.rs
+expression: "test_folding_range(\"\n# First section ----\na\nb\")"
+---
+[
+    FoldingRange {
+        start_line: 1,
+        start_character: None,
+        end_line: 3,
+        end_character: None,
+        kind: Some(
+            Region,
+        ),
+        collapsed_text: None,
+    },
+]