nvim-treesitter · PeterCardenas · Apr 12, 2025 · Apr 12, 2025
diff --git a/test/contexts_spec.lua b/test/contexts_spec.lua
@@ -114,6 +114,34 @@ for _, lang in ipairs(langs_with_queries) do
       setup(function()
         cmd([[let $XDG_CACHE_HOME='scratch/cache']])
         install_langs(lang)
+        cmd('edit ' .. test_file)
+        local bufnr = api.nvim_get_current_buf()
+        exec_lua(
+          [[
+            local bufnr, lang = ...
+            local root_parser = vim.treesitter.get_parser(bufnr, lang)
+            local line_count = vim.api.nvim_buf_line_count(bufnr)
+            root_parser:parse({ 0, line_count })
+            local langs_to_iterate = vim.tbl_keys(root_parser:children())
+            while #langs_to_iterate > 0 do
+              local current_lang = table.remove(langs_to_iterate, 1)
+              require('nvim-treesitter.configs').setup({
+                ensure_installed = current_lang,
+                sync_install = true,
+              })
+              local ok, current_parser = pcall(vim.treesitter.get_parser, bufnr, current_lang)
+              if ok then
+                current_parser:parse({ 0, line_count })
+                local child_langs = vim.tbl_keys(current_parser:children())
+                for _, child in ipairs(child_langs) do
+                  table.insert(langs_to_iterate, child)
+                end
+              end
+            end
+          ]],
+          bufnr,
+          lang
+        )
       end)
 
       for cursor_row, context_rows in pairs(contexts) do

diff --git a/test/lang/test.html b/test/lang/test.html
@@ -48,7 +48,7 @@
 
 
 
-      color: red; /* BUG: cannot test for injected lang here */
+      color: red; /* {{CURSOR}} */
 
 
 
@@ -133,8 +133,14 @@
 
 
     var b = 2;
-    function test() {
+    function test() { // {{CONTEXT}}
       let test = "asdasd";
+
+
+
+
+
+      // {{CURSOR}}
     }
 
     var c = a + b;

diff --git a/test/lang/test.md b/test/lang/test.md
@@ -1,13 +1,13 @@
-```html
-<html>
-  <body>
+```html <!-- {{CONTEXT}} -->
+<html> <!-- {{CONTEXT}} -->
+  <body> <!-- {{CONTEXT}} -->
 
 
 
 
 
 
-    <script>
+    <script> <!-- {{CONTEXT}} -->
 
 
 
@@ -20,9 +20,8 @@
 
 
 
-      function test() {
-        if test != "" {
-
+      function test() { // {{CONTEXT}}
+        if test != "" { // {{CONTEXT}}
 
 
 
@@ -37,13 +36,14 @@
 
 
 
+          // {{CURSOR}}
         }
       }
     </script>
   </body>
 </html>
 ```
-
+{{TEST}}
 # Title {{CONTEXT}}