tests/indent: add basic examples for C and Rust

6 files changed, 67 insertions, 1 deletions

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/lua/tests/indent/c/basic.c
@@ -0,0 +1,11 @@
+int foo(int x)
+{
+    if (x > 10) {
+        return x;
+    } else {
+        while (x > 0) {
+            x--;
+        }
+        return 0;
+    }
+}
diff --git a/lua/tests/indent/c/no_braces.c b/lua/tests/indent/c/no_braces.c
new file mode 100644
index 000000000..4c14ab05f
--- /dev/null
+++ b/lua/tests/indent/c/no_braces.c
@@ -0,0 +1,6 @@
+int foo(int x) {
+    if (x > 10)
+        return 10;
+    else
+        return x;
+}
diff --git a/lua/tests/indent/c_spec.lua b/lua/tests/indent/c_spec.lua
new file mode 100644
index 000000000..5176312b6
--- /dev/null
+++ b/lua/tests/indent/c_spec.lua
@@ -0,0 +1,18 @@
+local whole_file = require('nvim-treesitter.test_utils').indent_whole_file
+
+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
+end)
diff --git a/lua/tests/indent/python_spec.lua b/lua/tests/indent/python_spec.lua
index 9fcdf4637..70017bde1 100644
--- a/lua/tests/indent/python_spec.lua
+++ b/lua/tests/indent/python_spec.lua
@@ -14,6 +14,13 @@ describe('indent python', function()
     'strings.py',
   }
   for _, file in ipairs(files) do
-    it(file, function() whole_file('lua/tests/indent/python/' .. file) end)
+    it(file, function()
+      whole_file('lua/tests/indent/python/' .. file, {
+        tabstop = 4,
+        shiftwidth = 4,
+        softtabstop = 0,
+        expandtab = true,
+      })
+    end)
   end
 end)
diff --git a/lua/tests/indent/rust/basic.rs b/lua/tests/indent/rust/basic.rs
new file mode 100644
index 000000000..54d38cf68
--- /dev/null
+++ b/lua/tests/indent/rust/basic.rs
@@ -0,0 +1,7 @@
+fn foo(x: i32) {
+    if (x > 10) {
+        return 10;
+    } else {
+        return x;
+    }
+}
diff --git a/lua/tests/indent/rust_spec.lua b/lua/tests/indent/rust_spec.lua
new file mode 100644
index 000000000..374e3db6d
--- /dev/null
+++ b/lua/tests/indent/rust_spec.lua
@@ -0,0 +1,17 @@
+local whole_file = require('nvim-treesitter.test_utils').indent_whole_file
+
+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
+end)