neovim/test/functional/plugin/lsp/linked_editing_range_spec.lua

local t = require('test.testutil')
local n = require('test.functional.testnvim')()
local t_lsp = require('test.functional.plugin.lsp.testutil')

local eq = t.eq
local exec_lua = n.exec_lua
local insert = n.insert
local feed = n.feed

local clear_notrace = t_lsp.clear_notrace
local create_server_definition = t_lsp.create_server_definition

describe('vim.lsp.linked_editing_range', function()
  before_each(function()
    clear_notrace()

    insert([[
    hello
    hello
    hello]])

    exec_lua(create_server_definition)
    exec_lua(function()
      vim.lsp.linked_editing_range.enable()

      _G.server = _G._create_server({
        capabilities = {
          linkedEditingRangeProvider = true,
        },
        handlers = {
          ['textDocument/linkedEditingRange'] = function(_, _, callback)
            callback(nil, {
              ranges = {
                { start = { line = 0, character = 0 }, ['end'] = { line = 0, character = 5 } },
                { start = { line = 1, character = 0 }, ['end'] = { line = 1, character = 5 } },
                { start = { line = 2, character = 0 }, ['end'] = { line = 2, character = 5 } },
              },
            })
          end,
        },
      })

      _G.server_id = vim.lsp.start({ name = 'dummy', cmd = _G.server.cmd })
    end)
  end)

  it('initiates linked editing', function()
    exec_lua(function()
      local win = vim.api.nvim_get_current_win()
      vim.api.nvim_win_set_cursor(win, { 1, 0 })
    end)
    -- Deletion
    feed('ldw')
    eq(
      {
        'h',
        'h',
        'h',
      },
      exec_lua(function()
        return vim.api.nvim_buf_get_lines(0, 0, -1, false)
      end)
    )
    -- Insertion
    feed('Apt<Esc>')
    eq(
      {
        'hpt',
        'hpt',
        'hpt',
      },
      exec_lua(function()
        return vim.api.nvim_buf_get_lines(0, 0, -1, false)
      end)
    )
    -- Undo/redo
    feed('0xx')
    eq(
      {
        't',
        't',
        't',
      },
      exec_lua(function()
        return vim.api.nvim_buf_get_lines(0, 0, -1, false)
      end)
    )
    feed('u')
    eq(
      {
        'pt',
        'pt',
        'pt',
      },
      exec_lua(function()
        return vim.api.nvim_buf_get_lines(0, 0, -1, false)
      end)
    )
    feed('u')
    eq(
      {
        'hpt',
        'hpt',
        'hpt',
      },
      exec_lua(function()
        return vim.api.nvim_buf_get_lines(0, 0, -1, false)
      end)
    )
    feed('<C-r><C-r>')
    eq(
      {
        't',
        't',
        't',
      },
      exec_lua(function()
        return vim.api.nvim_buf_get_lines(0, 0, -1, false)
      end)
    )
    -- Disabling
    exec_lua(function()
      vim.lsp.linked_editing_range.enable(false, { client_id = _G.server_id })
    end)
    feed('Ipp<Esc>')
    eq(
      {
        'ppt',
        't',
        't',
      },
      exec_lua(function()
        return vim.api.nvim_buf_get_lines(0, 0, -1, false)
      end)
    )
  end)
end)