tests/indent: add basic examples for C and Rust

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)'