feat(ChildProcessStream): capture exit code on exit

Author: vigoux

nvim/child_process_stream.lua

@@ -19,7 +19,9 @@ function ChildProcessStream.spawn(argv, env, io_extra)
     stdio = {self._child_stdin, self._child_stdout, 2, io_extra},
     args = args,
     env = env,
-  }, function()
+  }, function(ecode, signal)
+    self._exit_code = ecode
+    self._signal = signal
     self:close()
   end)
 

test/session_spec.lua

@@ -4,6 +4,7 @@ local SocketStream = require('nvim.socket_stream')
 local Session = require('nvim.session')
 local coxpcall = require('coxpcall')
 local busted = require('busted')
+local uv = require('luv')
 require('nvim._compat')
 
 local nvim_prog = os.getenv('NVIM_PROG') or 'nvim'
@@ -168,6 +169,38 @@ test_session("Session using ChidProcessStream", function ()
   return Session.new(proc_stream)
 end)
 
+describe("ChildProcessStream", function()
+  local session, proc_stream
+
+  before_each(function()
+    proc_stream = ChildProcessStream.spawn({
+      nvim_prog, '--clean', '--embed',
+    })
+    session = Session.new(proc_stream)
+    session:request "nvim_get_api_info"
+  end)
+
+  after_each(function()
+    session:close()
+    proc_stream:close()
+  end)
+
+  it("can capture exit code", function()
+    session:request("nvim_command", "qall!")
+    uv.run()
+
+    assert.is.same(proc_stream._exit_code, 0)
+    assert.is.same(proc_stream._signal, 0)
+  end)
+
+  it("can can capture signal", function()
+    session:close "kill"
+
+    assert.is.same(proc_stream._exit_code, 0)
+    assert.is.same(proc_stream._signal, 9)
+  end)
+end)
+
 -- Session using SocketStream
 test_session(string.format("Session using SocketStream [%s]", socket_file), function ()
   child_session = Session.new(ChildProcessStream.spawn({