Merge remote-tracking branch 'upstream/master' into mix_20240309

1 files changed, 24 insertions, 0 deletions

diff --git a/test/functional/treesitter/node_spec.lua b/test/functional/treesitter/node_spec.lua
index d07ed35368..c87a56b160 100644
--- a/test/functional/treesitter/node_spec.lua
+++ b/test/functional/treesitter/node_spec.lua
@@ -186,4 +186,28 @@ describe('treesitter node API', function()
     )
     eq(vim.NIL, lua_eval('declarator:child_containing_descendant(value)'))
   end)
+
+  it('child_with_descendant() works', function()
+    insert([[
+      int main() {
+        int x = 3;
+      }]])
+
+    exec_lua(function()
+      local tree = vim.treesitter.get_parser(0, 'c'):parse()[1]
+      _G.root = assert(tree:root())
+      _G.main = assert(_G.root:child(0))
+      _G.body = assert(_G.main:child(2))
+      _G.statement = assert(_G.body:child(1))
+      _G.declarator = assert(_G.statement:child(1))
+      _G.value = assert(_G.declarator:child(1))
+    end)
+
+    eq(lua_eval('main:type()'), lua_eval('root:child_with_descendant(value):type()'))
+    eq(lua_eval('body:type()'), lua_eval('main:child_with_descendant(value):type()'))
+    eq(lua_eval('statement:type()'), lua_eval('body:child_with_descendant(value):type()'))
+    eq(lua_eval('declarator:type()'), lua_eval('statement:child_with_descendant(value):type()'))
+    eq(lua_eval('value:type()'), lua_eval('declarator:child_with_descendant(value):type()'))
+    eq(vim.NIL, lua_eval('value:child_with_descendant(value)'))
+  end)
 end)