diff options
Diffstat (limited to 'lua/nvim-treesitter/test_utils.lua')
| -rw-r--r-- | lua/nvim-treesitter/test_utils.lua | 15 |
1 files changed, 13 insertions, 2 deletions
diff --git a/lua/nvim-treesitter/test_utils.lua b/lua/nvim-treesitter/test_utils.lua index e3206b7d..a62b2307 100644 --- a/lua/nvim-treesitter/test_utils.lua +++ b/lua/nvim-treesitter/test_utils.lua @@ -1,6 +1,15 @@ local M = {} -function M.indent_whole_file(file) +function M.set_buf_indent_opts(opts) + local optnames = {'tabstop', 'shiftwidth', 'softtabstop', 'expandtab'} + for _, opt in ipairs(optnames) do + if opts[opt] ~= nil then + vim.bo[opt] = opts[opt] + end + end +end + +function M.indent_whole_file(file, opts) assert.are.same(1, vim.fn.filereadable(file)) -- load reference file @@ -11,8 +20,10 @@ function M.indent_whole_file(file) vim.bo.indentexpr = 'nvim_treesitter#indent()' assert.are.same('nvim_treesitter#indent()', vim.bo.indentexpr) + M.set_buf_indent_opts(opts) + -- indent the whole file - vim.cmd 'normal gg=G' + vim.cmd 'silent normal gg=G' local indented = vim.api.nvim_buf_get_lines(0, 0, -1, true) -- clear any changes to avoid 'No write since last change (add ! to override)' |