aboutsummaryrefslogtreecommitdiffstats
path: root/lua
diff options
context:
space:
mode:
Diffstat (limited to 'lua')
-rw-r--r--lua/nvim-treesitter/test_utils.lua15
-rw-r--r--lua/tests/indent/c/basic.c11
-rw-r--r--lua/tests/indent/c/no_braces.c6
-rw-r--r--lua/tests/indent/c_spec.lua18
-rw-r--r--lua/tests/indent/python_spec.lua9
-rw-r--r--lua/tests/indent/rust/basic.rs7
-rw-r--r--lua/tests/indent/rust_spec.lua17
7 files changed, 80 insertions, 3 deletions
diff --git a/lua/nvim-treesitter/test_utils.lua b/lua/nvim-treesitter/test_utils.lua
index e3206b7dd..a62b23070 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)'
diff --git a/lua/tests/indent/c/basic.c b/lua/tests/indent/c/basic.c
new file mode 100644
index 000000000..b20b742d5
--- /dev/null
+++ b/