local t = require('test.testutil')

local n = require('test.functional.testnvim')()

local Screen = require('test.functional.ui.screen')

local eq = t.eq

local feed = n.feed

local api = n.api

local clear = n.clear

local source = n.source

local command = n.command

local exc_exec = n.exc_exec

local async_meths = n.async_meths

local NIL = vim.NIL

local screen

before_each(function()

clear()

screen = Screen.new(25, 5)

source([[

hi Test ctermfg=Red guifg=Red term=bold

function CustomCompl(...)

return 'TEST'

endfunction

function CustomListCompl(...)

return ['FOO']

endfunction

highlight RBP1 guibg=Red

highlight RBP2 guibg=Yellow

highlight RBP3 guibg=Green

highlight RBP4 guibg=Blue

let g:NUM_LVLS = 4

function Redraw()

redraw!

return ''

endfunction

cnoremap <expr> {REDRAW} Redraw()

function RainBowParens(cmdline)

let ret = []

let i = 0

let lvl = 0

while i < len(a:cmdline)

if a:cmdline[i] is# '('

call add(ret, [i, i + 1, 'RBP' . ((lvl % g:NUM_LVLS) + 1)])

let lvl += 1

elseif a:cmdline[i] is# ')'

let lvl -= 1

call add(ret, [i, i + 1, 'RBP' . ((lvl % g:NUM_LVLS) + 1)])

endif

let i += 1

endwhile

return ret

endfunction

]])

screen:set_default_attr_ids({

EOB = { bold = true, foreground = Screen.colors.Blue1 },

T = { foreground = Screen.colors.Red },

RBP1 = { background = Screen.colors.Red },

RBP2 = { background = Screen.colors.Yellow },

RBP3 = { background = Screen.colors.Green },

RBP4 = { background = Screen.colors.Blue },

SEP = { bold = true, reverse = true },

CONFIRM = { bold = true, foreground = Screen.colors.SeaGreen4 },

})

end)

describe('input()', function()

it('works with multiline prompts', function()

feed([[:call input("Test\nFoo")<CR>]])

screen:expect([[

|

{EOB:~ }|

{SEP: }|

Test |

Foo^ |

]])

end)

it('works with multiline prompts and :echohl', function()

feed([[:echohl Test | call input("Test\nFoo")<CR>]])

screen:expect([[

|

{EOB:~ }|

{SEP: }|

{T:Test} |

{T:Foo}^ |

]])

command('redraw!')

screen:expect([[

|

{EOB:~ }|*3

{T:Foo}^ |

]])

end)

it('allows unequal numeric arguments when using multiple args', function()

command('echohl Test')

feed([[:call input(1, 2)<CR>]])

screen:expect([[

|

{EOB:~ }|*3

{T:1}2^ |

]])

feed('<BS>')

screen:expect([[

|

{EOB:~ }|*3

{T:1}^ |

]])

end)

it('allows unequal numeric values when using {opts} dict', function()

command('echohl Test')

api.nvim_set_var('opts', { prompt = 1, default = 2, cancelreturn = 3 })

feed([[:echo input(opts)<CR>]])

screen:expect([[

|

{EOB:~ }|*3

{T:1}2^ |

]])

feed('<BS>')

screen:expect([[

|

{EOB:~ }|*3

{T:1}^ |

]])

feed('<Esc>')

screen:expect([[

^ |

{EOB:~ }|*3

{T:3} |

]])

end)

it('works with redraw', function()

command('echohl Test')

api.nvim_set_var('opts', { prompt = 'Foo>', default = 'Bar' })

feed([[:echo inputdialog(opts)<CR>]])

screen:expect([[

|

{EOB:~ }|*3

{T:Foo>}Bar^ |

]])

command('mode')

screen:expect {

grid = [[

|

{EOB:~ }|*3

{T:Foo>}Bar^ |

]],

reset = true,

}

feed('<BS>')

screen:expect([[

|

{EOB:~ }|*3

{T:Foo>}Ba^ |

]])

command('mode')

screen:expect {

grid = [[

|

{EOB:~ }|*3

{T:Foo>}Ba^ |

]],

reset = true,

}

end)

it('allows omitting everything with dict argument', function()

command('echohl Test')

feed([[:call input({})<CR>]])

screen:expect([[

|

{EOB:~ }|*3

^ |

]])

end)

it('supports completion', function()

feed(':let var = input("", "", "custom,CustomCompl")<CR>')

feed('<Tab><CR>')

eq('TEST', api.nvim_get_var('var'))

feed(':let var = input({"completion": "customlist,CustomListCompl"})<CR>')

feed('<Tab><CR>')

eq('FOO', api.nvim_get_var('var'))

end)

it('supports cancelreturn', function()

feed(':let var = input({"cancelreturn": "BAR"})<CR>')

feed('<Esc>')

eq('BAR', api.nvim_get_var('var'))

feed(':let var = input({"cancelreturn": []})<CR>')

feed('<Esc>')

eq({}, api.nvim_get_var('var'))

feed(':let var = input({"cancelreturn": v:false})<CR>')

feed('<Esc>')

eq(false, api.nvim_get_var('var'))

feed(':let var = input({"cancelreturn": v:null})<CR>')

feed('<Esc>')

eq(NIL, api.nvim_get_var('var'))

end)

it('supports default string', function()

feed(':let var = input("", "DEF1")<CR>')

feed('<CR>')

eq('DEF1', api.nvim_get_var('var'))

feed(':let var = input({"default": "DEF2"})<CR>')

feed('<CR>')

eq('DEF2', api.nvim_get_var('var'))

end)

it('errors out on invalid inputs', function()

eq('Vim(call):E730: Using a List as a String', exc_exec('call input([])'))

eq('Vim(call):E730: Using a List as a String', exc_exec('call input("", [])'))

eq('Vim(call):E730: Using a List as a String', exc_exec('call input("", "", [])'))

eq('Vim(call):E730: Using a List as a String', exc_exec('call input({"prompt": []})'))

eq('Vim(call):E730: Using a List as a String', exc_exec('call input({"default": []})'))

eq('Vim(call):E730: Using a List as a String', exc_exec('call input({"completion": []})'))

eq('Vim(call):E5050: {opts} must be the only argument', exc_exec('call input({}, "default")'))

eq(

'Vim(call):E118: Too many arguments for function: input',

exc_exec('call input("prompt> ", "default", "file", "extra")')

)

end)

it('supports highlighting', function()

command('nnoremap <expr> X input({"highlight": "RainBowParens"})[-1]')

feed([[X]])

feed('(())')

screen:expect([[

|

{EOB:~ }|*3

{RBP1:(}{RBP2:()}{RBP1:)}^ |

]])

end)

it('is not hidden by :silent', function()

feed([[:silent call input('Foo: ')<CR>]])

screen:expect([[

|

{EOB:~ }|

{SEP: }|

Foo: ^ |

|

]])

feed('Bar')

screen:expect([[

|

{EOB:~ }|

{SEP: }|

Foo: Bar^ |

|

]])

feed('<CR>')

end)

end)

