1 files changed, 83 insertions, 0 deletions
diff --git a/tests/whitespace_spec.lua b/tests/whitespace_spec.lua
new file mode 100644
index 0000000..0933276
--- /dev/null
+++ b/tests/whitespace_spec.lua
@@ -0,0 +1,83 @@
+local quicker = require("quicker")
+local test_util = require("tests.test_util")
+
+describe("whitespace", function()
+  before_each(function()
+    require("quicker.config").trim_leading_whitespace = "common"
+  end)
+  after_each(function()
+    test_util.reset_editor()
+  end)
+
+  it("removes common leading whitespace from valid results", function()
+    local bufnr = vim.fn.bufadd(test_util.make_tmp_file("whitespace.txt", {
+      "    line 1",
+      "  line 2",
+      "    line 3",
+      "",
+      "      line 4",
+    }))
+    vim.fn.setqflist({
+      {
+        bufnr = bufnr,
+        text = "  line 2",
+        lnum = 2,
+      },
+      {
+        bufnr = bufnr,
+        text = "    line 3",
+        lnum = 3,
+      },
+    })
+    vim.cmd.copen()
+    test_util.assert_snapshot(0, "trim_whitespace")
+    quicker.expand()
+    test_util.assert_snapshot(0, "trim_whitespace_expanded")
+  end)
+
+  it("handles mixed tabs and spaces", function()
+    local bufnr = vim.fn.bufadd(test_util.make_tmp_file("mixed_whitespace.txt", {
+      "  line 1",
+      "\t\tline 2",
+    }))
+    vim.fn.setqflist({
+      {
+        bufnr = bufnr,
+        text = "  line 1",
+        lnum = 1,
+      },
+      {
+        bufnr = bufnr,
+        text = "\t\tline 2",
+        lnum = 2,
+      },
+    })
+    vim.cmd.copen()
+    test_util.assert_snapshot(0, "trim_mixed_whitespace")
+  end)
+
+  it("removes all leading whitespace", function()
+    require("quicker.config").trim_leading_whitespace = "all"
+    local bufnr = vim.fn.bufadd(test_util.make_tmp_file("whitespace_1.txt", {
+      "    line 1",
+      "  line 2",
+      "    line 3",
+      "",
+      "      line 4",
+    }))
+    vim.fn.setqflist({
+      {
+        bufnr = bufnr,
+        text = "  line 2",
+        lnum = 2,
+      },
+      {
+        bufnr = bufnr,
+        text = "    line 3",
+        lnum = 3,
+      },
+    })
+    vim.cmd.copen()
+    test_util.assert_snapshot(0, "trim_all_whitespace")
+  end)
+end)