mirror of
https://github.com/neovim/neovim.git
synced 2025-09-08 20:38:18 +00:00

likely fixes #7768 #7913 If multiple internal stream callbacks were recieved before vimL callbacks got called, only invoke one vimL callback with all data.
870 lines
30 KiB
Lua
870 lines
30 KiB
Lua
local helpers = require('test.functional.helpers')(after_each)
|
|
local clear, eq, eval, exc_exec, feed_command, feed, insert, neq, next_msg, nvim,
|
|
nvim_dir, ok, source, write_file, mkdir, rmdir = helpers.clear,
|
|
helpers.eq, helpers.eval, helpers.exc_exec, helpers.feed_command, helpers.feed,
|
|
helpers.insert, helpers.neq, helpers.next_msg, helpers.nvim,
|
|
helpers.nvim_dir, helpers.ok, helpers.source,
|
|
helpers.write_file, helpers.mkdir, helpers.rmdir
|
|
local command = helpers.command
|
|
local funcs = helpers.funcs
|
|
local retry = helpers.retry
|
|
local meths = helpers.meths
|
|
local NIL = helpers.NIL
|
|
local wait = helpers.wait
|
|
local iswin = helpers.iswin
|
|
local get_pathsep = helpers.get_pathsep
|
|
local pathroot = helpers.pathroot
|
|
local nvim_set = helpers.nvim_set
|
|
local expect_twostreams = helpers.expect_twostreams
|
|
local expect_msg_seq = helpers.expect_msg_seq
|
|
local Screen = require('test.functional.ui.screen')
|
|
|
|
-- Kill process with given pid
|
|
local function os_kill(pid)
|
|
return os.execute((iswin()
|
|
and 'taskkill /f /t /pid '..pid..' > nul'
|
|
or 'kill -9 '..pid..' > /dev/null'))
|
|
end
|
|
|
|
describe('jobs', function()
|
|
local channel
|
|
|
|
before_each(function()
|
|
clear()
|
|
channel = nvim('get_api_info')[1]
|
|
nvim('set_var', 'channel', channel)
|
|
source([[
|
|
function! Normalize(data) abort
|
|
" Windows: remove ^M
|
|
return type([]) == type(a:data)
|
|
\ ? map(a:data, 'substitute(v:val, "\r", "", "g")')
|
|
\ : a:data
|
|
endfunction
|
|
function! OnEvent(id, data, event) dict
|
|
let userdata = get(self, 'user')
|
|
let data = Normalize(a:data)
|
|
call rpcnotify(g:channel, a:event, userdata, data)
|
|
endfunction
|
|
let g:job_opts = {
|
|
\ 'on_stdout': function('OnEvent'),
|
|
\ 'on_exit': function('OnEvent'),
|
|
\ 'user': 0
|
|
\ }
|
|
]])
|
|
end)
|
|
|
|
it('uses &shell and &shellcmdflag if passed a string', function()
|
|
nvim('command', "let $VAR = 'abc'")
|
|
if iswin() then
|
|
nvim('command', "let j = jobstart('echo %VAR%', g:job_opts)")
|
|
else
|
|
nvim('command', "let j = jobstart('echo $VAR', g:job_opts)")
|
|
end
|
|
eq({'notification', 'stdout', {0, {'abc', ''}}}, next_msg())
|
|
eq({'notification', 'stdout', {0, {''}}}, next_msg())
|
|
eq({'notification', 'exit', {0, 0}}, next_msg())
|
|
end)
|
|
|
|
it('changes to given / directory', function()
|
|
nvim('command', "let g:job_opts.cwd = '/'")
|
|
if iswin() then
|
|
nvim('command', "let j = jobstart('cd', g:job_opts)")
|
|
else
|
|
nvim('command', "let j = jobstart('pwd', g:job_opts)")
|
|
end
|
|
eq({'notification', 'stdout',
|
|
{0, {pathroot(), ''}}}, next_msg())
|
|
eq({'notification', 'stdout', {0, {''}}}, next_msg())
|
|
eq({'notification', 'exit', {0, 0}}, next_msg())
|
|
end)
|
|
|
|
it('changes to given `cwd` directory', function()
|
|
local dir = eval("resolve(tempname())"):gsub("/", get_pathsep())
|
|
mkdir(dir)
|
|
nvim('command', "let g:job_opts.cwd = '" .. dir .. "'")
|
|
if iswin() then
|
|
nvim('command', "let j = jobstart('cd', g:job_opts)")
|
|
else
|
|
nvim('command', "let j = jobstart('pwd', g:job_opts)")
|
|
end
|
|
expect_msg_seq(
|
|
{ {'notification', 'stdout', {0, {dir, ''} } },
|
|
{'notification', 'stdout', {0, {''} } },
|
|
{'notification', 'exit', {0, 0} }
|
|
},
|
|
-- Alternative sequence:
|
|
{ {'notification', 'stdout', {0, {dir} } },
|
|
{'notification', 'stdout', {0, {'', ''} } },
|
|
{'notification', 'stdout', {0, {''} } },
|
|
{'notification', 'exit', {0, 0} }
|
|
}
|
|
)
|
|
rmdir(dir)
|
|
end)
|
|
|
|
it('fails to change to invalid `cwd`', function()
|
|
local dir = eval('resolve(tempname())."-bogus"')
|
|
local _, err = pcall(function()
|
|
nvim('command', "let g:job_opts.cwd = '" .. dir .. "'")
|
|
if iswin() then
|
|
nvim('command', "let j = jobstart('cd', g:job_opts)")
|
|
else
|
|
nvim('command', "let j = jobstart('pwd', g:job_opts)")
|
|
end
|
|
end)
|
|
ok(string.find(err, "E475: Invalid argument: expected valid directory$") ~= nil)
|
|
end)
|
|
|
|
it('returns 0 when it fails to start', function()
|
|
eq("", eval("v:errmsg"))
|
|
feed_command("let g:test_jobid = jobstart([])")
|
|
eq(0, eval("g:test_jobid"))
|
|
eq("E474:", string.match(eval("v:errmsg"), "E%d*:"))
|
|
end)
|
|
|
|
it('returns -1 when target is not executable #5465', function()
|
|
local function new_job()
|
|
return eval([[jobstart('')]])
|
|
end
|
|
local executable_jobid = new_job()
|
|
local nonexecutable_jobid = eval("jobstart(['"..(iswin()
|
|
and './test/functional/fixtures'
|
|
or './test/functional/fixtures/non_executable.txt').."'])")
|
|
eq(-1, nonexecutable_jobid)
|
|
-- Should _not_ throw an error.
|
|
eq("", eval("v:errmsg"))
|
|
-- Non-executable job should not increment the job ids. #5465
|
|
eq(executable_jobid + 1, new_job())
|
|
end)
|
|
|
|
it('invokes callbacks when the job writes and exits', function()
|
|
nvim('command', "let g:job_opts.on_stderr = function('OnEvent')")
|
|
nvim('command', [[call jobstart(has('win32') ? 'echo:' : 'echo', g:job_opts)]])
|
|
expect_twostreams({{'notification', 'stdout', {0, {'', ''}}},
|
|
{'notification', 'stdout', {0, {''}}}},
|
|
{{'notification', 'stderr', {0, {''}}}})
|
|
eq({'notification', 'exit', {0, 0}}, next_msg())
|
|
end)
|
|
|
|
it('allows interactive commands', function()
|
|
nvim('command', "let j = jobstart(has('win32') ? ['find', '/v', ''] : ['cat', '-'], g:job_opts)")
|
|
neq(0, eval('j'))
|
|
nvim('command', 'call jobsend(j, "abc\\n")')
|
|
eq({'notification', 'stdout', {0, {'abc', ''}}}, next_msg())
|
|
nvim('command', 'call jobsend(j, "123\\nxyz\\n")')
|
|
expect_msg_seq(
|
|
{ {'notification', 'stdout', {0, {'123', 'xyz', ''}}}
|
|
},
|
|
-- Alternative sequence:
|
|
{ {'notification', 'stdout', {0, {'123', ''}}},
|
|
{'notification', 'stdout', {0, {'xyz', ''}}}
|
|
}
|
|
)
|
|
nvim('command', 'call jobsend(j, [123, "xyz", ""])')
|
|
expect_msg_seq(
|
|
{ {'notification', 'stdout', {0, {'123', 'xyz', ''}}}
|
|
},
|
|
-- Alternative sequence:
|
|
{ {'notification', 'stdout', {0, {'123', ''}}},
|
|
{'notification', 'stdout', {0, {'xyz', ''}}}
|
|
}
|
|
)
|
|
nvim('command', "call jobstop(j)")
|
|
eq({'notification', 'stdout', {0, {''}}}, next_msg())
|
|
eq({'notification', 'exit', {0, 0}}, next_msg())
|
|
end)
|
|
|
|
it('preserves NULs', function()
|
|
if helpers.pending_win32(pending) then return end -- TODO: Need `cat`.
|
|
-- Make a file with NULs in it.
|
|
local filename = helpers.tmpname()
|
|
write_file(filename, "abc\0def\n")
|
|
|
|
nvim('command', "let j = jobstart(['cat', '"..filename.."'], g:job_opts)")
|
|
eq({'notification', 'stdout', {0, {'abc\ndef', ''}}}, next_msg())
|
|
eq({'notification', 'stdout', {0, {''}}}, next_msg())
|
|
eq({'notification', 'exit', {0, 0}}, next_msg())
|
|
os.remove(filename)
|
|
|
|
-- jobsend() preserves NULs.
|
|
nvim('command', "let j = jobstart(['cat', '-'], g:job_opts)")
|
|
nvim('command', [[call jobsend(j, ["123\n456",""])]])
|
|
eq({'notification', 'stdout', {0, {'123\n456', ''}}}, next_msg())
|
|
nvim('command', "call jobstop(j)")
|
|
end)
|
|
|
|
it("will not buffer data if it doesn't end in newlines", function()
|
|
if helpers.pending_win32(pending) then return end -- TODO: Need `cat`.
|
|
if os.getenv("TRAVIS") and os.getenv("CC") == "gcc-4.9"
|
|
and helpers.os_name() == "osx" then
|
|
-- XXX: Hangs Travis macOS since e9061117a5b8f195c3f26a5cb94e18ddd7752d86.
|
|
pending("[Hangs on Travis macOS. #5002]", function() end)
|
|
return
|
|
end
|
|
|
|
nvim('command', "let j = jobstart(['cat', '-'], g:job_opts)")
|
|
nvim('command', 'call jobsend(j, "abc\\nxyz")')
|
|
eq({'notification', 'stdout', {0, {'abc', 'xyz'}}}, next_msg())
|
|
nvim('command', "call jobstop(j)")
|
|
eq({'notification', 'stdout', {0, {''}}}, next_msg())
|
|
eq({'notification', 'exit', {0, 0}}, next_msg())
|
|
end)
|
|
|
|
it('preserves newlines', function()
|
|
if helpers.pending_win32(pending) then return end -- TODO: Need `cat`.
|
|
nvim('command', "let j = jobstart(['cat', '-'], g:job_opts)")
|
|
nvim('command', 'call jobsend(j, "a\\n\\nc\\n\\n\\n\\nb\\n\\n")')
|
|
eq({'notification', 'stdout',
|
|
{0, {'a', '', 'c', '', '', '', 'b', '', ''}}}, next_msg())
|
|
end)
|
|
|
|
it('preserves NULs', function()
|
|
if helpers.pending_win32(pending) then return end -- TODO: Need `cat`.
|
|
nvim('command', "let j = jobstart(['cat', '-'], g:job_opts)")
|
|
nvim('command', 'call jobsend(j, ["\n123\n", "abc\\nxyz\n", ""])')
|
|
eq({'notification', 'stdout', {0, {'\n123\n', 'abc\nxyz\n', ''}}},
|
|
next_msg())
|
|
nvim('command', "call jobstop(j)")
|
|
eq({'notification', 'stdout', {0, {''}}}, next_msg())
|
|
eq({'notification', 'exit', {0, 0}}, next_msg())
|
|
end)
|
|
|
|
it('avoids sending final newline', function()
|
|
if helpers.pending_win32(pending) then return end -- TODO: Need `cat`.
|
|
nvim('command', "let j = jobstart(['cat', '-'], g:job_opts)")
|
|
nvim('command', 'call jobsend(j, ["some data", "without\nfinal nl"])')
|
|
eq({'notification', 'stdout', {0, {'some data', 'without\nfinal nl'}}},
|
|
next_msg())
|
|
nvim('command', "call jobstop(j)")
|
|
eq({'notification', 'stdout', {0, {''}}}, next_msg())
|
|
eq({'notification', 'exit', {0, 0}}, next_msg())
|
|
end)
|
|
|
|
it('closes the job streams with jobclose', function()
|
|
nvim('command', "let j = jobstart(has('win32') ? ['find', '/v', ''] : ['cat', '-'], g:job_opts)")
|
|
nvim('command', 'call jobclose(j, "stdin")')
|
|
eq({'notification', 'stdout', {0, {''}}}, next_msg())
|
|
eq({'notification', 'exit', {0, iswin() and 1 or 0}}, next_msg())
|
|
end)
|
|
|
|
it("disallows jobsend on a job that closed stdin", function()
|
|
nvim('command', "let j = jobstart(has('win32') ? ['find', '/v', ''] : ['cat', '-'], g:job_opts)")
|
|
nvim('command', 'call jobclose(j, "stdin")')
|
|
eq(false, pcall(function()
|
|
nvim('command', 'call jobsend(j, ["some data"])')
|
|
end))
|
|
end)
|
|
|
|
it('disallows jobsend/stop on a non-existent job', function()
|
|
eq(false, pcall(eval, "jobsend(-1, 'lol')"))
|
|
eq(false, pcall(eval, "jobstop(-1)"))
|
|
end)
|
|
|
|
it('disallows jobstop twice on the same job', function()
|
|
nvim('command', "let j = jobstart(has('win32') ? ['find', '/v', ''] : ['cat', '-'], g:job_opts)")
|
|
neq(0, eval('j'))
|
|
eq(true, pcall(eval, "jobstop(j)"))
|
|
eq(false, pcall(eval, "jobstop(j)"))
|
|
end)
|
|
|
|
it('will not leak memory if we leave a job running', function()
|
|
nvim('command', "call jobstart(['cat', '-'], g:job_opts)")
|
|
end)
|
|
|
|
it('can get the pid value using getpid', function()
|
|
nvim('command', "let j = jobstart(has('win32') ? ['find', '/v', ''] : ['cat', '-'], g:job_opts)")
|
|
local pid = eval('jobpid(j)')
|
|
neq(NIL, meths.get_proc(pid))
|
|
nvim('command', 'call jobstop(j)')
|
|
eq({'notification', 'stdout', {0, {''}}}, next_msg())
|
|
if iswin() then
|
|
expect_msg_seq(
|
|
-- win64
|
|
{ {'notification', 'exit', {0, 1}}
|
|
},
|
|
-- win32
|
|
{ {'notification', 'exit', {0, 15}}
|
|
}
|
|
)
|
|
else
|
|
eq({'notification', 'exit', {0, 0}}, next_msg())
|
|
end
|
|
eq(NIL, meths.get_proc(pid))
|
|
end)
|
|
|
|
it("do not survive the exit of nvim", function()
|
|
-- use sleep, which doesn't die on stdin close
|
|
nvim('command', "let g:j = jobstart(has('win32') ? ['ping', '-n', '1001', '127.0.0.1'] : ['sleep', '1000'], g:job_opts)")
|
|
local pid = eval('jobpid(g:j)')
|
|
neq(NIL, meths.get_proc(pid))
|
|
clear()
|
|
eq(NIL, meths.get_proc(pid))
|
|
end)
|
|
|
|
it('can survive the exit of nvim with "detach"', function()
|
|
nvim('command', 'let g:job_opts.detach = 1')
|
|
nvim('command', "let g:j = jobstart(has('win32') ? ['ping', '-n', '1001', '127.0.0.1'] : ['sleep', '1000'], g:job_opts)")
|
|
local pid = eval('jobpid(g:j)')
|
|
neq(NIL, meths.get_proc(pid))
|
|
clear()
|
|
neq(NIL, meths.get_proc(pid))
|
|
-- clean up after ourselves
|
|
eq(0, os_kill(pid))
|
|
end)
|
|
|
|
it('can pass user data to the callback', function()
|
|
nvim('command', 'let g:job_opts.user = {"n": 5, "s": "str", "l": [1]}')
|
|
nvim('command', [[call jobstart('echo foo', g:job_opts)]])
|
|
local data = {n = 5, s = 'str', l = {1}}
|
|
expect_msg_seq(
|
|
{ {'notification', 'stdout', {data, {'foo', ''}}},
|
|
{'notification', 'stdout', {data, {''}}},
|
|
},
|
|
-- Alternative sequence:
|
|
{ {'notification', 'stdout', {data, {'foo'}}},
|
|
{'notification', 'stdout', {data, {'', ''}}},
|
|
{'notification', 'stdout', {data, {''}}},
|
|
}
|
|
)
|
|
eq({'notification', 'exit', {data, 0}}, next_msg())
|
|
end)
|
|
|
|
it('can omit data callbacks', function()
|
|
nvim('command', 'unlet g:job_opts.on_stdout')
|
|
nvim('command', 'let g:job_opts.user = 5')
|
|
nvim('command', [[call jobstart('echo foo', g:job_opts)]])
|
|
eq({'notification', 'exit', {5, 0}}, next_msg())
|
|
end)
|
|
|
|
it('can omit exit callback', function()
|
|
nvim('command', 'unlet g:job_opts.on_exit')
|
|
nvim('command', 'let g:job_opts.user = 5')
|
|
nvim('command', [[call jobstart('echo foo', g:job_opts)]])
|
|
expect_msg_seq(
|
|
{ {'notification', 'stdout', {5, {'foo', ''} } },
|
|
{'notification', 'stdout', {5, {''} } },
|
|
},
|
|
-- Alternative sequence:
|
|
{ {'notification', 'stdout', {5, {'foo'} } },
|
|
{'notification', 'stdout', {5, {'', ''} } },
|
|
{'notification', 'stdout', {5, {''} } },
|
|
}
|
|
)
|
|
end)
|
|
|
|
it('will pass return code with the exit event', function()
|
|
nvim('command', 'let g:job_opts.user = 5')
|
|
nvim('command', "call jobstart('exit 55', g:job_opts)")
|
|
eq({'notification', 'stdout', {5, {''}}}, next_msg())
|
|
eq({'notification', 'exit', {5, 55}}, next_msg())
|
|
end)
|
|
|
|
it('can receive dictionary functions', function()
|
|
source([[
|
|
let g:dict = {'id': 10}
|
|
function g:dict.on_exit(id, code, event)
|
|
call rpcnotify(g:channel, a:event, a:code, self.id)
|
|
endfunction
|
|
call jobstart('exit 45', g:dict)
|
|
]])
|
|
eq({'notification', 'exit', {45, 10}}, next_msg())
|
|
end)
|
|
|
|
it('can redefine callbacks being used by a job', function()
|
|
local screen = Screen.new()
|
|
screen:attach()
|
|
screen:set_default_attr_ids({
|
|
[1] = {bold=true, foreground=Screen.colors.Blue},
|
|
})
|
|
source([[
|
|
function! g:JobHandler(job_id, data, event)
|
|
endfunction
|
|
|
|
let g:callbacks = {
|
|
\ 'on_stdout': function('g:JobHandler'),
|
|
\ 'on_stderr': function('g:JobHandler'),
|
|
\ 'on_exit': function('g:JobHandler')
|
|
\ }
|
|
let job = jobstart(['cat', '-'], g:callbacks)
|
|
]])
|
|
wait()
|
|
source([[
|
|
function! g:JobHandler(job_id, data, event)
|
|
endfunction
|
|
]])
|
|
|
|
eq("", eval("v:errmsg"))
|
|
end)
|
|
|
|
it('requires funcrefs for script-local (s:) functions', function()
|
|
local screen = Screen.new(60, 5)
|
|
screen:attach()
|
|
screen:set_default_attr_ids({
|
|
[1] = {bold = true, foreground = Screen.colors.Blue1},
|
|
[2] = {foreground = Screen.colors.Grey100, background = Screen.colors.Red},
|
|
[3] = {bold = true, foreground = Screen.colors.SeaGreen4}
|
|
})
|
|
|
|
-- Pass job callback names _without_ `function(...)`.
|
|
source([[
|
|
function! s:OnEvent(id, data, event) dict
|
|
let g:job_result = get(self, 'user')
|
|
endfunction
|
|
let s:job = jobstart('echo "foo"', {
|
|
\ 'on_stdout': 's:OnEvent',
|
|
\ 'on_stderr': 's:OnEvent',
|
|
\ 'on_exit': 's:OnEvent',
|
|
\ })
|
|
]])
|
|
|
|
screen:expect("{2:E120: Using <SID> not in a script context: s:OnEvent}",nil,nil,nil,true)
|
|
end)
|
|
|
|
it('does not repeat output with slow output handlers', function()
|
|
source([[
|
|
let d = {'data': []}
|
|
function! d.on_stdout(job, data, event) dict
|
|
call add(self.data, Normalize(a:data))
|
|
sleep 200m
|
|
endfunction
|
|
if has('win32')
|
|
let cmd = 'for /L %I in (1,1,5) do @(echo %I& ping -n 2 127.0.0.1 > nul)'
|
|
else
|
|
let cmd = ['sh', '-c', 'for i in $(seq 1 5); do echo $i; sleep 0.1; done']
|
|
endif
|
|
call jobwait([jobstart(cmd, d)])
|
|
]])
|
|
|
|
local expected = {'1', '2', '3', '4', '5', ''}
|
|
local chunks = eval('d.data')
|
|
local received = {''}
|
|
for i, chunk in ipairs(chunks) do
|
|
if i < #chunks then
|
|
-- if chunks got joined, a spurious [''] callback was not sent
|
|
neq({''}, chunk)
|
|
else
|
|
-- but EOF callback is still sent
|
|
eq({''}, chunk)
|
|
end
|
|
received[#received] = received[#received]..chunk[1]
|
|
for j = 2, #chunk do
|
|
received[#received+1] = chunk[j]
|
|
end
|
|
end
|
|
eq(expected, received)
|
|
end)
|
|
|
|
it('jobstart() works with partial functions', function()
|
|
source([[
|
|
function PrintArgs(a1, a2, id, data, event)
|
|
" Windows: remove ^M
|
|
let normalized = map(a:data, 'substitute(v:val, "\r", "", "g")')
|
|
call rpcnotify(g:channel, '1', a:a1, a:a2, normalized, a:event)
|
|
endfunction
|
|
let Callback = function('PrintArgs', ["foo", "bar"])
|
|
let g:job_opts = {'on_stdout': Callback}
|
|
call jobstart('echo some text', g:job_opts)
|
|
]])
|
|
expect_msg_seq(
|
|
{ {'notification', '1', {'foo', 'bar', {'some text', ''}, 'stdout'}},
|
|
},
|
|
-- Alternative sequence:
|
|
{ {'notification', '1', {'foo', 'bar', {'some text'}, 'stdout'}},
|
|
{'notification', '1', {'foo', 'bar', {'', ''}, 'stdout'}},
|
|
}
|
|
)
|
|
end)
|
|
|
|
it('jobstart() works with closures', function()
|
|
source([[
|
|
fun! MkFun()
|
|
let a1 = 'foo'
|
|
let a2 = 'bar'
|
|
return {id, data, event -> rpcnotify(g:channel, '1', a1, a2, Normalize(data), event)}
|
|
endfun
|
|
let g:job_opts = {'on_stdout': MkFun()}
|
|
call jobstart('echo some text', g:job_opts)
|
|
]])
|
|
expect_msg_seq(
|
|
{ {'notification', '1', {'foo', 'bar', {'some text', ''}, 'stdout'}},
|
|
},
|
|
-- Alternative sequence:
|
|
{ {'notification', '1', {'foo', 'bar', {'some text'}, 'stdout'}},
|
|
{'notification', '1', {'foo', 'bar', {'', ''}, 'stdout'}},
|
|
}
|
|
)
|
|
end)
|
|
|
|
it('jobstart() works when closure passed directly to `jobstart`', function()
|
|
source([[
|
|
let g:job_opts = {'on_stdout': {id, data, event -> rpcnotify(g:channel, '1', 'foo', 'bar', Normalize(data), event)}}
|
|
call jobstart('echo some text', g:job_opts)
|
|
]])
|
|
expect_msg_seq(
|
|
{ {'notification', '1', {'foo', 'bar', {'some text', ''}, 'stdout'}},
|
|
},
|
|
-- Alternative sequence:
|
|
{ {'notification', '1', {'foo', 'bar', {'some text'}, 'stdout'}},
|
|
{'notification', '1', {'foo', 'bar', {'', ''}, 'stdout'}},
|
|
}
|
|
)
|
|
end)
|
|
|
|
describe('jobwait', function()
|
|
before_each(function()
|
|
if iswin() then
|
|
helpers.set_shell_powershell()
|
|
end
|
|
end)
|
|
|
|
it('returns a list of status codes', function()
|
|
source([[
|
|
call rpcnotify(g:channel, 'wait', jobwait(has('win32') ? [
|
|
\ jobstart('Start-Sleep -Milliseconds 100; exit 4'),
|
|
\ jobstart('Start-Sleep -Milliseconds 300; exit 5'),
|
|
\ jobstart('Start-Sleep -Milliseconds 500; exit 6'),
|
|
\ jobstart('Start-Sleep -Milliseconds 700; exit 7')
|
|
\ ] : [
|
|
\ jobstart('sleep 0.10; exit 4'),
|
|
\ jobstart('sleep 0.110; exit 5'),
|
|
\ jobstart('sleep 0.210; exit 6'),
|
|
\ jobstart('sleep 0.310; exit 7')
|
|
\ ]))
|
|
]])
|
|
eq({'notification', 'wait', {{4, 5, 6, 7}}}, next_msg())
|
|
end)
|
|
|
|
it('will run callbacks while waiting', function()
|
|
source([[
|
|
let g:dict = {'id': 10}
|
|
let g:exits = 0
|
|
function g:dict.on_exit(id, code, event)
|
|
if a:code != 5
|
|
throw 'Error!'
|
|
endif
|
|
let g:exits += 1
|
|
endfunction
|
|
call jobwait(has('win32') ? [
|
|
\ jobstart('Start-Sleep -Milliseconds 100; exit 5', g:dict),
|
|
\ jobstart('Start-Sleep -Milliseconds 300; exit 5', g:dict),
|
|
\ jobstart('Start-Sleep -Milliseconds 500; exit 5', g:dict),
|
|
\ jobstart('Start-Sleep -Milliseconds 700; exit 5', g:dict)
|
|
\ ] : [
|
|
\ jobstart('sleep 0.010; exit 5', g:dict),
|
|
\ jobstart('sleep 0.030; exit 5', g:dict),
|
|
\ jobstart('sleep 0.050; exit 5', g:dict),
|
|
\ jobstart('sleep 0.070; exit 5', g:dict)
|
|
\ ])
|
|
call rpcnotify(g:channel, 'wait', g:exits)
|
|
]])
|
|
eq({'notification', 'wait', {4}}, next_msg())
|
|
end)
|
|
|
|
it('will return status codes in the order of passed ids', function()
|
|
source([[
|
|
call rpcnotify(g:channel, 'wait', jobwait(has('win32') ? [
|
|
\ jobstart('Start-Sleep -Milliseconds 700; exit 4'),
|
|
\ jobstart('Start-Sleep -Milliseconds 500; exit 5'),
|
|
\ jobstart('Start-Sleep -Milliseconds 300; exit 6'),
|
|
\ jobstart('Start-Sleep -Milliseconds 100; exit 7')
|
|
\ ] : [
|
|
\ jobstart('sleep 0.070; exit 4'),
|
|
\ jobstart('sleep 0.050; exit 5'),
|
|
\ jobstart('sleep 0.030; exit 6'),
|
|
\ jobstart('sleep 0.010; exit 7')
|
|
\ ]))
|
|
]])
|
|
eq({'notification', 'wait', {{4, 5, 6, 7}}}, next_msg())
|
|
end)
|
|
|
|
it('will return -3 for invalid job ids', function()
|
|
source([[
|
|
call rpcnotify(g:channel, 'wait', jobwait([
|
|
\ -10,
|
|
\ jobstart((has('win32') ? 'Start-Sleep -Milliseconds 100' : 'sleep 0.01').'; exit 5'),
|
|
\ ]))
|
|
]])
|
|
eq({'notification', 'wait', {{-3, 5}}}, next_msg())
|
|
end)
|
|
|
|
it('will return -2 when interrupted without timeout', function()
|
|
feed_command('call rpcnotify(g:channel, "ready") | '..
|
|
'call rpcnotify(g:channel, "wait", '..
|
|
'jobwait([jobstart("'..
|
|
(iswin() and 'Start-Sleep 10' or 'sleep 10')..
|
|
'; exit 55")]))')
|
|
eq({'notification', 'ready', {}}, next_msg())
|
|
feed('<c-c>')
|
|
eq({'notification', 'wait', {{-2}}}, next_msg())
|
|
end)
|
|
|
|
it('will return -2 when interrupted with timeout', function()
|
|
feed_command('call rpcnotify(g:channel, "ready") | '..
|
|
'call rpcnotify(g:channel, "wait", '..
|
|
'jobwait([jobstart("'..
|
|
(iswin() and 'Start-Sleep 10' or 'sleep 10')..
|
|
'; exit 55")], 10000))')
|
|
eq({'notification', 'ready', {}}, next_msg())
|
|
feed('<c-c>')
|
|
eq({'notification', 'wait', {{-2}}}, next_msg())
|
|
end)
|
|
|
|
it('can be called recursively', function()
|
|
if helpers.pending_win32(pending) then return end -- TODO: Need `cat`.
|
|
source([[
|
|
let g:opts = {}
|
|
let g:counter = 0
|
|
function g:opts.on_stdout(id, msg, _event)
|
|
if self.state == 0
|
|
if self.counter < 10
|
|
call Run()
|
|
endif
|
|
let self.state = 1
|
|
call jobsend(a:id, "line1\n")
|
|
elseif self.state == 1
|
|
let self.state = 2
|
|
call jobsend(a:id, "line2\n")
|
|
elseif self.state == 2
|
|
let self.state = 3
|
|
call jobsend(a:id, "line3\n")
|
|
elseif self.state == 3
|
|
let self.state = 4
|
|
call rpcnotify(g:channel, 'w', printf('job %d closed', self.counter))
|
|
call jobclose(a:id, 'stdin')
|
|
endif
|
|
endfunction
|
|
function g:opts.on_exit(...)
|
|
call rpcnotify(g:channel, 'w', printf('job %d exited', self.counter))
|
|
endfunction
|
|
function Run()
|
|
let g:counter += 1
|
|
let j = copy(g:opts)
|
|
let j.state = 0
|
|
let j.counter = g:counter
|
|
call jobwait([
|
|
\ jobstart('echo ready; cat -', j),
|
|
\ ])
|
|
endfunction
|
|
]])
|
|
feed_command('call Run()')
|
|
local r
|
|
for i = 10, 1, -1 do
|
|
r = next_msg()
|
|
eq('job '..i..' closed', r[3][1])
|
|
r = next_msg()
|
|
eq('job '..i..' exited', r[3][1])
|
|
end
|
|
eq(10, nvim('eval', 'g:counter'))
|
|
end)
|
|
|
|
describe('with timeout argument', function()
|
|
it('will return -1 if the wait timed out', function()
|
|
source([[
|
|
call rpcnotify(g:channel, 'wait', jobwait([
|
|
\ jobstart('exit 4'),
|
|
\ jobstart((has('win32') ? 'Start-Sleep 10' : 'sleep 10').'; exit 5'),
|
|
\ ], has('win32') ? 3000 : 100))
|
|
]])
|
|
eq({'notification', 'wait', {{4, -1}}}, next_msg())
|
|
end)
|
|
|
|
it('can pass 0 to check if a job exists', function()
|
|
source([[
|
|
call rpcnotify(g:channel, 'wait', jobwait(has('win32') ? [
|
|
\ jobstart('Start-Sleep -Milliseconds 50; exit 4'),
|
|
\ jobstart('Start-Sleep -Milliseconds 300; exit 5'),
|
|
\ ] : [
|
|
\ jobstart('sleep 0.05; exit 4'),
|
|
\ jobstart('sleep 0.3; exit 5'),
|
|
\ ], 0))
|
|
]])
|
|
eq({'notification', 'wait', {{-1, -1}}}, next_msg())
|
|
end)
|
|
end)
|
|
end)
|
|
|
|
-- FIXME need to wait until jobsend succeeds before calling jobstop
|
|
pending('will only emit the "exit" event after "stdout" and "stderr"', function()
|
|
nvim('command', "let g:job_opts.on_stderr = function('s:OnEvent')")
|
|
nvim('command', "let j = jobstart(['cat', '-'], g:job_opts)")
|
|
local jobid = nvim('eval', 'j')
|
|
nvim('eval', 'jobsend(j, "abcdef")')
|
|
nvim('eval', 'jobstop(j)')
|
|
eq({'notification', 'j', {0, {jobid, 'stdout', {'abcdef'}}}}, next_msg())
|
|
eq({'notification', 'j', {0, {jobid, 'exit'}}}, next_msg())
|
|
end)
|
|
|
|
it('cannot have both rpc and pty options', function()
|
|
command("let g:job_opts.pty = v:true")
|
|
command("let g:job_opts.rpc = v:true")
|
|
local _, err = pcall(command, "let j = jobstart(has('win32') ? ['find', '/v', ''] : ['cat', '-'], g:job_opts)")
|
|
ok(string.find(err, "E475: Invalid argument: job cannot have both 'pty' and 'rpc' options set") ~= nil)
|
|
end)
|
|
|
|
it('does not crash when repeatedly failing to start shell', function()
|
|
source([[
|
|
set shell=nosuchshell
|
|
func! DoIt()
|
|
call jobstart('true')
|
|
call jobstart('true')
|
|
endfunc
|
|
]])
|
|
-- The crash only triggered if both jobs are cleaned up on the same event
|
|
-- loop tick. This is also prevented by try-block, so feed must be used.
|
|
feed_command("call DoIt()")
|
|
feed('<cr>') -- press RETURN
|
|
eq(2,eval('1+1'))
|
|
end)
|
|
|
|
it('jobstop() kills entire process tree #6530', function()
|
|
-- XXX: Using `nvim` isn't a good test, it reaps its children on exit.
|
|
-- local c = 'call jobstart([v:progpath, "-u", "NONE", "-i", "NONE", "--headless"])'
|
|
-- local j = eval("jobstart([v:progpath, '-u', 'NONE', '-i', 'NONE', '--headless', '-c', '"
|
|
-- ..c.."', '-c', '"..c.."'])")
|
|
|
|
-- Create child with several descendants.
|
|
local sleep_cmd = (iswin()
|
|
and 'ping -n 31 127.0.0.1'
|
|
or 'sleep 30')
|
|
local j = eval("jobstart('"..sleep_cmd..' | '..sleep_cmd..' | '..sleep_cmd.."')")
|
|
local ppid = funcs.jobpid(j)
|
|
local children
|
|
retry(nil, nil, function()
|
|
children = meths.get_proc_children(ppid)
|
|
eq(3, #children)
|
|
end)
|
|
-- Assert that nvim_get_proc() sees the children.
|
|
for _, child_pid in ipairs(children) do
|
|
local info = meths.get_proc(child_pid)
|
|
-- eq((iswin() and 'nvim.exe' or 'nvim'), info.name)
|
|
eq(ppid, info.ppid)
|
|
end
|
|
-- Kill the root of the tree.
|
|
funcs.jobstop(j)
|
|
-- Assert that the children were killed.
|
|
retry(nil, nil, function()
|
|
for _, child_pid in ipairs(children) do
|
|
eq(NIL, meths.get_proc(child_pid))
|
|
end
|
|
end)
|
|
end)
|
|
|
|
describe('running tty-test program', function()
|
|
if helpers.pending_win32(pending) then return end
|
|
local function next_chunk()
|
|
local rv
|
|
while true do
|
|
local msg = next_msg()
|
|
local data = msg[3][2]
|
|
for i = 1, #data do
|
|
data[i] = data[i]:gsub('\n', '\000')
|
|
end
|
|
rv = table.concat(data, '\n')
|
|
rv = rv:gsub('\r\n$', ''):gsub('^\r\n', '')
|
|
if rv ~= '' then
|
|
break
|
|
end
|
|
end
|
|
return rv
|
|
end
|
|
|
|
local function send(str)
|
|
nvim('command', 'call jobsend(j, "'..str..'")')
|
|
end
|
|
|
|
before_each(function()
|
|
-- Redefine Normalize() so that TTY data is not munged.
|
|
source([[
|
|
function! Normalize(data) abort
|
|
return a:data
|
|
endfunction
|
|
]])
|
|
local ext = iswin() and '.exe' or ''
|
|
insert(nvim_dir..'/tty-test'..ext) -- Full path to tty-test.
|
|
nvim('command', 'let g:job_opts.pty = 1')
|
|
nvim('command', 'let exec = [expand("<cfile>:p")]')
|
|
nvim('command', "let j = jobstart(exec, g:job_opts)")
|
|
eq('tty ready', next_chunk())
|
|
end)
|
|
|
|
it('echoing input', function()
|
|
send('test')
|
|
eq('test', next_chunk())
|
|
end)
|
|
|
|
it('resizing window', function()
|
|
nvim('command', 'call jobresize(j, 40, 10)')
|
|
eq('rows: 10, cols: 40', next_chunk())
|
|
nvim('command', 'call jobresize(j, 10, 40)')
|
|
eq('rows: 40, cols: 10', next_chunk())
|
|
end)
|
|
|
|
it('jobclose() sends SIGHUP', function()
|
|
nvim('command', 'call jobclose(j)')
|
|
local msg = next_msg()
|
|
msg = (msg[2] == 'stdout') and next_msg() or msg -- Skip stdout, if any.
|
|
eq({'notification', 'exit', {0, 42}}, msg)
|
|
end)
|
|
|
|
it('jobstart() does not keep ptmx file descriptor open', function()
|
|
-- Start another job (using libuv)
|
|
command('let g:job_opts.pty = 0')
|
|
local other_jobid = eval("jobstart(['cat', '-'], g:job_opts)")
|
|
local other_pid = eval('jobpid(' .. other_jobid .. ')')
|
|
|
|
-- Other job doesn't block first job from recieving SIGHUP on jobclose()
|
|
command('call jobclose(j)')
|
|
-- Have to wait so that the SIGHUP can be processed by tty-test on time.
|
|
-- Can't wait for the next message in case this test fails, if it fails
|
|
-- there won't be any more messages, and the test would hang.
|
|
helpers.sleep(100)
|
|
local err = exc_exec('call jobpid(j)')
|
|
eq('Vim(call):E900: Invalid channel id', err)
|
|
|
|
-- cleanup
|
|
eq(other_pid, eval('jobpid(' .. other_jobid .. ')'))
|
|
command('call jobstop(' .. other_jobid .. ')')
|
|
end)
|
|
end)
|
|
end)
|
|
|
|
describe("pty process teardown", function()
|
|
local screen
|
|
before_each(function()
|
|
clear()
|
|
screen = Screen.new(30, 6)
|
|
screen:attach()
|
|
screen:expect([[
|
|
^ |
|
|
~ |
|
|
~ |
|
|
~ |
|
|
~ |
|
|
|
|
|
]])
|
|
end)
|
|
after_each(function()
|
|
screen:detach()
|
|
end)
|
|
|
|
it("does not prevent/delay exit. #4798 #4900", function()
|
|
if helpers.pending_win32(pending) then return end
|
|
-- Use a nested nvim (in :term) to test without --headless.
|
|
feed_command(":terminal '"..helpers.nvim_prog
|
|
.."' -u NONE -i NONE --cmd '"..nvim_set.."' "
|
|
-- Use :term again in the _nested_ nvim to get a PTY process.
|
|
-- Use `sleep` to simulate a long-running child of the PTY.
|
|
.."+terminal +'!(sleep 300 &)' +qa")
|
|
|
|
-- Exiting should terminate all descendants (PTY, its children, ...).
|
|
screen:expect([[
|
|
^ |
|
|
[Process exited 0] |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
]])
|
|
end)
|
|
end)
|