benbot/nvim-treesitter
1
0
Fork 0
mirror of https://github.com/nvim-treesitter/nvim-treesitter.git synced 2026-07-06 05:20:00 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

test: fix tests broken after 0.10 update (#6714)

Browse source
This commit is contained in:
Riley Bruins 2024-05-31 20:33:56 -07:00 • committed by GitHub
parent b7d50e59b1
commit f9773e4a5e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 20 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

12
tests/query/highlights_spec.lua
View file

@ -29,6 +29,7 @@ local function check_assertions(file)
)
)
local parser = parsers.get_parser(buf, lang)
parser:parse(true)
local self = highlighter.new(parser, {})
@ -39,12 +40,13 @@ local function check_assertions(file)
local captures = {}
local highlights = {}
self.tree:for_each_tree(function(tstree, tree)
if not tstree then
self:prepare_highlight_states(row, row + 1)
self:for_each_highlight_state(function(state)
if not state.tstree then
return
end
local root = tstree:root()
local root = state.tstree:root()
local root_start_row, _, root_end_row, _ = root:range()
-- Only worry about trees within the line range
@ -52,7 +54,7 @@ local function check_assertions(file)
return
end
local query = self:get_query(tree:lang())
local query = state.highlighter_query
-- Some injected languages may not have highlight queries.
if not query:query() then
@ -62,7 +64,7 @@ local function check_assertions(file)
local iter = query:query():iter_captures(root, self.bufnr, row, row + 1)
for capture, node, _ in iter do
local hl = query.hl_cache[capture]
local hl = query:get_hl_from_capture(capture)
assert.is.truthy(hl)
assert.Truthy(node)