1 files changed, 15 insertions, 1 deletions
diff --git a/tests/query/highlights_spec.lua b/tests/query/highlights_spec.lua
index ca573e086..29bfc7c8d 100644
--- a/tests/query/highlights_spec.lua
+++ b/tests/query/highlights_spec.lua
@@ -4,6 +4,10 @@ local utils = require "nvim-treesitter.utils"
 local ts_utils = require "nvim-treesitter.ts_utils"
 local parsers = require "nvim-treesitter.parsers"
 
+local COMMENT_NODES = {
+  markdown = "html_block",
+}
+
 local function check_assertions(file)
   local buf = vim.fn.bufadd(file)
   vim.fn.bufload(file)
@@ -14,15 +18,25 @@ local function check_assertions(file)
     '"highlight-assertions" not executable!'
       .. ' Get it via "cargo install --git https://github.com/theHamsta/highlight-assertions"'
   )
+  local comment_node = COMMENT_NODES[lang] or "comment"
   local assertions = vim.fn.json_decode(
     vim.fn.system(
-      "highlight-assertions -p '" .. utils.get_parser_install_dir() .. "/" .. lang .. ".so'" .. " -s '" .. file .. "'"
+      "highlight-assertions -p '"
+        .. utils.get_parser_install_dir()
+        .. "/"
+        .. lang
+        .. ".so'"
+        .. " -s '"
+        .. file
+        .. "' -c "
+        .. comment_node
     )
   )
   local parser = parsers.get_parser(buf, lang)
 
   local self = highlighter.new(parser, {})
 
+  assert.True(#assertions > 0, "No assertions detected!")
   for _, assertion in ipairs(assertions) do
     local row = assertion.position.row
     local col = assertion.position.column