test: added scope tests

lua/snacks/init.lua

@@ -26,6 +26,9 @@ M.config = setmetatable({}, {
     config[k] = config[k] or {}
     return config[k]
   end,
+  __newindex = function(_, k, v)
+    config[k] = v
+  end,
 })
 
 --- Get an example config from the docs/examples directory.

tests/scope_spec.lua

@@ -0,0 +1,104 @@
+---@module 'luassert'
+
+local M = {}
+
+---@param lines string|string[]
+---@param opts? {ft?: string, ts?: boolean}
+function M.set_lines(lines, opts)
+  opts = opts or {}
+  lines = type(lines) == "string" and vim.split(lines, "\n") or lines
+  vim.api.nvim_buf_set_lines(0, 0, -1, false, lines --[[ @as string[] ]])
+  vim.bo.filetype = opts.ft or ""
+  if opts.ts then
+    vim.treesitter.stop()
+    assert(not vim.b.ts_highlight, "treesitter highlight is still enabled")
+    vim.treesitter.start()
+    assert(vim.b.ts_highlight, "treesitter highlight is not enabled")
+  end
+end
+
+function M.inspect(v)
+  return vim.inspect(v):gsub("%s+", " ")
+end
+
+local test = [[
+function foo()
+  while true do
+    if x == 1 then
+      break
+    end
+    local y = 2
+  end
+end
+]]
+
+describe("scope", function()
+  local tests = {
+    [1] = { 1, 8 },
+    [2] = { 2, 7 },
+    [3] = { 3, 5 },
+    [4] = { 3, 5 },
+    [5] = { 3, 5 },
+    [6] = { 2, 7 },
+    [7] = { 2, 7 },
+    [8] = { 1, 8 },
+  }
+  Snacks.config.scope = {
+    min_size = 2,
+    treesitter = {
+      blocks = false,
+    },
+  }
+
+  for _, ws in ipairs({ false, true }) do
+    local lines = vim.split(vim.trim(test), "\n")
+
+    local t = vim.deepcopy(tests)
+
+    if ws == true then
+      local c = #lines
+      -- insert empty lines
+      for i = 1, c do
+        table.insert(lines, i * 2, "")
+      end
+      local test2 = {}
+      -- transform tests
+      for line, s in pairs(t) do
+        test2[line * 2 - 1] = { s[1] * 2 - 1, s[2] * 2 - 1 }
+      end
+      t = test2
+    end
+
+    for _, ts in ipairs({ true, false }) do
+      for line, s in pairs(t) do
+        it("should get scope " .. M.inspect({ line = line, scope = s, ts = ts, ws = ws }), function()
+          M.set_lines(lines, {
+            ft = "lua",
+            ts = ts,
+          })
+          local scope = Snacks.scope.get({
+            pos = { line, 0 },
+            treesitter = { enabled = ts },
+          })
+          assert(scope)
+          assert((scope.node == nil) == not ts)
+          assert.same(scope.from, s[1])
+          assert.same(scope.to, s[2])
+        end)
+      end
+    end
+  end
+end)
+
+local function foo()
+  while true do
+    -- doo
+
+    if x == 1 and false then
+      break
+    end
+
+    local y = 2
+    local y = 2
+  end
+end