describe('inputdialog()', function()

it('works with multiline prompts', function()

feed([[:call inputdialog("Test\nFoo")<CR>]])

screen:expect([[

|

{EOB:~ }|

{SEP: }|

Test |

Foo^ |

]])

end)

it('works with multiline prompts and :echohl', function()

feed([[:echohl Test | call inputdialog("Test\nFoo")<CR>]])

screen:expect([[

|

{EOB:~ }|

{SEP: }|

{T:Test} |

{T:Foo}^ |

]])

command('redraw!')

screen:expect([[

|

{EOB:~ }|*3

{T:Foo}^ |

]])

end)

it('allows unequal numeric arguments when using multiple args', function()

command('echohl Test')

feed([[:call inputdialog(1, 2)<CR>]])

screen:expect([[

|

{EOB:~ }|*3

{T:1}2^ |

]])

feed('<BS>')

screen:expect([[

|

{EOB:~ }|*3

{T:1}^ |

]])

end)

it('allows unequal numeric values when using {opts} dict', function()

command('echohl Test')

api.nvim_set_var('opts', { prompt = 1, default = 2, cancelreturn = 3 })

feed([[:echo input(opts)<CR>]])

screen:expect([[

|

{EOB:~ }|*3

{T:1}2^ |

]])

feed('<BS>')

screen:expect([[

|

{EOB:~ }|*3

{T:1}^ |

]])

feed('<Esc>')

screen:expect([[

^ |

{EOB:~ }|*3

{T:3} |

]])

end)

it('works with redraw', function()

command('echohl Test')

api.nvim_set_var('opts', { prompt = 'Foo>', default = 'Bar' })

feed([[:echo input(opts)<CR>]])

screen:expect([[

|

{EOB:~ }|*3

{T:Foo>}Bar^ |

]])

command('mode')

screen:expect {

grid = [[

|

{EOB:~ }|*3

{T:Foo>}Bar^ |

]],

reset = true,

}

feed('<BS>')

screen:expect([[

|

{EOB:~ }|*3

{T:Foo>}Ba^ |

]])

command('mode')

screen:expect {

grid = [[

|

{EOB:~ }|*3

{T:Foo>}Ba^ |

]],

reset = true,

}

end)

it('allows omitting everything with dict argument', function()

command('echohl Test')

feed(':echo inputdialog({})<CR>')

screen:expect([[

|

{EOB:~ }|*3

^ |

]])

end)

it('supports completion', function()

feed(':let var = inputdialog({"completion": "customlist,CustomListCompl"})<CR>')

feed('<Tab><CR>')

eq('FOO', api.nvim_get_var('var'))

end)

it('supports cancelreturn', function()

feed(':let var = inputdialog("", "", "CR1")<CR>')

feed('<Esc>')

eq('CR1', api.nvim_get_var('var'))

feed(':let var = inputdialog({"cancelreturn": "BAR"})<CR>')

feed('<Esc>')

eq('BAR', api.nvim_get_var('var'))

end)

it('supports default string', function()

feed(':let var = inputdialog("", "DEF1")<CR>')

feed('<CR>')

eq('DEF1', api.nvim_get_var('var'))

feed(':let var = inputdialog({"default": "DEF2"})<CR>')

feed('<CR>')

eq('DEF2', api.nvim_get_var('var'))

end)

it('errors out on invalid inputs', function()

eq('Vim(call):E730: Using a List as a String', exc_exec('call inputdialog([])'))

eq('Vim(call):E730: Using a List as a String', exc_exec('call inputdialog("", [])'))

eq('Vim(call):E730: Using a List as a String', exc_exec('call inputdialog("", "", [])'))

eq('Vim(call):E730: Using a List as a String', exc_exec('call inputdialog({"prompt": []})'))

eq('Vim(call):E730: Using a List as a String', exc_exec('call inputdialog({"default": []})'))

eq('Vim(call):E730: Using a List as a String', exc_exec('call inputdialog({"completion": []})'))

eq(

'Vim(call):E5050: {opts} must be the only argument',

exc_exec('call inputdialog({}, "default")')

)

eq(

'Vim(call):E118: Too many arguments for function: inputdialog',

exc_exec('call inputdialog("prompt> ", "default", "file", "extra")')

)

end)

it('supports highlighting', function()

command('nnoremap <expr> X inputdialog({"highlight": "RainBowParens"})[-1]')

feed([[X]])

feed('(())')

screen:expect([[

|

{EOB:~ }|*3

{RBP1:(}{RBP2:()}{RBP1:)}^ |

]])

end)

end)

