wyj/neovim
1
0
Fork 0
mirror of https://github.com/neovim/neovim.git synced 2024-09-17 20:58:20 -04:00
Code Issues Packages Projects Releases Wiki Activity

test(treesitter): add a simple testutil file

Browse Source 
The util file, for now, just abstracts the common `run_query` function.
This commit is contained in:
Riley Bruins 2024-08-30 19:27:18 -07:00
parent b17481c77e
commit 8e55f7ca68
2 changed files with 36 additions and 48 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

59
test/functional/treesitter/parser_spec.lua
View File

@ -1,5 +1,6 @@
local t = require('test.testutil')
local n = require('test.functional.testnvim')()
local ts_t = require('test.functional.treesitter.testutil')
local clear = n.clear
local dedent = t.dedent
@ -8,6 +9,7 @@ local insert = n.insert
local exec_lua = n.exec_lua
local pcall_err = t.pcall_err
local feed = n.feed
local run_query = ts_t.run_query
describe('treesitter parser API', function()
before_each(function()
@ -686,26 +688,13 @@ print()
vim.treesitter.start(0, 'lua')
end)
local function run_query()
return exec_lua(function(query_str)
local query = vim.treesitter.query.parse('lua', query_str)
local parser = vim.treesitter.get_parser()
local tree = parser:parse()[1]
local res = {}
for id, _, metadata in query:iter_captures(tree:root(), 0) do
table.insert(res, { query.captures[id], metadata[id].range })
end
return res
end, query_text)
end
eq({
{ 'str', { 2, 12, 6, 10 } },
{ 'str', { 11, 10, 11, 10 } },
{ 'str', { 17, 10, 17, 10 } },
{ 'str', { 19, 10, 19, 10 } },
{ 'str', { 22, 15, 22, 25 } },
}, run_query())
}, run_query('lua', query_text))
end)
it('trims only empty lines by default (backwards compatible)', function()
@ -728,23 +717,10 @@ print()
vim.treesitter.start(0, 'markdown')
end)
local function run_query()
return exec_lua(function(query_str)
local query = vim.treesitter.query.parse('markdown', query_str)
local parser = vim.treesitter.get_parser()
local tree = parser:parse()[1]
local res = {}
for id, _, metadata in query:iter_captures(tree:root(), 0) do
table.insert(res, { query.captures[id], metadata[id].range })
end
return res
end, query_text)
end
eq({
{ 'fold', { 0, 0, 3, 0 } },
{ 'fold', { 4, 0, 7, 0 } },
}, run_query())
}, run_query('markdown', query_text))
end)
end)
@ -763,32 +739,19 @@ print()
vim.treesitter.start(0, 'c')
end)
local function run_query()
return exec_lua(function(query_str)
local query = vim.treesitter.query.parse('c', query_str)
local parser = vim.treesitter.get_parser()
local tree = parser:parse()[1]
local res = {}
for id, node in query:iter_captures(tree:root()) do
table.insert(res, { query.captures[id], node:range() })
end
return res
end, query0)
end
eq({
{ 'function', 0, 0, 2, 1 },
{ 'declaration', 1, 2, 1, 12 },
}, run_query())
{ 'function', { 0, 0, 2, 1 } },
{ 'declaration', { 1, 2, 1, 12 } },
}, run_query('c', query0))
n.command 'normal ggO'
insert('int a;')
eq({
{ 'declaration', 0, 0, 0, 6 },
{ 'function', 1, 0, 3, 1 },
{ 'declaration', 2, 2, 2, 12 },
}, run_query())
{ 'declaration', { 0, 0, 0, 6 } },
{ 'function', { 1, 0, 3, 1 } },
{ 'declaration', { 2, 2, 2, 12 } },
}, run_query('c', query0))
end)
it('handles ranges when source is a multiline string (#20419)', function()

25
test/functional/treesitter/testutil.lua Normal file
View File

@ -0,0 +1,25 @@
local n = require('test.functional.testnvim')()
local exec_lua = n.exec_lua
local M = {}
---@param language string
---@param query_string string
function M.run_query(language, query_string)
return exec_lua(function(lang, query_str)
local query = vim.treesitter.query.parse(lang, query_str)
local parser = vim.treesitter.get_parser()
local tree = parser:parse()[1]
local res = {}
for id, node, metadata in query:iter_captures(tree:root(), 0) do
table.insert(
res,
{ query.captures[id], metadata[id] and metadata[id].range or { node:range() } }
)
end
return res
end, language, query_string)
end
return M