From 136e5602af29c881c4ebdc66d300e4801df94044 Mon Sep 17 00:00:00 2001 From: Julien Vincent Date: Sun, 13 Oct 2024 12:18:04 +0100 Subject: [PATCH] Add fennel treesitter queries and tests --- lua/nvim-paredit/defaults.lua | 5 +- queries/fennel/paredit/forms.scm | 3 + tests/config.lua | 2 +- .../fennel/element_raise_spec.lua | 71 ++++++++++++ tests/nvim-paredit/fennel/form_raise_spec.lua | 67 +++++++++++ tests/nvim-paredit/fennel/motion_spec.lua | 49 ++++++++ tests/nvim-paredit/fennel/slurp_spec.lua | 107 ++++++++++++++++++ 7 files changed, 302 insertions(+), 2 deletions(-) create mode 100644 queries/fennel/paredit/forms.scm create mode 100644 tests/nvim-paredit/fennel/element_raise_spec.lua create mode 100644 tests/nvim-paredit/fennel/form_raise_spec.lua create mode 100644 tests/nvim-paredit/fennel/motion_spec.lua create mode 100644 tests/nvim-paredit/fennel/slurp_spec.lua diff --git a/lua/nvim-paredit/defaults.lua b/lua/nvim-paredit/defaults.lua index 5824c7c..8db835e 100644 --- a/lua/nvim-paredit/defaults.lua +++ b/lua/nvim-paredit/defaults.lua @@ -119,11 +119,14 @@ M.defaults = { enabled = false, indentor = require("nvim-paredit.indentation.native").indentor, }, - filetypes = { "clojure" }, + filetypes = { "clojure", "fennel" }, languages = { clojure = { whitespace_chars = { " ", "," }, }, + fennel = { + whitespace_chars = { " ", "," }, + }, }, keys = {}, } diff --git a/queries/fennel/paredit/forms.scm b/queries/fennel/paredit/forms.scm new file mode 100644 index 0000000..252f69e --- /dev/null +++ b/queries/fennel/paredit/forms.scm @@ -0,0 +1,3 @@ +(list) @form +(sequence) @form +(table) @form diff --git a/tests/config.lua b/tests/config.lua index bc416c5..a4c60d3 100644 --- a/tests/config.lua +++ b/tests/config.lua @@ -11,6 +11,6 @@ vim.bo.swapfile = false require("nvim-treesitter.configs").setup({ parser_install_dir = vim.fn.getcwd() .. "/.build/parsers", - ensure_installed = { "clojure" }, + ensure_installed = { "clojure", "fennel" }, sync_install = true, }) diff --git a/tests/nvim-paredit/fennel/element_raise_spec.lua b/tests/nvim-paredit/fennel/element_raise_spec.lua new file mode 100644 index 0000000..97b2620 --- /dev/null +++ b/tests/nvim-paredit/fennel/element_raise_spec.lua @@ -0,0 +1,71 @@ +local paredit = require("nvim-paredit.api") + +local prepare_buffer = require("tests.nvim-paredit.utils").prepare_buffer +local expect = require("tests.nvim-paredit.utils").expect + +describe("element raising", function() + vim.api.nvim_set_option_value("filetype", "fennel", { + buf = 0, + }) + + it("should raise the element", function() + prepare_buffer({ + content = "(a (b))", + cursor = { 1, 4 }, + }) + paredit.raise_element() + expect({ + content = "(a b)", + cursor = { 1, 3 }, + }) + end) + + it("should raise form elements when cursor is placed on edge", function() + prepare_buffer({ + content = "(a (b))", + cursor = { 1, 3 }, + }) + + paredit.raise_element() + expect({ + content = "(b)", + cursor = { 1, 0 }, + }) + + prepare_buffer({ + content = "(a #(b))", + cursor = { 1, 3 }, + }) + + paredit.raise_element() + expect({ + content = "#(b)", + cursor = { 1, 0 }, + }) + end) + + it("should raise a multi-line element", function() + prepare_buffer({ + content = { "(a (b", " c))" }, + cursor = { 1, 3 }, + }) + + paredit.raise_element() + expect({ + content = { "(b", " c)" }, + cursor = { 1, 0 }, + }) + end) + + it("should do nothing if it is a direct child of the document root", function() + prepare_buffer({ + content = { "a", "b" }, + cursor = { 1, 0 }, + }) + paredit.raise_form() + expect({ + content = { "a", "b" }, + cursor = { 1, 0 }, + }) + end) +end) diff --git a/tests/nvim-paredit/fennel/form_raise_spec.lua b/tests/nvim-paredit/fennel/form_raise_spec.lua new file mode 100644 index 0000000..e7f42fe --- /dev/null +++ b/tests/nvim-paredit/fennel/form_raise_spec.lua @@ -0,0 +1,67 @@ +local paredit = require("nvim-paredit.api") + +local prepare_buffer = require("tests.nvim-paredit.utils").prepare_buffer +local expect_all = require("tests.nvim-paredit.utils").expect_all +local expect = require("tests.nvim-paredit.utils").expect + +describe("form raising", function() + vim.api.nvim_set_option_value("filetype", "fennel", { + buf = 0, + }) + + it("should raise the form", function() + expect_all(paredit.raise_form, { + { + "list", + before_content = "(a (b c))", + before_cursor = { 1, 6 }, + after_content = "(b c)", + after_cursor = { 1, 0 }, + }, + { + "list with reader", + before_content = "(a #(b c))", + before_cursor = { 1, 5 }, + after_content = "#(b c)", + after_cursor = { 1, 0 }, + }, + }) + end) + + it("should raise a multi-line form", function() + prepare_buffer({ + content = { "(a (b ", "c))" }, + cursor = { 1, 4 }, + }) + + paredit.raise_form() + expect({ + content = { "(b ", "c)" }, + cursor = { 1, 0 }, + }) + end) + + it("should do nothing if it is a direct child of the document root", function() + prepare_buffer({ + content = { "(a)", "b" }, + cursor = { 1, 1 }, + }) + paredit.raise_form() + expect({ + content = { "(a)", "b" }, + cursor = { 1, 1 }, + }) + end) + + it("should do nothing if it is outside of a form", function() + prepare_buffer({ + content = { "a", "b" }, + cursor = { 1, 0 }, + }) + paredit.raise_form() + expect({ + content = { "a", "b" }, + cursor = { 1, 0 }, + }) + end) +end) diff --git a/tests/nvim-paredit/fennel/motion_spec.lua b/tests/nvim-paredit/fennel/motion_spec.lua new file mode 100644 index 0000000..6dda149 --- /dev/null +++ b/tests/nvim-paredit/fennel/motion_spec.lua @@ -0,0 +1,49 @@ +local paredit = require("nvim-paredit.api") + +local prepare_buffer = require("tests.nvim-paredit.utils").prepare_buffer +local expect = require("tests.nvim-paredit.utils").expect + +describe("motions", function() + vim.api.nvim_set_option_value("filetype", "fennel", { + buf = 0, + }) + + it("should jump to next element in form (tail)", function() + prepare_buffer({ + content = "(aa (bb) #(cc))", + cursor = { 1, 2 }, + }) + + paredit.move_to_next_element_tail() + expect({ + cursor = { 1, 7 }, + }) + + paredit.move_to_next_element_tail() + expect({ + cursor = { 1, 13 }, + }) + + paredit.move_to_next_element_tail() + expect({ + cursor = { 1, 13 }, + }) + end) + + it("should jump to next element in form (head)", function() + prepare_buffer({ + content = "(aa (bb) #(cc))", + cursor = { 1, 2 }, + }) + + paredit.move_to_next_element_head() + expect({ + cursor = { 1, 4 }, + }) + + paredit.move_to_next_element_head() + expect({ + cursor = { 1, 9 }, + }) + end) +end) diff --git a/tests/nvim-paredit/fennel/slurp_spec.lua b/tests/nvim-paredit/fennel/slurp_spec.lua new file mode 100644 index 0000000..e9f1265 --- /dev/null +++ b/tests/nvim-paredit/fennel/slurp_spec.lua @@ -0,0 +1,107 @@ +local paredit = require("nvim-paredit.api") + +local prepare_buffer = require("tests.nvim-paredit.utils").prepare_buffer +local expect_all = require("tests.nvim-paredit.utils").expect_all +local expect = require("tests.nvim-paredit.utils").expect + +describe("slurping forward", function() + vim.api.nvim_set_option_value("filetype", "fennel", { + buf = 0, + }) + local parser = vim.treesitter.get_parser(0) + if not parser then + return error("Failed to get parser for fennel") + end + + it("should slurp forward different form types", function() + expect_all(paredit.slurp_forwards, { + { + "list", + before_content = "() a", + before_cursor = { 1, 1 }, + after_content = "( a)", + after_cursor = { 1, 1 }, + }, + { + "vector", + before_content = "[] a", + before_cursor = { 1, 1 }, + after_content = "[ a]", + after_cursor = { 1, 1 }, + }, + { + "quoted list", + before_content = "`() a", + before_cursor = { 1, 2 }, + after_content = "`( a)", + after_cursor = { 1, 2 }, + }, + }) + end) + + it("should skip comments", function() + prepare_buffer({ + content = { "()", ";; comment", "a" }, + cursor = { 1, 1 }, + }) + paredit.slurp_forwards() + expect({ + content = { "(", ";; comment", "a)" }, + cursor = { 1, 0 }, + }) + end) + + it("should recursively slurp the next sibling", function() + prepare_buffer({ + content = "(()) 1 2", + cursor = { 1, 2 }, + }) + + paredit.slurp_forwards() + expect({ + content = "(() 1) 2", + cursor = { 1, 2 }, + }) + + parser:parse() + + paredit.slurp_forwards() + expect({ + content = "(( 1)) 2", + cursor = { 1, 2 }, + }) + + parser:parse() + + prepare_buffer({ + content = "(( 1)) 2", + cursor = { 1, 2 }, + }) + + paredit.slurp_forwards() + expect({ + content = "(( 1) 2)", + cursor = { 1, 2 }, + }) + + parser:parse() + + paredit.slurp_forwards() + expect({ + content = "(( 1 2))", + cursor = { 1, 2 }, + }) + end) + + it("should follow the bracket with cursor", function() + prepare_buffer({ + content = "() 1 2", + cursor = { 1, 1 }, + }) + paredit.slurp_forwards({ cursor_behaviour = "follow" }) + expect({ + content = "( 1) 2", + cursor = { 1, 3 }, + }) + end) +end)