Add test with argument enabled

Author: lionel-

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

@@ -0,0 +1,131 @@
+---
+source: crates/ark/src/lsp/symbols.rs
+expression: symbols
+---
+[
+    DocumentSymbol {
+        name: "inner1",
+        detail: None,
+        kind: Variable,
+        tags: None,
+        deprecated: None,
+        range: Range {
+            start: Position {
+                line: 2,
+                character: 2,
+            },
+            end: Position {
+                line: 2,
+                character: 8,
+            },
+        },
+        selection_range: Range {
+            start: Position {
+                line: 2,
+                character: 2,
+            },
+            end: Position {
+                line: 2,
+                character: 8,
+            },
+        },
+        children: Some(
+            [],
+        ),
+    },
+    DocumentSymbol {
+        name: "a",
+        detail: Some(
+            "function()",
+        ),
+        kind: Function,
+        tags: None,
+        deprecated: None,
+        range: Range {
+            start: Position {
+                line: 4,
+                character: 0,
+            },
+            end: Position {
+                line: 7,
+                character: 1,
+            },
+        },
+        selection_range: Range {
+            start: Position {
+                line: 4,
+                character: 0,
+            },
+            end: Position {
+                line: 7,
+                character: 1,
+            },
+        },
+        children: Some(
+            [
+                DocumentSymbol {
+                    name: "inner2",
+                    detail: None,
+                    kind: Variable,
+                    tags: None,
+                    deprecated: None,
+                    range: Range {
+                        start: Position {
+                            line: 5,
+                            character: 2,
+                        },
+                        end: Position {
+                            line: 5,
+                            character: 8,
+                        },
+                    },
+                    selection_range: Range {
+                        start: Position {
+                            line: 5,
+                            character: 2,
+                        },
+                        end: Position {
+                            line: 5,
+                            character: 8,
+                        },
+                    },
+                    children: Some(
+                        [],
+                    ),
+                },
+                DocumentSymbol {
+                    name: "inner3",
+                    detail: Some(
+                        "function()",
+                    ),
+                    kind: Function,
+                    tags: None,
+                    deprecated: None,
+                    range: Range {
+                        start: Position {
+                            line: 6,
+                            character: 2,
+                        },
+                        end: Position {
+                            line: 6,
+                            character: 24,
+                        },
+                    },
+                    selection_range: Range {
+                        start: Position {
+                            line: 6,
+                            character: 2,
+                        },
+                        end: Position {
+                            line: 6,
+                            character: 24,
+                        },
+                    },
+                    children: Some(
+                        [],
+                    ),
+                },
+            ],
+        ),
+    },
+]

crates/ark/src/lsp/symbols.rs

@@ -849,4 +849,29 @@ a <- function() {
         ));
         assert_eq!(test_symbol("{ foo <- 1 }"), vec![]);
     }
+
+    #[test]
+    fn test_symbol_nested_assignments_enabled() {
+        let doc = Document::new(
+            "
+local({
+  inner1 <- 1
+})
+a <- function() {
+  inner2 <- 2
+  inner3 <- function() 3
+}
+",
+            None,
+        );
+        let node = doc.ast.root_node();
+
+        let ctx = &mut CollectContext::new();
+        ctx.include_assignments_in_blocks = true;
+
+        let mut symbols = Vec::new();
+        collect_symbols(ctx, &node, &doc.contents, 0, &mut symbols).unwrap();
+
+        insta::assert_debug_snapshot!(symbols);
+    }
 }