From 2985c17867bec5dfdc77f1bae123b50cac533f1b Mon Sep 17 00:00:00 2001
From: Steven Sojka <steelsojka@gmail.com>
Date: Fri, 4 Dec 2020 16:15:47 -0600
Subject: [PATCH] fix(treesitter): incorrect method name call

---
 runtime/lua/vim/treesitter/languagetree.lua |  2 +-
 test/functional/lua/treesitter_spec.lua     | 21 +++++++++++++++++++++
 2 files changed, 22 insertions(+), 1 deletion(-)

diff --git a/runtime/lua/vim/treesitter/languagetree.lua b/runtime/lua/vim/treesitter/languagetree.lua
index a8b62e21b9..885460c9fd 100644
--- a/runtime/lua/vim/treesitter/languagetree.lua
+++ b/runtime/lua/vim/treesitter/languagetree.lua
@@ -445,7 +445,7 @@ end
 function LanguageTree:language_for_range(range)
   for _, child in pairs(self._children) do
     if child:contains(range) then
-      return child:node_for_range(range)
+      return child:language_for_range(range)
     end
   end
 
diff --git a/test/functional/lua/treesitter_spec.lua b/test/functional/lua/treesitter_spec.lua
index 65dc1b3e03..0e823426ae 100644
--- a/test/functional/lua/treesitter_spec.lua
+++ b/test/functional/lua/treesitter_spec.lua
@@ -919,4 +919,25 @@ local hl_query = [[
     end)
   end)
 
+  describe("when getting the language for a range", function()
+    before_each(function()
+      insert([[
+int x = INT_MAX;
+#define VALUE 123456789
+      ]])
+    end)
+
+    it("should return the correct language tree", function()
+      local result = exec_lua([[
+      parser = vim.treesitter.get_parser(0, "c", {
+        queries = { c = "(preproc_def (preproc_arg) @c)"}})
+
+      local sub_tree = parser:language_for_range({1, 18, 1, 19})
+
+      return sub_tree == parser:children().c
+      ]])
+
+      eq(result, true)
+    end)
+  end)
 end)