1 files changed, 45 insertions, 0 deletions
diff --git a/spec/ruby/core/process/wait2_spec.rb b/spec/ruby/core/process/wait2_spec.rb
new file mode 100644
index 0000000000..8ba429dc96
--- /dev/null
+++ b/spec/ruby/core/process/wait2_spec.rb
@@ -0,0 +1,45 @@
+require_relative '../../spec_helper'
+
+describe "Process.wait2" do
+  before :all do
+    # HACK: this kludge is temporarily necessary because some
+    # misbehaving spec somewhere else does not clear processes
+    # Note: background processes are unavoidable with RJIT,
+    # but we shouldn't reap them from Ruby-space
+    begin
+      Process.wait(-1, Process::WNOHANG)
+      $stderr.puts "Leaked process before wait2 specs! Waiting for it"
+      leaked = Process.waitall
+      $stderr.puts "leaked before wait2 specs: #{leaked}" unless leaked.empty?
+      # Ruby-space should not see PIDs used by rjit
+      leaked.should be_empty
+    rescue Errno::ECHILD # No child processes
+    rescue NotImplementedError
+    end
+  end
+
+  platform_is_not :windows do
+    it "returns the pid and status of child process" do
+      pidf = Process.fork { Process.exit! 99 }
+      results = Process.wait2
+      results.size.should == 2
+      pidw, status = results
+      pidf.should == pidw
+      status.exitstatus.should == 99
+    end
+  end
+
+  it "raises a StandardError if no child processes exist" do
+    -> { Process.wait2 }.should raise_error(Errno::ECHILD)
+    -> { Process.wait2 }.should raise_error(StandardError)
+  end
+
+  it "returns nil if the child process is still running when given the WNOHANG flag" do
+    IO.popen(ruby_cmd('STDIN.getbyte'), "w") do |io|
+      pid, status = Process.wait2(io.pid, Process::WNOHANG)
+      pid.should be_nil
+      status.should be_nil
+      io.write('a')
+    end
+  end
+end