describe('confirm()', function()

it('works', function()

api.nvim_set_option_value('more', false, {}) -- Avoid hit-enter prompt

api.nvim_set_option_value('laststatus', 2, {})

-- screen:expect() calls are needed to avoid feeding input too early

screen:expect({ any = '%[No Name%]' })

async_meths.nvim_command([[let a = confirm('Press O to proceed')]])

screen:expect({ any = '{CONFIRM:.+: }' })

feed('o')

screen:expect({ any = '%[No Name%]' })

eq(1, api.nvim_get_var('a'))

async_meths.nvim_command([[let a = 'Are you sure?'->confirm("&Yes\n&No")]])

screen:expect({ any = '{CONFIRM:.+: }' })

feed('y')

screen:expect({ any = '%[No Name%]' })

eq(1, api.nvim_get_var('a'))

async_meths.nvim_command([[let a = confirm('Are you sure?', "&Yes\n&No")]])

screen:expect({ any = '{CONFIRM:.+: }' })

feed('n')

screen:expect({ any = '%[No Name%]' })

eq(2, api.nvim_get_var('a'))

-- Not possible to match Vim's CTRL-C test here as CTRL-C always sets got_int in Nvim.

-- confirm() should return 0 when pressing ESC.

async_meths.nvim_command([[let a = confirm('Are you sure?', "&Yes\n&No")]])

screen:expect({ any = '{CONFIRM:.+: }' })

feed('<Esc>')

screen:expect({ any = '%[No Name%]' })

eq(0, api.nvim_get_var('a'))

-- Default choice is returned when pressing <CR>.

async_meths.nvim_command([[let a = confirm('Are you sure?', "&Yes\n&No")]])

screen:expect({ any = '{CONFIRM:.+: }' })

feed('<CR>')

screen:expect({ any = '%[No Name%]' })

eq(1, api.nvim_get_var('a'))

async_meths.nvim_command([[let a = confirm('Are you sure?', "&Yes\n&No", 2)]])

screen:expect({ any = '{CONFIRM:.+: }' })

feed('<CR>')

screen:expect({ any = '%[No Name%]' })

eq(2, api.nvim_get_var('a'))

async_meths.nvim_command([[let a = confirm('Are you sure?', "&Yes\n&No", 0)]])

screen:expect({ any = '{CONFIRM:.+: }' })

feed('<CR>')

screen:expect({ any = '%[No Name%]' })

eq(0, api.nvim_get_var('a'))

-- Test with the {type} 4th argument

for _, type in ipairs({ 'Error', 'Question', 'Info', 'Warning', 'Generic' }) do

async_meths.nvim_command(

([[let a = confirm('Are you sure?', "&Yes\n&No", 1, '%s')]]):format(type)

)

screen:expect({ any = '{CONFIRM:.+: }' })

feed('y')

screen:expect({ any = '%[No Name%]' })

eq(1, api.nvim_get_var('a'))

end

end)

it('shows dialog even if :silent #8788', function()

command("autocmd BufNewFile * call confirm('test')")

local function check_and_clear(edit_line)

end

-- With shortmess-=F

command('set shortmess-=F')

feed(':edit foo<cr>')

check_and_clear('"foo" [New] |\n')

-- With shortmess+=F

command('set shortmess+=F')

feed(':edit foo<cr>')

check_and_clear(':edit foo |\n')

-- With :silent

feed(':silent edit foo<cr>')

check_and_clear(':silent edit foo |\n')

-- With API (via eval/Vimscript) call and shortmess+=F

feed(':call nvim_command("edit x")<cr>')

check_and_clear(':call nvim_command("edit |\n')

async_meths.nvim_command('edit x')

check_and_clear(' |\n')

end)

end)