local helpers = require('test.functional.helpers')(after_each) local Screen = require('test.functional.ui.screen') local clear = helpers.clear local eq = helpers.eq local ok = helpers.ok local describe_lua_and_rpc = helpers.describe_lua_and_rpc(describe) local api = helpers.api local fn = helpers.fn local request = helpers.request local exc_exec = helpers.exc_exec local exec_lua = helpers.exec_lua local feed_command = helpers.feed_command local insert = helpers.insert local NIL = vim.NIL local command = helpers.command local feed = helpers.feed local pcall_err = helpers.pcall_err local assert_alive = helpers.assert_alive describe('api/buf', function() before_each(clear) -- access deprecated functions local function curbuf_depr(method, ...) return request('buffer_' .. method, 0, ...) end describe('nvim_buf_set_lines, nvim_buf_line_count', function() it('deprecated forms', function() eq(1, curbuf_depr('line_count')) curbuf_depr('insert', -1, { 'line' }) eq(2, curbuf_depr('line_count')) curbuf_depr('insert', -1, { 'line' }) eq(3, curbuf_depr('line_count')) curbuf_depr('del_line', -1) eq(2, curbuf_depr('line_count')) curbuf_depr('del_line', -1) curbuf_depr('del_line', -1) -- There's always at least one line eq(1, curbuf_depr('line_count')) end) it("doesn't crash just after set undolevels=1 #24894", function() local buf = api.nvim_create_buf(false, true) api.nvim_buf_set_option(buf, 'undolevels', -1) api.nvim_buf_set_lines(buf, 0, 1, false, {}) assert_alive() end) it('cursor position is maintained after lines are inserted #9961', function() -- replace the buffer contents with these three lines. api.nvim_buf_set_lines(0, 0, -1, true, { 'line1', 'line2', 'line3', 'line4' }) -- Set the current cursor to {3, 2}. api.nvim_win_set_cursor(0, { 3, 2 }) -- add 2 lines and delete 1 line above the current cursor position. api.nvim_buf_set_lines(0, 1, 2, true, { 'line5', 'line6' }) -- check the current set of lines in the buffer. eq({ 'line1', 'line5', 'line6', 'line3', 'line4' }, api.nvim_buf_get_lines(0, 0, -1, true)) -- cursor should be moved below by 1 line. eq({ 4, 2 }, api.nvim_win_get_cursor(0)) -- add a line after the current cursor position. api.nvim_buf_set_lines(0, 5, 5, true, { 'line7' }) -- check the current set of lines in the buffer. eq( { 'line1', 'line5', 'line6', 'line3', 'line4', 'line7' }, api.nvim_buf_get_lines(0, 0, -1, true) ) -- cursor position is unchanged. eq({ 4, 2 }, api.nvim_win_get_cursor(0)) -- overwrite current cursor line. api.nvim_buf_set_lines(0, 3, 5, true, { 'line8', 'line9' }) -- check the current set of lines in the buffer. eq( { 'line1', 'line5', 'line6', 'line8', 'line9', 'line7' }, api.nvim_buf_get_lines(0, 0, -1, true) ) -- cursor position is unchanged. eq({ 4, 2 }, api.nvim_win_get_cursor(0)) -- delete current cursor line. api.nvim_buf_set_lines(0, 3, 5, true, {}) -- check the current set of lines in the buffer. eq({ 'line1', 'line5', 'line6', 'line7' }, api.nvim_buf_get_lines(0, 0, -1, true)) -- cursor position is unchanged. eq({ 4, 2 }, api.nvim_win_get_cursor(0)) end) it('cursor position is maintained in non-current window', function() api.nvim_buf_set_lines(0, 0, -1, true, { 'line1', 'line2', 'line3', 'line4' }) api.nvim_win_set_cursor(0, { 3, 2 }) local win = api.nvim_get_current_win() local buf = api.nvim_get_current_buf() command('new') api.nvim_buf_set_lines(buf, 1, 2, true, { 'line5', 'line6' }) eq({ 'line1', 'line5', 'line6', 'line3', 'line4' }, api.nvim_buf_get_lines(buf, 0, -1, true)) eq({ 4, 2 }, api.nvim_win_get_cursor(win)) end) it('cursor position is maintained in TWO non-current windows', function() api.nvim_buf_set_lines(0, 0, -1, true, { 'line1', 'line2', 'line3', 'line4' }) api.nvim_win_set_cursor(0, { 3, 2 }) local win = api.nvim_get_current_win() local buf = api.nvim_get_current_buf() command('split') api.nvim_win_set_cursor(0, { 4, 2 }) local win2 = api.nvim_get_current_win() -- set current window to third one with another buffer command('new') api.nvim_buf_set_lines(buf, 1, 2, true, { 'line5', 'line6' }) eq({ 'line1', 'line5', 'line6', 'line3', 'line4' }, api.nvim_buf_get_lines(buf, 0, -1, true)) eq({ 4, 2 }, api.nvim_win_get_cursor(win)) eq({ 5, 2 }, api.nvim_win_get_cursor(win2)) end) it('line_count has defined behaviour for unloaded buffers', function() -- we'll need to know our bufnr for when it gets unloaded local bufnr = api.nvim_buf_get_number(0) -- replace the buffer contents with these three lines api.nvim_buf_set_lines(bufnr, 0, -1, true, { 'line1', 'line2', 'line3', 'line4' }) -- check the line count is correct eq(4, api.nvim_buf_line_count(bufnr)) -- force unload the buffer (this will discard changes) command('new') command('bunload! ' .. bufnr) -- line count for an unloaded buffer should always be 0 eq(0, api.nvim_buf_line_count(bufnr)) end) it('get_lines has defined behaviour for unloaded buffers', function() -- we'll need to know our bufnr for when it gets unloaded local bufnr = api.nvim_buf_get_number(0) -- replace the buffer contents with these three lines api.nvim_buf_set_lines(bufnr, 0, -1, true, { 'line1', 'line2', 'line3', 'line4' }) -- confirm that getting lines works eq({ 'line2', 'line3' }, api.nvim_buf_get_lines(bufnr, 1, 3, true)) -- force unload the buffer (this will discard changes) command('new') command('bunload! ' .. bufnr) -- attempting to get lines now always gives empty list eq({}, api.nvim_buf_get_lines(bufnr, 1, 3, true)) -- it's impossible to get out-of-bounds errors for an unloaded buffer eq({}, api.nvim_buf_get_lines(bufnr, 8888, 9999, true)) end) describe('handles topline', function() local screen before_each(function() screen = Screen.new(20, 12) screen:set_default_attr_ids { [1] = { bold = true, foreground = Screen.colors.Blue1 }, [2] = { reverse = true, bold = true }, [3] = { reverse = true }, } screen:attach() api.nvim_buf_set_lines( 0, 0, -1, true, { 'aaa', 'bbb', 'ccc', 'ddd', 'www', 'xxx', 'yyy', 'zzz' } ) api.nvim_set_option_value('modified', false, {}) end) it('of current window', function() local win = api.nvim_get_current_win() local buf = api.nvim_get_current_buf() command('new | wincmd w') api.nvim_win_set_cursor(win, { 8, 0 }) screen:expect { grid = [[ | {1:~ }|*4 {3:[No Name] }| www | xxx | yyy | ^zzz | {2:[No Name] }| | ]], } api.nvim_buf_set_lines(buf, 0, 2, true, { 'aaabbb' }) screen:expect { grid = [[ | {1:~ }|*4 {3:[No Name] }| www | xxx | yyy | ^zzz | {2:[No Name] [+] }| | ]], } -- replacing topline keeps it the topline api.nvim_buf_set_lines(buf, 3, 4, true, { 'wwweeee' }) screen:expect { grid = [[ | {1:~ }|*4 {3:[No Name] }| wwweeee | xxx | yyy | ^zzz | {2:[No Name] [+] }| | ]], } -- inserting just before topline does not scroll up if cursor would be moved api.nvim_buf_set_lines(buf, 3, 3, true, { 'mmm' }) screen:expect { grid = [[ | {1:~ }|*4 {3:[No Name] }| wwweeee | xxx | yyy | ^zzz | {2:[No Name] [+] }| | ]], unchanged = true, } api.nvim_win_set_cursor(0, { 7, 0 }) screen:expect { grid = [[ | {1:~ }|*4 {3:[No Name] }| wwweeee | xxx | ^yyy | zzz | {2:[No Name] [+] }| | ]], } api.nvim_buf_set_lines(buf, 4, 4, true, { 'mmmeeeee' }) screen:expect { grid = [[ | {1:~ }|*4 {3:[No Name] }| mmmeeeee | wwweeee | xxx | ^yyy | {2:[No Name] [+] }| | ]], } end) it('of non-current window', function() local win = api.nvim_get_current_win() local buf = api.nvim_get_current_buf() command('new') api.nvim_win_set_cursor(win, { 8, 0 }) screen:expect { grid = [[ ^ | {1:~ }|*4 {2:[No Name] }| www | xxx | yyy | zzz | {3:[No Name] }| | ]], } api.nvim_buf_set_lines(buf, 0, 2, true, { 'aaabbb' }) screen:expect { grid = [[ ^ | {1:~ }|*4 {2:[No Name] }| www | xxx | yyy | zzz | {3:[No Name] [+] }| | ]], } -- replacing topline keeps it the topline api.nvim_buf_set_lines(buf, 3, 4, true, { 'wwweeee' }) screen:expect { grid = [[ ^ | {1:~ }|*4 {2:[No Name] }| wwweeee | xxx | yyy | zzz | {3:[No Name] [+] }| | ]], } -- inserting just before topline scrolls up api.nvim_buf_set_lines(buf, 3, 3, true, { 'mmm' }) screen:expect { grid = [[ ^ | {1:~ }|*4 {2:[No Name] }| mmm | wwweeee | xxx | yyy | {3:[No Name] [+] }| | ]], } end) it('of split windows with same buffer', function() local win = api.nvim_get_current_win() local buf = api.nvim_get_current_buf() command('split') api.nvim_win_set_cursor(win, { 8, 0 }) api.nvim_win_set_cursor(0, { 1, 0 }) screen:expect { grid = [[ ^aaa | bbb | ccc | ddd | www | {2:[No Name] }| www | xxx | yyy | zzz | {3:[No Name] }| | ]], } api.nvim_buf_set_lines(buf, 0, 2, true, { 'aaabbb' }) screen:expect { grid = [[ ^aaabbb | ccc | ddd | www | xxx | {2:[No Name] [+] }| www | xxx | yyy | zzz | {3:[No Name] [+] }| | ]], } -- replacing topline keeps it the topline api.nvim_buf_set_lines(buf, 3, 4, true, { 'wwweeee' }) screen:expect { grid = [[ ^aaabbb | ccc | ddd | wwweeee | xxx | {2:[No Name] [+] }| wwweeee | xxx | yyy | zzz | {3:[No Name] [+] }| | ]], } -- inserting just before topline scrolls up api.nvim_buf_set_lines(buf, 3, 3, true, { 'mmm' }) screen:expect { grid = [[ ^aaabbb | ccc | ddd | mmm | wwweeee | {2:[No Name] [+] }| mmm | wwweeee | xxx | yyy | {3:[No Name] [+] }| | ]], } end) end) it('handles clearing out non-current buffer #24911', function() local buf = api.nvim_get_current_buf() api.nvim_buf_set_lines(buf, 0, -1, true, { 'aaa', 'bbb', 'ccc' }) command('new') api.nvim_buf_set_lines(0, 0, -1, true, { 'xxx', 'yyy', 'zzz' }) api.nvim_buf_set_lines(buf, 0, -1, true, {}) eq({ 'xxx', 'yyy', 'zzz' }, api.nvim_buf_get_lines(0, 0, -1, true)) eq({ '' }, api.nvim_buf_get_lines(buf, 0, -1, true)) end) end) describe('deprecated: {get,set,del}_line', function() it('works', function() eq('', curbuf_depr('get_line', 0)) curbuf_depr('set_line', 0, 'line1') eq('line1', curbuf_depr('get_line', 0)) curbuf_depr('set_line', 0, 'line2') eq('line2', curbuf_depr('get_line', 0)) curbuf_depr('del_line', 0) eq('', curbuf_depr('get_line', 0)) end) it('get_line: out-of-bounds is an error', function() curbuf_depr('set_line', 0, 'line1.a') eq(1, curbuf_depr('line_count')) -- sanity eq(false, pcall(curbuf_depr, 'get_line', 1)) eq(false, pcall(curbuf_depr, 'get_line', -2)) end) it('set_line, del_line: out-of-bounds is an error', function() curbuf_depr('set_line', 0, 'line1.a') eq(false, pcall(curbuf_depr, 'set_line', 1, 'line1.b')) eq(false, pcall(curbuf_depr, 'set_line', -2, 'line1.b')) eq(false, pcall(curbuf_depr, 'del_line', 2)) eq(false, pcall(curbuf_depr, 'del_line', -3)) end) it('can handle NULs', function() curbuf_depr('set_line', 0, 'ab\0cd') eq('ab\0cd', curbuf_depr('get_line', 0)) end) end) describe('deprecated: {get,set}_line_slice', function() it('get_line_slice: out-of-bounds returns empty array', function() curbuf_depr('set_line_slice', 0, 0, true, true, { 'a', 'b', 'c' }) eq({ 'a', 'b', 'c' }, curbuf_depr('get_line_slice', 0, 2, true, true)) --sanity eq({}, curbuf_depr('get_line_slice', 2, 3, false, true)) eq({}, curbuf_depr('get_line_slice', 3, 9, true, true)) eq({}, curbuf_depr('get_line_slice', 3, -1, true, true)) eq({}, curbuf_depr('get_line_slice', -3, -4, false, true)) eq({}, curbuf_depr('get_line_slice', -4, -5, true, true)) end) it('set_line_slice: out-of-bounds extends past end', function() curbuf_depr('set_line_slice', 0, 0, true, true, { 'a', 'b', 'c' }) eq({ 'a', 'b', 'c' }, curbuf_depr('get_line_slice', 0, 2, true, true)) --sanity eq({ 'c' }, curbuf_depr('get_line_slice', -1, 4, true, true)) eq({ 'a', 'b', 'c' }, curbuf_depr('get_line_slice', 0, 5, true, true)) curbuf_depr('set_line_slice', 4, 5, true, true, { 'd' }) eq({ 'a', 'b', 'c', 'd' }, curbuf_depr('get_line_slice', 0, 5, true, true)) curbuf_depr('set_line_slice', -4, -5, true, true, { 'e' }) eq({ 'e', 'a', 'b', 'c', 'd' }, curbuf_depr('get_line_slice', 0, 5, true, true)) end) it('works', function() eq({ '' }, curbuf_depr('get_line_slice', 0, -1, true, true)) -- Replace buffer curbuf_depr('set_line_slice', 0, -1, true, true, { 'a', 'b', 'c' }) eq({ 'a', 'b', 'c' }, curbuf_depr('get_line_slice', 0, -1, true, true)) eq({ 'b', 'c' }, curbuf_depr('get_line_slice', 1, -1, true, true)) eq({ 'b' }, curbuf_depr('get_line_slice', 1, 2, true, false)) eq({}, curbuf_depr('get_line_slice', 1, 1, true, false)) eq({ 'a', 'b' }, curbuf_depr('get_line_slice', 0, -1, true, false)) eq({ 'b' }, curbuf_depr('get_line_slice', 1, -1, true, false)) eq({ 'b', 'c' }, curbuf_depr('get_line_slice', -2, -1, true, true)) curbuf_depr('set_line_slice', 1, 2, true, false, { 'a', 'b', 'c' }) eq({ 'a', 'a', 'b', 'c', 'c' }, curbuf_depr('get_line_slice', 0, -1, true, true)) curbuf_depr('set_line_slice', -1, -1, true, true, { 'a', 'b', 'c' }) eq({ 'a', 'a', 'b', 'c', 'a', 'b', 'c' }, curbuf_depr('get_line_slice', 0, -1, true, true)) curbuf_depr('set_line_slice', 0, -3, true, false, {}) eq({ 'a', 'b', 'c' }, curbuf_depr('get_line_slice', 0, -1, true, true)) curbuf_depr('set_line_slice', 0, -1, true, true, {}) eq({ '' }, curbuf_depr('get_line_slice', 0, -1, true, true)) end) end) describe_lua_and_rpc('nvim_buf_get_lines, nvim_buf_set_lines', function(lua_or_rpc) local function get_lines(...) return lua_or_rpc.nvim_buf_get_lines(0, ...) end local function set_lines(...) return lua_or_rpc.nvim_buf_set_lines(0, ...) end local function line_count() return lua_or_rpc.nvim_buf_line_count(0) end it('fails correctly when input is not valid', function() eq(1, lua_or_rpc.nvim_buf_get_number(0)) eq( [['replacement string' item contains newlines]], pcall_err(lua_or_rpc.nvim_buf_set_lines, 1, 1, 2, false, { 'b\na' }) ) end) it("fails if 'nomodifiable'", function() command('set nomodifiable') eq( [[Buffer is not 'modifiable']], pcall_err(lua_or_rpc.nvim_buf_set_lines, 1, 1, 2, false, { 'a', 'b' }) ) end) it('has correct line_count when inserting and deleting', function() eq(1, line_count()) set_lines(-1, -1, true, { 'line' }) eq(2, line_count()) set_lines(-1, -1, true, { 'line' }) eq(3, line_count()) set_lines(-2, -1, true, {}) eq(2, line_count()) set_lines(-2, -1, true, {}) set_lines(-2, -1, true, {}) -- There's always at least one line eq(1, line_count()) end) it('can get, set and delete a single line', function() eq({ '' }, get_lines(0, 1, true)) set_lines(0, 1, true, { 'line1' }) eq({ 'line1' }, get_lines(0, 1, true)) set_lines(0, 1, true, { 'line2' }) eq({ 'line2' }, get_lines(0, 1, true)) set_lines(0, 1, true, {}) eq({ '' }, get_lines(0, 1, true)) end) it('can get a single line with strict indexing', function() set_lines(0, 1, true, { 'line1.a' }) eq(1, line_count()) -- sanity eq('Index out of bounds', pcall_err(get_lines, 1, 2, true)) eq('Index out of bounds', pcall_err(get_lines, -3, -2, true)) end) it('can get a single line with non-strict indexing', function() set_lines(0, 1, true, { 'line1.a' }) eq(1, line_count()) -- sanity eq({}, get_lines(1, 2, false)) eq({}, get_lines(-3, -2, false)) end) it('can set and delete a single line with strict indexing', function() set_lines(0, 1, true, { 'line1.a' }) eq('Index out of bounds', pcall_err(set_lines, 1, 2, true, { 'line1.b' })) eq('Index out of bounds', pcall_err(set_lines, -3, -2, true, { 'line1.c' })) eq({ 'line1.a' }, get_lines(0, -1, true)) eq('Index out of bounds', pcall_err(set_lines, 1, 2, true, {})) eq('Index out of bounds', pcall_err(set_lines, -3, -2, true, {})) eq({ 'line1.a' }, get_lines(0, -1, true)) end) it('can set and delete a single line with non-strict indexing', function() set_lines(0, 1, true, { 'line1.a' }) set_lines(1, 2, false, { 'line1.b' }) set_lines(-4, -3, false, { 'line1.c' }) eq({ 'line1.c', 'line1.a', 'line1.b' }, get_lines(0, -1, true)) set_lines(3, 4, false, {}) set_lines(-5, -4, false, {}) eq({ 'line1.c', 'line1.a', 'line1.b' }, get_lines(0, -1, true)) end) it('can handle NULs', function() set_lines(0, 1, true, { 'ab\0cd' }) eq({ 'ab\0cd' }, get_lines(0, -1, true)) end) it('works with multiple lines', function() eq({ '' }, get_lines(0, -1, true)) -- Replace buffer for _, mode in pairs({ false, true }) do set_lines(0, -1, mode, { 'a', 'b', 'c' }) eq({ 'a', 'b', 'c' }, get_lines(0, -1, mode)) eq({ 'b', 'c' }, get_lines(1, -1, mode)) eq({ 'b' }, get_lines(1, 2, mode)) eq({}, get_lines(1, 1, mode)) eq({ 'a', 'b' }, get_lines(0, -2, mode)) eq({ 'b' }, get_lines(1, -2, mode)) eq({ 'b', 'c' }, get_lines(-3, -1, mode)) set_lines(1, 2, mode, { 'a', 'b', 'c' }) eq({ 'a', 'a', 'b', 'c', 'c' }, get_lines(0, -1, mode)) set_lines(-2, -1, mode, { 'a', 'b', 'c' }) eq({ 'a', 'a', 'b', 'c', 'a', 'b', 'c' }, get_lines(0, -1, mode)) set_lines(0, -4, mode, {}) eq({ 'a', 'b', 'c' }, get_lines(0, -1, mode)) set_lines(0, -1, mode, {}) eq({ '' }, get_lines(0, -1, mode)) end end) it('can get line ranges with non-strict indexing', function() set_lines(0, -1, true, { 'a', 'b', 'c' }) eq({ 'a', 'b', 'c' }, get_lines(0, -1, true)) --sanity eq({}, get_lines(3, 4, false)) eq({}, get_lines(3, 10, false)) eq({}, get_lines(-5, -5, false)) eq({}, get_lines(3, -1, false)) eq({}, get_lines(-3, -4, false)) end) it('can get line ranges with strict indexing', function() set_lines(0, -1, true, { 'a', 'b', 'c' }) eq({ 'a', 'b', 'c' }, get_lines(0, -1, true)) --sanity eq('Index out of bounds', pcall_err(get_lines, 3, 4, true)) eq('Index out of bounds', pcall_err(get_lines, 3, 10, true)) eq('Index out of bounds', pcall_err(get_lines, -5, -5, true)) -- empty or inverted ranges are not errors eq({}, get_lines(3, -1, true)) eq({}, get_lines(-3, -4, true)) end) it('set_lines: out-of-bounds can extend past end', function() set_lines(0, -1, true, { 'a', 'b', 'c' }) eq({ 'a', 'b', 'c' }, get_lines(0, -1, true)) --sanity eq({ 'c' }, get_lines(-2, 5, false)) eq({ 'a', 'b', 'c' }, get_lines(0, 6, false)) eq('Index out of bounds', pcall_err(set_lines, 4, 6, true, { 'd' })) set_lines(4, 6, false, { 'd' }) eq({ 'a', 'b', 'c', 'd' }, get_lines(0, -1, true)) eq('Index out of bounds', pcall_err(set_lines, -6, -6, true, { 'e' })) set_lines(-6, -6, false, { 'e' }) eq({ 'e', 'a', 'b', 'c', 'd' }, get_lines(0, -1, true)) end) it('set_lines on alternate buffer does not access invalid line (E315)', function() feed_command('set hidden') insert('Initial file') command('enew') insert([[ More Lines Than In The Other Buffer]]) feed_command('$') local retval = exc_exec("call nvim_buf_set_lines(1, 0, 1, v:false, ['test'])") eq(0, retval) end) it("set_lines of invisible buffer doesn't move cursor in current window", function() local screen = Screen.new(20, 5) screen:set_default_attr_ids({ [1] = { bold = true, foreground = Screen.colors.Blue1 }, [2] = { bold = true }, }) screen:attach() insert([[ Who would win? A real window with proper text]]) local buf = lua_or_rpc.nvim_create_buf(false, true) screen:expect([[ Who would win? | A real window | with proper tex^t | {1:~ }| | ]]) lua_or_rpc.nvim_buf_set_lines(buf, 0, -1, true, { 'or some', 'scratchy text' }) feed('i') -- provoke redraw screen:expect([[ Who would win? | A real window | with proper tex^t | {1:~ }| {2:-- INSERT --} | ]]) end) it('set_lines on hidden buffer preserves "previous window" #9741', function() insert([[ visible buffer line 1 line 2 ]]) local hiddenbuf = lua_or_rpc.nvim_create_buf(false, true) command('vsplit') command('vsplit') feed('lll') eq(3, fn.winnr()) feed('h') eq(2, fn.winnr()) lua_or_rpc.nvim_buf_set_lines(hiddenbuf, 0, -1, true, { 'hidden buffer line 1', 'line 2' }) feed('p') eq(3, fn.winnr()) end) it('set_lines on unloaded buffer #8659 #22670', function() local bufnr = api.nvim_get_current_buf() lua_or_rpc.nvim_buf_set_lines(bufnr, 0, -1, false, { 'a', 'b', 'c' }) lua_or_rpc.nvim_buf_set_name(bufnr, 'set_lines') finally(function() os.remove('set_lines') end) command('write!') command('new') command('bunload! ' .. bufnr) local new_bufnr = fn.bufnr('set_lines', true) lua_or_rpc.nvim_buf_set_lines(new_bufnr, 0, -1, false, {}) eq({ '' }, lua_or_rpc.nvim_buf_get_lines(new_bufnr, 0, -1, false)) end) end) describe('nvim_buf_set_text', function() local function get_lines(...) return api.nvim_buf_get_lines(0, ...) end local function set_text(...) return api.nvim_buf_set_text(0, ...) end it('works', function() insert([[ hello foo! text ]]) eq({ 'hello foo!' }, get_lines(0, 1, true)) -- can replace a single word set_text(0, 6, 0, 9, { 'world' }) eq({ 'hello world!', 'text' }, get_lines(0, 2, true)) -- can insert text set_text(0, 0, 0, 0, { 'well ' }) eq({ 'well hello world!', 'text' }, get_lines(0, 2, true)) -- can delete text set_text(0, 0, 0, 5, { '' }) eq({ 'hello world!', 'text' }, get_lines(0, 2, true)) -- can replace with multiple lines set_text(0, 6, 0, 11, { 'foo', 'wo', 'more' }) eq({ 'hello foo', 'wo', 'more!', 'text' }, get_lines(0, 4, true)) -- will join multiple lines if needed set_text(0, 6, 3, 4, { 'bar' }) eq({ 'hello bar' }, get_lines(0, 1, true)) -- can use negative line numbers set_text(-2, 0, -2, 5, { 'goodbye' }) eq({ 'goodbye bar', '' }, get_lines(0, -1, true)) set_text(-1, 0, -1, 0, { 'text' }) eq({ 'goodbye bar', 'text' }, get_lines(0, 2, true)) -- can append to a line set_text(1, 4, -1, 4, { ' and', 'more' }) eq({ 'goodbye bar', 'text and', 'more' }, get_lines(0, 3, true)) -- can use negative column numbers set_text(0, -5, 0, -1, { '!' }) eq({ 'goodbye!' }, get_lines(0, 1, true)) end) it('works with undo', function() insert([[ hello world! foo bar ]]) -- setting text set_text(0, 0, 0, 0, { 'well ' }) feed('u') eq({ 'hello world!' }, get_lines(0, 1, true)) -- deleting text set_text(0, 0, 0, 6, { '' }) feed('u') eq({ 'hello world!' }, get_lines(0, 1, true)) -- inserting newlines set_text(0, 0, 0, 0, { 'hello', 'mr ' }) feed('u') eq({ 'hello world!' }, get_lines(0, 1, true)) -- deleting newlines set_text(0, 0, 1, 4, { 'hello' }) feed('u') eq({ 'hello world!' }, get_lines(0, 1, true)) end) it('updates the cursor position', function() insert([[ hello world! ]]) -- position the cursor on `!` api.nvim_win_set_cursor(0, { 1, 11 }) -- replace 'world' with 'foo' set_text(0, 6, 0, 11, { 'foo' }) eq('hello foo!', curbuf_depr('get_line', 0)) -- cursor should be moved left by two columns (replacement is shorter by 2 chars) eq({ 1, 9 }, api.nvim_win_get_cursor(0)) end) it('updates the cursor position in non-current window', function() insert([[ hello world!]]) -- position the cursor on `!` api.nvim_win_set_cursor(0, { 1, 11 }) local win = api.nvim_get_current_win() local buf = api.nvim_get_current_buf() command('new') -- replace 'world' with 'foo' api.nvim_buf_set_text(buf, 0, 6, 0, 11, { 'foo' }) eq({ 'hello foo!' }, api.nvim_buf_get_lines(buf, 0, -1, true)) -- cursor should be moved left by two columns (replacement is shorter by 2 chars) eq({ 1, 9 }, api.nvim_win_get_cursor(win)) end) it('updates the cursor position in TWO non-current windows', function() insert([[ hello world!]]) -- position the cursor on `!` api.nvim_win_set_cursor(0, { 1, 11 }) local win = api.nvim_get_current_win() local buf = api.nvim_get_current_buf() command('split') local win2 = api.nvim_get_current_win() -- position the cursor on `w` api.nvim_win_set_cursor(0, { 1, 6 }) command('new') -- replace 'hello' with 'foo' api.nvim_buf_set_text(buf, 0, 0, 0, 5, { 'foo' }) eq({ 'foo world!' }, api.nvim_buf_get_lines(buf, 0, -1, true)) -- both cursors should be moved left by two columns (replacement is shorter by 2 chars) eq({ 1, 9 }, api.nvim_win_get_cursor(win)) eq({ 1, 4 }, api.nvim_win_get_cursor(win2)) end) describe('when text is being added right at cursor position #22526', function() it('updates the cursor position in NORMAL mode', function() insert([[ abcd]]) -- position the cursor on 'c' api.nvim_win_set_cursor(0, { 1, 2 }) -- add 'xxx' before 'c' set_text(0, 2, 0, 2, { 'xxx' }) eq({ 'abxxxcd' }, get_lines(0, -1, true)) -- cursor should be on 'c' eq({ 1, 5 }, api.nvim_win_get_cursor(0)) end) it('updates the cursor position only in non-current window when in INSERT mode', function() insert([[ abcd]]) -- position the cursor on 'c' api.nvim_win_set_cursor(0, { 1, 2 }) -- open vertical split feed('v') -- get into INSERT mode to treat cursor -- as being after 'b', not on 'c' feed('i') -- add 'xxx' between 'b' and 'c' set_text(0, 2, 0, 2, { 'xxx' }) eq({ 'abxxxcd' }, get_lines(0, -1, true)) -- in the current window cursor should stay after 'b' eq({ 1, 2 }, api.nvim_win_get_cursor(0)) -- quit INSERT mode feed('') -- close current window feed('c') -- in another window cursor should be on 'c' eq({ 1, 5 }, api.nvim_win_get_cursor(0)) end) end) describe('when text is being deleted right at cursor position', function() it('leaves cursor at the same position in NORMAL mode', function() insert([[ abcd]]) -- position the cursor on 'b' api.nvim_win_set_cursor(0, { 1, 1 }) -- delete 'b' set_text(0, 1, 0, 2, {}) eq({ 'acd' }, get_lines(0, -1, true)) -- cursor is now on 'c' eq({ 1, 1 }, api.nvim_win_get_cursor(0)) end) it('maintains INSERT-mode cursor position current/non-current window', function() insert([[ abcd]]) -- position the cursor on 'b' api.nvim_win_set_cursor(0, { 1, 1 }) -- open vertical split feed('v') -- get into INSERT mode to treat cursor -- as being after 'a', not on 'b' feed('i') -- delete 'b' set_text(0, 1, 0, 2, {}) eq({ 'acd' }, get_lines(0, -1, true)) -- cursor in the current window should stay after 'a' eq({ 1, 1 }, api.nvim_win_get_cursor(0)) -- quit INSERT mode feed('') -- close current window feed('c') -- cursor in non-current window should stay on 'c' eq({ 1, 1 }, api.nvim_win_get_cursor(0)) end) end) describe('when cursor is inside replaced row range', function() it('maintains cursor position if at start_row, but before start_col', function() insert([[ This should be first then there is a line we do not want and finally the last one]]) -- position the cursor on ' ' before 'first' api.nvim_win_set_cursor(0, { 1, 14 }) set_text(0, 15, 2, 11, { 'the line we do not want', 'but hopefully', }) eq({ 'This should be the line we do not want', 'but hopefully the last one', }, get_lines(0, -1, true)) -- cursor should stay at the same position eq({ 1, 14 }, api.nvim_win_get_cursor(0)) end) it('maintains cursor position if at start_row and column is still valid', function() insert([[ This should be first then there is a line we do not want and finally the last one]]) -- position the cursor on 'f' in 'first' api.nvim_win_set_cursor(0, { 1, 15 }) set_text(0, 15, 2, 11, { 'the line we do not want', 'but hopefully', }) eq({ 'This should be the line we do not want', 'but hopefully the last one', }, get_lines(0, -1, true)) -- cursor should stay at the same position eq({ 1, 15 }, api.nvim_win_get_cursor(0)) end) it('adjusts cursor column to keep it valid if start_row got smaller', function() insert([[ This should be first then there is a line we do not want and finally the last one]]) -- position the cursor on 't' in 'first' api.nvim_win_set_cursor(0, { 1, 19 }) local cursor = exec_lua([[ vim.api.nvim_buf_set_text(0, 0, 15, 2, 24, {'last'}) return vim.api.nvim_win_get_cursor(0) ]]) eq({ 'This should be last' }, get_lines(0, -1, true)) -- cursor should end up on 't' in 'last' eq({ 1, 18 }, api.nvim_win_get_cursor(0)) -- immediate call to nvim_win_get_cursor should have returned the same position eq({ 1, 18 }, cursor) end) it('adjusts cursor column to keep it valid if start_row decreased in INSERT mode', function() insert([[ This should be first then there is a line we do not want and finally the last one]]) -- position the cursor on 't' in 'first' api.nvim_win_set_cursor(0, { 1, 19 }) -- enter INSERT mode to treat cursor as being after 't' feed('a') local cursor = exec_lua([[ vim.api.nvim_buf_set_text(0, 0, 15, 2, 24, {'last'}) return vim.api.nvim_win_get_cursor(0) ]]) eq({ 'This should be last' }, get_lines(0, -1, true)) -- cursor should end up after 't' in 'last' eq({ 1, 19 }, api.nvim_win_get_cursor(0)) -- immediate call to nvim_win_get_cursor should have returned the same position eq({ 1, 19 }, cursor) end) it('adjusts cursor to valid column in row after start_row if it got smaller', function() insert([[ This should be first then there is a line we do not want and finally the last one]]) -- position the cursor on 'w' in 'want' api.nvim_win_set_cursor(0, { 2, 31 }) local cursor = exec_lua([[ vim.api.nvim_buf_set_text(0, 0, 15, 2, 11, { '1', 'then 2', 'and then', }) return vim.api.nvim_win_get_cursor(0) ]]) eq({ 'This should be 1', 'then 2', 'and then the last one', }, get_lines(0, -1, true)) -- cursor column should end up at the end of a row eq({ 2, 5 }, api.nvim_win_get_cursor(0)) -- immediate call to nvim_win_get_cursor should have returned the same position eq({ 2, 5 }, cursor) end) it( 'adjusts cursor to valid column in row after start_row if it got smaller in INSERT mode', function() insert([[ This should be first then there is a line we do not want and finally the last one]]) -- position the cursor on 'w' in 'want' api.nvim_win_set_cursor(0, { 2, 31 }) -- enter INSERT mode feed('a') local cursor = exec_lua([[ vim.api.nvim_buf_set_text(0, 0, 15, 2, 11, { '1', 'then 2', 'and then', }) return vim.api.nvim_win_get_cursor(0) ]]) eq({ 'This should be 1', 'then 2', 'and then the last one', }, get_lines(0, -1, true)) -- cursor column should end up at the end of a row eq({ 2, 6 }, api.nvim_win_get_cursor(0)) -- immediate call to nvim_win_get_cursor should have returned the same position eq({ 2, 6 }, cursor) end ) it('adjusts cursor line and column to keep it inside replacement range', function() insert([[ This should be first then there is a line we do not want and finally the last one]]) -- position the cursor on 'n' in 'finally' api.nvim_win_set_cursor(0, { 3, 6 }) local cursor = exec_lua([[ vim.api.nvim_buf_set_text(0, 0, 15, 2, 11, { 'the line we do not want', 'but hopefully', }) return vim.api.nvim_win_get_cursor(0) ]]) eq({ 'This should be the line we do not want', 'but hopefully the last one', }, get_lines(0, -1, true)) -- cursor should end up on 'y' in 'hopefully' -- to stay in the range, because it got smaller eq({ 2, 12 }, api.nvim_win_get_cursor(0)) -- immediate call to nvim_win_get_cursor should have returned the same position eq({ 2, 12 }, cursor) end) it('adjusts cursor line and column if replacement is empty', function() insert([[ This should be first then there is a line we do not want and finally the last one]]) -- position the cursor on 'r' in 'there' api.nvim_win_set_cursor(0, { 2, 8 }) local cursor = exec_lua([[ vim.api.nvim_buf_set_text(0, 0, 15, 2, 12, {}) return vim.api.nvim_win_get_cursor(0) ]]) eq({ 'This should be the last one' }, get_lines(0, -1, true)) -- cursor should end up on the next column after deleted range eq({ 1, 15 }, api.nvim_win_get_cursor(0)) -- immediate call to nvim_win_get_cursor should have returned the same position eq({ 1, 15 }, cursor) end) it('adjusts cursor line and column if replacement is empty and start_col == 0', function() insert([[ This should be first then there is a line we do not want and finally the last one]]) -- position the cursor on 'r' in 'there' api.nvim_win_set_cursor(0, { 2, 8 }) local cursor = exec_lua([[ vim.api.nvim_buf_set_text(0, 0, 0, 2, 4, {}) return vim.api.nvim_win_get_cursor(0) ]]) eq({ 'finally the last one' }, get_lines(0, -1, true)) -- cursor should end up in column 0 eq({ 1, 0 }, api.nvim_win_get_cursor(0)) -- immediate call to nvim_win_get_cursor should have returned the same position eq({ 1, 0 }, cursor) end) it('adjusts cursor column if replacement ends at cursor row, after cursor column', function() insert([[ This should be first then there is a line we do not want and finally the last one]]) -- position the cursor on 'y' in 'finally' api.nvim_win_set_cursor(0, { 3, 10 }) set_text(0, 15, 2, 11, { '1', 'this 2', 'and then' }) eq({ 'This should be 1', 'this 2', 'and then the last one', }, get_lines(0, -1, true)) -- cursor should end up on 'n' in 'then' eq({ 3, 7 }, api.nvim_win_get_cursor(0)) end) it( 'adjusts cursor column if replacement ends at cursor row, at cursor column in INSERT mode', function() insert([[ This should be first then there is a line we do not want and finally the last one]]) -- position the cursor on 'y' at 'finally' api.nvim_win_set_cursor(0, { 3, 10 }) -- enter INSERT mode to treat cursor as being between 'l' and 'y' feed('i') set_text(0, 15, 2, 11, { '1', 'this 2', 'and then' }) eq({ 'This should be 1', 'this 2', 'and then the last one', }, get_lines(0, -1, true)) -- cursor should end up after 'n' in 'then' eq({ 3, 8 }, api.nvim_win_get_cursor(0)) end ) it('adjusts cursor column if replacement is inside of a single line', function() insert([[ This should be first then there is a line we do not want and finally the last one]]) -- position the cursor on 'y' in 'finally' api.nvim_win_set_cursor(0, { 3, 10 }) set_text(2, 4, 2, 11, { 'then' }) eq({ 'This should be first', 'then there is a line we do not want', 'and then the last one', }, get_lines(0, -1, true)) -- cursor should end up on 'n' in 'then' eq({ 3, 7 }, api.nvim_win_get_cursor(0)) end) it('does not move cursor column after end of a line', function() insert([[ This should be the only line here !!!]]) -- position cursor on the last '1' api.nvim_win_set_cursor(0, { 2, 2 }) local cursor = exec_lua([[ vim.api.nvim_buf_set_text(0, 0, 33, 1, 3, {}) return vim.api.nvim_win_get_cursor(0) ]]) eq({ 'This should be the only line here' }, get_lines(0, -1, true)) -- cursor should end up on '!' eq({ 1, 32 }, api.nvim_win_get_cursor(0)) -- immediate call to nvim_win_get_cursor should have returned the same position eq({ 1, 32 }, cursor) end) it('does not move cursor column before start of a line', function() insert('\n!!!') -- position cursor on the last '1' api.nvim_win_set_cursor(0, { 2, 2 }) local cursor = exec_lua([[ vim.api.nvim_buf_set_text(0, 0, 0, 1, 3, {}) return vim.api.nvim_win_get_cursor(0) ]]) eq({ '' }, get_lines(0, -1, true)) -- cursor should end up on '!' eq({ 1, 0 }, api.nvim_win_get_cursor(0)) -- immediate call to nvim_win_get_cursor should have returned the same position eq({ 1, 0 }, cursor) end) describe('with virtualedit', function() it('adjusts cursor line/col to keep inside replacement range if not after eol', function() insert([[ This should be first then there is a line we do not want and finally the last one]]) -- position cursor on 't' in 'want' api.nvim_win_set_cursor(0, { 2, 34 }) -- turn on virtualedit command('set virtualedit=all') local cursor = exec_lua([[ vim.api.nvim_buf_set_text(0, 0, 15, 2, 11, { 'the line we do not want', 'but hopefully', }) return vim.api.nvim_win_get_cursor(0) ]]) eq({ 'This should be the line we do not want', 'but hopefully the last one', }, get_lines(0, -1, true)) -- cursor should end up on 'y' in 'hopefully' -- to stay in the range eq({ 2, 12 }, api.nvim_win_get_cursor(0)) -- immediate call to nvim_win_get_cursor should have returned the same position eq({ 2, 12 }, cursor) -- coladd should be 0 eq( 0, exec_lua([[ return vim.fn.winsaveview().coladd ]]) ) end) it('does not change cursor screen column when cursor >EOL and row got shorter', function() insert([[ This should be first then there is a line we do not want and finally the last one]]) -- position cursor on 't' in 'want' api.nvim_win_set_cursor(0, { 2, 34 }) -- turn on virtualedit command('set virtualedit=all') -- move cursor after eol exec_lua([[ vim.fn.winrestview({ coladd = 5 }) ]]) local cursor = exec_lua([[ vim.api.nvim_buf_set_text(0, 0, 15, 2, 11, { 'the line we do not want', 'but hopefully', }) return vim.api.nvim_win_get_cursor(0) ]]) eq({ 'This should be the line we do not want', 'but hopefully the last one', }, get_lines(0, -1, true)) -- cursor should end up at eol of a new row eq({ 2, 26 }, api.nvim_win_get_cursor(0)) -- immediate call to nvim_win_get_cursor should have returned the same position eq({ 2, 26 }, cursor) -- coladd should be increased so that cursor stays in the same screen column eq( 13, exec_lua([[ return vim.fn.winsaveview().coladd ]]) ) end) it( 'does not change cursor screen column when cursor is after eol and row got longer', function() insert([[ This should be first then there is a line we do not want and finally the last one]]) -- position cursor on 't' in 'first' api.nvim_win_set_cursor(0, { 1, 19 }) -- turn on virtualedit command('set virtualedit=all') -- move cursor after eol exec_lua([[ vim.fn.winrestview({ coladd = 21 }) ]]) local cursor = exec_lua([[ vim.api.nvim_buf_set_text(0, 0, 15, 2, 11, { 'the line we do not want', 'but hopefully', }) return vim.api.nvim_win_get_cursor(0) ]]) eq({ 'This should be the line we do not want', 'but hopefully the last one', }, get_lines(0, -1, true)) -- cursor should end up at eol of a new row eq({ 1, 38 }, api.nvim_win_get_cursor(0)) -- immediate call to nvim_win_get_cursor should have returned the same position eq({ 1, 38 }, cursor) -- coladd should be increased so that cursor stays in the same screen column eq( 2, exec_lua([[ return vim.fn.winsaveview().coladd ]]) ) end ) it( 'does not change cursor screen column when cursor is after eol and row extended past cursor column', function() insert([[ This should be first then there is a line we do not want and finally the last one]]) -- position cursor on 't' in 'first' api.nvim_win_set_cursor(0, { 1, 19 }) -- turn on virtualedit command('set virtualedit=all') -- move cursor after eol just a bit exec_lua([[ vim.fn.winrestview({ coladd = 3 }) ]]) local cursor = exec_lua([[ vim.api.nvim_buf_set_text(0, 0, 15, 2, 11, { 'the line we do not want', 'but hopefully', }) return vim.api.nvim_win_get_cursor(0) ]]) eq({ 'This should be the line we do not want', 'but hopefully the last one', }, get_lines(0, -1, true)) -- cursor should stay at the same screen column eq({ 1, 22 }, api.nvim_win_get_cursor(0)) -- immediate call to nvim_win_get_cursor should have returned the same position eq({ 1, 22 }, cursor) -- coladd should become 0 eq( 0, exec_lua([[ return vim.fn.winsaveview().coladd ]]) ) end ) it( 'does not change cursor screen column when cursor is after eol and row range decreased', function() insert([[ This should be first then there is a line we do not want and one more and finally the last one]]) -- position cursor on 'e' in 'more' api.nvim_win_set_cursor(0, { 3, 11 }) -- turn on virtualedit command('set virtualedit=all') -- move cursor after eol exec_lua([[ vim.fn.winrestview({ coladd = 28 }) ]]) local cursor = exec_lua([[ vim.api.nvim_buf_set_text(0, 0, 15, 3, 11, { 'the line we do not want', 'but hopefully', }) return vim.api.nvim_win_get_cursor(0) ]]) eq({ 'This should be the line we do not want', 'but hopefully the last one', }, get_lines(0, -1, true)) -- cursor should end up at eol of a new row eq({ 2, 26 }, api.nvim_win_get_cursor(0)) -- immediate call to nvim_win_get_cursor should have returned the same position eq({ 2, 26 }, cursor) -- coladd should be increased so that cursor stays in the same screen column eq( 13, exec_lua([[ return vim.fn.winsaveview().coladd ]]) ) end ) end) end) describe('when cursor is at end_row and after end_col', function() it('adjusts cursor column when only a newline is added or deleted', function() insert([[ first line second line]]) -- position the cursor on 'i' api.nvim_win_set_cursor(0, { 3, 2 }) set_text(1, 6, 2, 0, {}) eq({ 'first line', 'second line' }, get_lines(0, -1, true)) -- cursor should stay on 'i' eq({ 2, 8 }, api.nvim_win_get_cursor(0)) -- add a newline back set_text(1, 6, 1, 6, { '', '' }) eq({ 'first line', 'second', ' line' }, get_lines(0, -1, true)) -- cursor should return back to the original position eq({ 3, 2 }, api.nvim_win_get_cursor(0)) end) it( 'adjusts cursor column if the range is not bound to either start or end of a line', function() insert([[ This should be first then there is a line we do not want and finally the last one]]) -- position the cursor on 'h' in 'the' api.nvim_win_set_cursor(0, { 3, 13 }) set_text(0, 14, 2, 11, {}) eq({ 'This should be the last one' }, get_lines(0, -1, true)) -- cursor should stay on 'h' eq({ 1, 16 }, api.nvim_win_get_cursor(0)) -- add deleted lines back set_text(0, 14, 0, 14, { ' first', 'then there is a line we do not want', 'and finally', }) eq({ 'This should be first', 'then there is a line we do not want', 'and finally the last one', }, get_lines(0, -1, true)) -- cursor should return back to the original position eq({ 3, 13 }, api.nvim_win_get_cursor(0)) end ) it( 'adjusts cursor column if replacing lines in range, not just deleting and adding', function() insert([[ This should be first then there is a line we do not want and finally the last one]]) -- position the cursor on 's' in 'last' api.nvim_win_set_cursor(0, { 3, 18 }) set_text(0, 15, 2, 11, { 'the line we do not want', 'but hopefully', }) eq({ 'This should be the line we do not want', 'but hopefully the last one', }, get_lines(0, -1, true)) -- cursor should stay on 's' eq({ 2, 20 }, api.nvim_win_get_cursor(0)) set_text(0, 15, 1, 13, { 'first', 'then there is a line we do not want', 'and finally', }) eq({ 'This should be first', 'then there is a line we do not want', 'and finally the last one', }, get_lines(0, -1, true)) -- cursor should return back to the original position eq({ 3, 18 }, api.nvim_win_get_cursor(0)) end ) it('does not move cursor column after end of a line', function() insert([[ This should be the only line here ]]) -- position cursor at the empty line api.nvim_win_set_cursor(0, { 2, 0 }) local cursor = exec_lua([[ vim.api.nvim_buf_set_text(0, 0, 33, 1, 0, {'!'}) return vim.api.nvim_win_get_cursor(0) ]]) eq({ 'This should be the only line here!' }, get_lines(0, -1, true)) -- cursor should end up on '!' eq({ 1, 33 }, api.nvim_win_get_cursor(0)) -- immediate call to nvim_win_get_cursor should have returned the same position eq({ 1, 33 }, cursor) end) it('does not move cursor column before start of a line', function() insert('\n') eq({ '', '' }, get_lines(0, -1, true)) -- position cursor on the last '1' api.nvim_win_set_cursor(0, { 2, 2 }) local cursor = exec_lua([[ vim.api.nvim_buf_set_text(0, 0, 0, 1, 0, {''}) return vim.api.nvim_win_get_cursor(0) ]]) eq({ '' }, get_lines(0, -1, true)) -- cursor should end up on '!' eq({ 1, 0 }, api.nvim_win_get_cursor(0)) -- immediate call to nvim_win_get_cursor should have returned the same position eq({ 1, 0 }, cursor) end) end) it('can handle NULs', function() set_text(0, 0, 0, 0, { 'ab\0cd' }) eq('ab\0cd', curbuf_depr('get_line', 0)) end) it('adjusts extmarks', function() local ns = api.nvim_create_namespace('my-fancy-plugin') insert([[ foo bar baz ]]) local id1 = api.nvim_buf_set_extmark(0, ns, 0, 1, {}) local id2 = api.nvim_buf_set_extmark(0, ns, 0, 7, {}) local id3 = api.nvim_buf_set_extmark(0, ns, 1, 1, {}) set_text(0, 4, 0, 7, { 'q' }) eq({ 'foo q', 'baz' }, get_lines(0, 2, true)) -- mark before replacement point is unaffected eq({ 0, 1 }, api.nvim_buf_get_extmark_by_id(0, ns, id1, {})) -- mark gets shifted back because the replacement was shorter eq({ 0, 5 }, api.nvim_buf_get_extmark_by_id(0, ns, id2, {})) -- mark on the next line is unaffected eq({ 1, 1 }, api.nvim_buf_get_extmark_by_id(0, ns, id3, {})) -- replacing the text spanning two lines will adjust the mark on the next line set_text(0, 3, 1, 3, { 'qux' }) eq({ 'fooqux', '' }, get_lines(0, 2, true)) eq({ 0, 6 }, api.nvim_buf_get_extmark_by_id(0, ns, id3, {})) -- but mark before replacement point is still unaffected eq({ 0, 1 }, api.nvim_buf_get_extmark_by_id(0, ns, id1, {})) -- and the mark in the middle was shifted to the end of the insertion eq({ 0, 6 }, api.nvim_buf_get_extmark_by_id(0, ns, id2, {})) -- marks should be put back into the same place after undoing set_text(0, 0, 0, 2, { '' }) feed('u') eq({ 0, 1 }, api.nvim_buf_get_extmark_by_id(0, ns, id1, {})) eq({ 0, 6 }, api.nvim_buf_get_extmark_by_id(0, ns, id2, {})) eq({ 0, 6 }, api.nvim_buf_get_extmark_by_id(0, ns, id3, {})) -- marks should be shifted over by the correct number of bytes for multibyte -- chars set_text(0, 0, 0, 0, { 'Ø' }) eq({ 0, 3 }, api.nvim_buf_get_extmark_by_id(0, ns, id1, {})) eq({ 0, 8 }, api.nvim_buf_get_extmark_by_id(0, ns, id2, {})) eq({ 0, 8 }, api.nvim_buf_get_extmark_by_id(0, ns, id3, {})) end) it('correctly marks changed region for redraw #13890', function() local screen = Screen.new(20, 5) screen:attach() insert([[ AAA BBB ]]) api.nvim_buf_set_text(0, 0, 0, 1, 3, { 'XXX', 'YYY' }) screen:expect([[ XXX | YYY | ^ | ~ | | ]]) end) it('errors on out-of-range', function() insert([[ hello foo! text]]) eq("Invalid 'start_row': out of range", pcall_err(set_text, 2, 0, 3, 0, {})) eq("Invalid 'start_row': out of range", pcall_err(set_text, -3, 0, 0, 0, {})) eq("Invalid 'end_row': out of range", pcall_err(set_text, 0, 0, 2, 0, {})) eq("Invalid 'end_row': out of range", pcall_err(set_text, 0, 0, -3, 0, {})) eq("Invalid 'start_col': out of range", pcall_err(set_text, 1, 5, 1, 5, {})) eq("Invalid 'end_col': out of range", pcall_err(set_text, 1, 0, 1, 5, {})) end) it('errors when start is greater than end', function() insert([[ hello foo! text]]) eq("'start' is higher than 'end'", pcall_err(set_text, 1, 0, 0, 0, {})) eq("'start' is higher than 'end'", pcall_err(set_text, 0, 1, 0, 0, {})) end) it('no heap-use-after-free when called consecutively #19643', function() set_text(0, 0, 0, 0, { 'one', '', '', 'two' }) eq({ 'one', '', '', 'two' }, get_lines(0, 4, true)) api.nvim_win_set_cursor(0, { 1, 0 }) exec_lua([[ vim.api.nvim_buf_set_text(0, 0, 3, 1, 0, {''}) vim.api.nvim_buf_set_text(0, 0, 3, 1, 0, {''}) ]]) eq({ 'one', 'two' }, get_lines(0, 2, true)) end) describe('handles topline', function() local screen before_each(function() screen = Screen.new(20, 12) screen:set_default_attr_ids { [1] = { bold = true, foreground = Screen.colors.Blue1 }, [2] = { reverse = true, bold = true }, [3] = { reverse = true }, } screen:attach() api.nvim_buf_set_lines( 0, 0, -1, true, { 'aaa', 'bbb', 'ccc', 'ddd', 'www', 'xxx', 'yyy', 'zzz' } ) api.nvim_set_option_value('modified', false, {}) end) it('of current window', function() local win = api.nvim_get_current_win() local buf = api.nvim_get_current_buf() command('new | wincmd w') api.nvim_win_set_cursor(win, { 8, 0 }) screen:expect { grid = [[ | {1:~ }|*4 {3:[No Name] }| www | xxx | yyy | ^zzz | {2:[No Name] }| | ]], } api.nvim_buf_set_text(buf, 0, 3, 1, 0, { 'X' }) screen:expect { grid = [[ | {1:~ }|*4 {3:[No Name] }| www | xxx | yyy | ^zzz | {2:[No Name] [+] }| | ]], } end) it('of non-current window', function() local win = api.nvim_get_current_win() local buf = api.nvim_get_current_buf() command('new') api.nvim_win_set_cursor(win, { 8, 0 }) screen:expect { grid = [[ ^ | {1:~ }|*4 {2:[No Name] }| www | xxx | yyy | zzz | {3:[No Name] }| | ]], } api.nvim_buf_set_text(buf, 0, 3, 1, 0, { 'X' }) screen:expect { grid = [[ ^ | {1:~ }|*4 {2:[No Name] }| www | xxx | yyy | zzz | {3:[No Name] [+] }| | ]], } end) it('of split windows with same buffer', function() local win = api.nvim_get_current_win() local buf = api.nvim_get_current_buf() command('split') api.nvim_win_set_cursor(win, { 8, 0 }) api.nvim_win_set_cursor(0, { 1, 1 }) screen:expect { grid = [[ a^aa | bbb | ccc | ddd | www | {2:[No Name] }| www | xxx | yyy | zzz | {3:[No Name] }| | ]], } api.nvim_buf_set_text(buf, 0, 3, 1, 0, { 'X' }) screen:expect { grid = [[ a^aaXbbb | ccc | ddd | www | xxx | {2:[No Name] [+] }| www | xxx | yyy | zzz | {3:[No Name] [+] }| | ]], } end) end) end) describe_lua_and_rpc('nvim_buf_get_text', function(lua_or_rpc) local get_text = lua_or_rpc.nvim_buf_get_text before_each(function() insert([[ hello foo! text more]]) end) it('works', function() eq({ 'hello' }, get_text(0, 0, 0, 0, 5, {})) eq({ 'hello foo!' }, get_text(0, 0, 0, 0, 42, {})) eq({ 'foo!' }, get_text(0, 0, 6, 0, 10, {})) eq({ 'foo!', 'tex' }, get_text(0, 0, 6, 1, 3, {})) eq({ 'foo!', 'tex' }, get_text(0, -3, 6, -2, 3, {})) eq({ '' }, get_text(0, 0, 18, 0, 20, {})) eq({ 'ext' }, get_text(0, -2, 1, -2, 4, {})) eq({ 'hello foo!', 'text', 'm' }, get_text(0, 0, 0, 2, 1, {})) end) it('errors on out-of-range', function() eq('Index out of bounds', pcall_err(get_text, 0, 2, 0, 4, 0, {})) eq('Index out of bounds', pcall_err(get_text, 0, -4, 0, 0, 0, {})) eq('Index out of bounds', pcall_err(get_text, 0, 0, 0, 3, 0, {})) eq('Index out of bounds', pcall_err(get_text, 0, 0, 0, -4, 0, {})) -- no ml_get errors should happen #19017 eq('', api.nvim_get_vvar('errmsg')) end) it('errors when start is greater than end', function() eq("'start' is higher than 'end'", pcall_err(get_text, 0, 1, 0, 0, 0, {})) eq('start_col must be less than end_col', pcall_err(get_text, 0, 0, 1, 0, 0, {})) end) end) describe('nvim_buf_get_offset', function() local get_offset = api.nvim_buf_get_offset it('works', function() api.nvim_buf_set_lines(0, 0, -1, true, { 'Some\r', 'exa\000mple', '', 'buf\rfer', 'text' }) eq(5, api.nvim_buf_line_count(0)) eq(0, get_offset(0, 0)) eq(6, get_offset(0, 1)) eq(15, get_offset(0, 2)) eq(16, get_offset(0, 3)) eq(24, get_offset(0, 4)) eq(29, get_offset(0, 5)) eq('Index out of bounds', pcall_err(get_offset, 0, 6)) eq('Index out of bounds', pcall_err(get_offset, 0, -1)) api.nvim_set_option_value('eol', false, {}) api.nvim_set_option_value('fixeol', false, {}) eq(28, get_offset(0, 5)) -- fileformat is ignored api.nvim_set_option_value('fileformat', 'dos', {}) eq(0, get_offset(0, 0)) eq(6, get_offset(0, 1)) eq(15, get_offset(0, 2)) eq(16, get_offset(0, 3)) eq(24, get_offset(0, 4)) eq(28, get_offset(0, 5)) api.nvim_set_option_value('eol', true, {}) eq(29, get_offset(0, 5)) command('set hidden') command('enew') eq(6, api.nvim_buf_get_offset(1, 1)) command('bunload! 1') eq(-1, api.nvim_buf_get_offset(1, 1)) eq(-1, api.nvim_buf_get_offset(1, 0)) end) it('works in empty buffer', function() eq(0, get_offset(0, 0)) eq(1, get_offset(0, 1)) eq(-1, fn.line2byte('$')) end) it('works in buffer with one line inserted', function() feed('itext') eq(0, get_offset(0, 0)) eq(5, get_offset(0, 1)) end) end) describe('nvim_buf_get_var, nvim_buf_set_var, nvim_buf_del_var', function() it('works', function() api.nvim_buf_set_var(0, 'lua', { 1, 2, { ['3'] = 1 } }) eq({ 1, 2, { ['3'] = 1 } }, api.nvim_buf_get_var(0, 'lua')) eq({ 1, 2, { ['3'] = 1 } }, api.nvim_eval('b:lua')) eq(1, fn.exists('b:lua')) api.nvim_buf_del_var(0, 'lua') eq(0, fn.exists('b:lua')) eq('Key not found: lua', pcall_err(api.nvim_buf_del_var, 0, 'lua')) api.nvim_buf_set_var(0, 'lua', 1) command('lockvar b:lua') eq('Key is locked: lua', pcall_err(api.nvim_buf_del_var, 0, 'lua')) eq('Key is locked: lua', pcall_err(api.nvim_buf_set_var, 0, 'lua', 1)) eq('Key is read-only: changedtick', pcall_err(api.nvim_buf_del_var, 0, 'changedtick')) eq('Key is read-only: changedtick', pcall_err(api.nvim_buf_set_var, 0, 'changedtick', 1)) end) end) describe('nvim_buf_get_changedtick', function() it('works', function() eq(2, api.nvim_buf_get_changedtick(0)) api.nvim_buf_set_lines(0, 0, 1, false, { 'abc\0', '\0def', 'ghi' }) eq(3, api.nvim_buf_get_changedtick(0)) eq(3, api.nvim_buf_get_var(0, 'changedtick')) end) it('buffer_set_var returns the old value', function() local val1 = { 1, 2, { ['3'] = 1 } } local val2 = { 4, 7 } eq(NIL, request('buffer_set_var', 0, 'lua', val1)) eq(val1, request('buffer_set_var', 0, 'lua', val2)) end) it('buffer_del_var returns the old value', function() local val1 = { 1, 2, { ['3'] = 1 } } local val2 = { 4, 7 } eq(NIL, request('buffer_set_var', 0, 'lua', val1)) eq(val1, request('buffer_set_var', 0, 'lua', val2)) eq(val2, request('buffer_del_var', 0, 'lua')) end) end) describe('nvim_get_option_value, nvim_set_option_value', function() it('works', function() eq(8, api.nvim_get_option_value('shiftwidth', {})) api.nvim_set_option_value('shiftwidth', 4, {}) eq(4, api.nvim_get_option_value('shiftwidth', {})) -- global-local option api.nvim_set_option_value('define', 'test', { buf = 0 }) eq('test', api.nvim_get_option_value('define', { buf = 0 })) -- Doesn't change the global value eq('', api.nvim_get_option_value('define', { scope = 'global' })) end) it('returns values for unset local options', function() -- 'undolevels' is only set to its "unset" value when a new buffer is -- created command('enew') eq(-123456, api.nvim_get_option_value('undolevels', { buf = 0 })) end) end) describe('nvim_buf_get_name, nvim_buf_set_name', function() it('works', function() command('new') eq('', api.nvim_buf_get_name(0)) local new_name = api.nvim_eval('resolve(tempname())') api.nvim_buf_set_name(0, new_name) eq(new_name, api.nvim_buf_get_name(0)) command('w!') eq(1, fn.filereadable(new_name)) os.remove(new_name) end) end) describe('nvim_buf_is_loaded', function() it('works', function() -- record our buffer number for when we unload it local bufnr = api.nvim_buf_get_number(0) -- api should report that the buffer is loaded ok(api.nvim_buf_is_loaded(bufnr)) -- hide the current buffer by switching to a new empty buffer -- Careful! we need to modify the buffer first or vim will just reuse it api.nvim_buf_set_lines(bufnr, 0, -1, true, { 'line1' }) command('hide enew') -- confirm the buffer is hidden, but still loaded local infolist = api.nvim_eval('getbufinfo(' .. bufnr .. ')') eq(1, #infolist) eq(1, infolist[1].hidden) eq(1, infolist[1].loaded) -- now force unload the buffer command('bunload! ' .. bufnr) -- confirm the buffer is unloaded infolist = api.nvim_eval('getbufinfo(' .. bufnr .. ')') eq(0, infolist[1].loaded) -- nvim_buf_is_loaded() should also report the buffer as unloaded eq(false, api.nvim_buf_is_loaded(bufnr)) end) end) describe('nvim_buf_is_valid', function() it('works', function() command('new') local b = api.nvim_get_current_buf() ok(api.nvim_buf_is_valid(b)) command('bw!') ok(not api.nvim_buf_is_valid(b)) end) end) describe('nvim_buf_delete', function() it('allows for just deleting', function() command('new') local b = api.nvim_get_current_buf() ok(api.nvim_buf_is_valid(b)) api.nvim_buf_delete(b, {}) ok(not api.nvim_buf_is_loaded(b)) ok(not api.nvim_buf_is_valid(b)) end) it('allows for just unloading', function() command('new') local b = api.nvim_get_current_buf() ok(api.nvim_buf_is_valid(b)) api.nvim_buf_delete(b, { unload = true }) ok(not api.nvim_buf_is_loaded(b)) ok(api.nvim_buf_is_valid(b)) end) end) describe('nvim_buf_get_mark', function() it('works', function() api.nvim_buf_set_lines(0, -1, -1, true, { 'a', 'bit of', 'text' }) api.nvim_win_set_cursor(0, { 3, 4 }) command('mark v') eq({ 3, 0 }, api.nvim_buf_get_mark(0, 'v')) end) end) describe('nvim_buf_set_mark', function() it('works with buffer local marks', function() api.nvim_buf_set_lines(0, -1, -1, true, { 'a', 'bit of', 'text' }) eq(true, api.nvim_buf_set_mark(0, 'z', 1, 1, {})) eq({ 1, 1 }, api.nvim_buf_get_mark(0, 'z')) eq({ 0, 1, 2, 0 }, fn.getpos("'z")) end) it('works with file/uppercase marks', function() api.nvim_buf_set_lines(0, -1, -1, true, { 'a', 'bit of', 'text' }) eq(true, api.nvim_buf_set_mark(0, 'Z', 3, 2, {})) eq({ 3, 2 }, api.nvim_buf_get_mark(0, 'Z')) eq({ api.nvim_get_current_buf(), 3, 3, 0 }, fn.getpos("'Z")) end) it('fails when invalid marks names are used', function() eq(false, pcall(api.nvim_buf_set_mark, 0, '!', 1, 0, {})) eq(false, pcall(api.nvim_buf_set_mark, 0, 'fail', 1, 0, {})) end) it('fails when invalid buffer number is used', function() eq(false, pcall(api.nvim_buf_set_mark, 99, 'a', 1, 1, {})) end) end) describe('nvim_buf_del_mark', function() it('works with buffer local marks', function() api.nvim_buf_set_lines(0, -1, -1, true, { 'a', 'bit of', 'text' }) api.nvim_buf_set_mark(0, 'z', 3, 1, {}) eq(true, api.nvim_buf_del_mark(0, 'z')) eq({ 0, 0 }, api.nvim_buf_get_mark(0, 'z')) end) it('works with file/uppercase marks', function() api.nvim_buf_set_lines(0, -1, -1, true, { 'a', 'bit of', 'text' }) api.nvim_buf_set_mark(0, 'Z', 3, 3, {}) eq(true, api.nvim_buf_del_mark(0, 'Z')) eq({ 0, 0 }, api.nvim_buf_get_mark(0, 'Z')) end) it('returns false in marks not set in this buffer', function() local abuf = api.nvim_create_buf(false, true) api.nvim_buf_set_lines(abuf, -1, -1, true, { 'a', 'bit of', 'text' }) api.nvim_buf_set_mark(abuf, 'A', 2, 2, {}) eq(false, api.nvim_buf_del_mark(0, 'A')) eq({ 2, 2 }, api.nvim_buf_get_mark(abuf, 'A')) end) it('returns false if mark was not deleted', function() api.nvim_buf_set_lines(0, -1, -1, true, { 'a', 'bit of', 'text' }) api.nvim_buf_set_mark(0, 'z', 3, 1, {}) eq(true, api.nvim_buf_del_mark(0, 'z')) eq(false, api.nvim_buf_del_mark(0, 'z')) -- Mark was already deleted end) it('fails when invalid marks names are used', function() eq(false, pcall(api.nvim_buf_del_mark, 0, '!')) eq(false, pcall(api.nvim_buf_del_mark, 0, 'fail')) end) it('fails when invalid buffer number is used', function() eq(false, pcall(api.nvim_buf_del_mark, 99, 'a')) end) end) end)