tests/indent: refactor indent test runner and auto-discover language test files

3 files changed, 45 insertions, 52 deletions

diff --git a/lua/tests/indent/c_spec.lua b/lua/tests/indent/c_spec.lua
index 5176312b6..386131036 100644
--- a/lua/tests/indent/c_spec.lua
+++ b/lua/tests/indent/c_spec.lua
@@ -1,18 +1,18 @@
 local whole_file = require('nvim-treesitter.test_utils').indent_whole_file
+local scan_dir = require('plenary.scandir').scan_dir
 
-describe('indent C', function()
-  local files = {
-    'basic.c',
-    'no_braces.c',
-  }
-  for _, file in ipairs(files) do
-    it(file, function()
-      whole_file('lua/tests/indent/c/' .. file, {
-        tabstop = 4,
-        shiftwidth = 4,
-        softtabstop = 0,
-        expandtab = true,
-      })
-    end)
-  end
+describe('indent C:', function()
+  describe('whole file:', function()
+    local files = scan_dir('lua/tests/indent/c');
+    for _, file in ipairs(files) do
+      it(vim.fn.fnamemodify(file, ':t'), function()
+        whole_file(file, {
+          tabstop = 4,
+          shiftwidth = 4,
+          softtabstop = 0,
+          expandtab = true,
+        })
+      end)
+    end
+  end)
 end)
diff --git a/lua/tests/indent/python_spec.lua b/lua/tests/indent/python_spec.lua
index 70017bde1..1bdb4aa48 100644
--- a/lua/tests/indent/python_spec.lua
+++ b/lua/tests/indent/python_spec.lua
@@ -1,26 +1,18 @@
 local whole_file = require('nvim-treesitter.test_utils').indent_whole_file
+local scan_dir = require('plenary.scandir').scan_dir
 
-describe('indent python', function()
-  local files = {
-    'aligned_indent.py',
-    'basic_blocks.py',
-    'basic_collections.py',
-    'branches.py',
-    'comprehensions.py',
-    'control_flow.py',
-    'hanging_indent.py',
-    'join_lines.py',
-    'nested_collections.py',
-    'strings.py',
-  }
-  for _, file in ipairs(files) do
-    it(file, function()
-      whole_file('lua/tests/indent/python/' .. file, {
-        tabstop = 4,
-        shiftwidth = 4,
-        softtabstop = 0,
-        expandtab = true,
-      })
-    end)
-  end
+describe('indent Python:', function()
+  describe('whole file:', function()
+    local files = scan_dir('lua/tests/indent/python');
+    for _, file in ipairs(files) do
+      it(vim.fn.fnamemodify(file, ':t'), function()
+        whole_file(file, {
+          tabstop = 4,
+          shiftwidth = 4,
+          softtabstop = 0,
+          expandtab = true,
+        })
+      end)
+    end
+  end)
 end)
diff --git a/lua/tests/indent/rust_spec.lua b/lua/tests/indent/rust_spec.lua
index 374e3db6d..ad9e948c4 100644
--- a/lua/tests/indent/rust_spec.lua
+++ b/lua/tests/indent/rust_spec.lua
@@ -1,17 +1,18 @@
 local whole_file = require('nvim-treesitter.test_utils').indent_whole_file
+local scan_dir = require('plenary.scandir').scan_dir
 
-describe('indent rust', function()
-  local files = {
-    'basic.rs',
-  }
-  for _, file in ipairs(files) do
-    it(file, function()
-      whole_file('lua/tests/indent/rust/' .. file, {
-        tabstop = 4,
-        shiftwidth = 4,
-        softtabstop = 0,
-        expandtab = true,
-      })
-    end)
-  end
+describe('indent Rust:', function()
+  describe('whole file:', function()
+    local files = scan_dir('lua/tests/indent/rust');
+    for _, file in ipairs(files) do
+      it(vim.fn.fnamemodify(file, ':t'), function()
+        whole_file(file, {
+          tabstop = 4,
+          shiftwidth = 4,
+          softtabstop = 0,
+          expandtab = true,
+        })
+      end)
+    end
+  end)
 end)