From f9773e4a5e0dbeda0d6f8a78d7d61eb802b6d0e1 Mon Sep 17 00:00:00 2001 From: Riley Bruins Date: Fri, 31 May 2024 20:33:56 -0700 Subject: test: fix tests broken after 0.10 update (#6714) --- tests/query/highlights_spec.lua | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) (limited to 'tests/query/highlights_spec.lua') diff --git a/tests/query/highlights_spec.lua b/tests/query/highlights_spec.lua index ff6a7523d..3bb03d49d 100644 --- a/tests/query/highlights_spec.lua +++ b/tests/query/highlights_spec.lua @@ -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) -- cgit v1.2.3-70-g09d2