1 files changed, 85 insertions, 0 deletions
diff --git a/spec/ruby/library/socket/tcpserver/accept_nonblock_spec.rb b/spec/ruby/library/socket/tcpserver/accept_nonblock_spec.rb
new file mode 100644
index 0000000000..ac08fe37c6
--- /dev/null
+++ b/spec/ruby/library/socket/tcpserver/accept_nonblock_spec.rb
@@ -0,0 +1,85 @@
+require_relative '../spec_helper'
+require_relative '../fixtures/classes'
+
+describe "Socket::TCPServer.accept_nonblock" do
+  before :each do
+    @server = TCPServer.new("127.0.0.1", 0)
+    @port = @server.addr[1]
+  end
+
+  after :each do
+    @server.close unless @server.closed?
+  end
+
+  it "accepts non blocking connections" do
+    @server.listen(5)
+    -> {
+      @server.accept_nonblock
+    }.should.raise(IO::WaitReadable)
+
+    c = TCPSocket.new("127.0.0.1", @port)
+    IO.select([@server])
+    s = @server.accept_nonblock
+
+    port, address = Socket.unpack_sockaddr_in(s.getsockname)
+
+    port.should == @port
+    address.should == "127.0.0.1"
+    s.should.is_a?(TCPSocket)
+
+    c.close
+    s.close
+  end
+
+  it "raises an IOError if the socket is closed" do
+    @server.close
+    -> { @server.accept }.should.raise(IOError)
+  end
+
+  describe 'without a connected client' do
+    it 'raises error' do
+      -> { @server.accept_nonblock }.should.raise(IO::WaitReadable)
+    end
+
+    it 'returns :wait_readable in exceptionless mode' do
+      @server.accept_nonblock(exception: false).should == :wait_readable
+    end
+  end
+end
+
+describe 'TCPServer#accept_nonblock' do
+  SocketSpecs.each_ip_protocol do |family, ip_address|
+    before do
+      @server = TCPServer.new(ip_address, 0)
+    end
+
+    after do
+      @server.close
+    end
+
+    describe 'without a connected client' do
+      it 'raises IO::WaitReadable' do
+        -> { @server.accept_nonblock }.should.raise(IO::WaitReadable)
+      end
+    end
+
+    platform_is_not :windows do # spurious
+      describe 'with a connected client' do
+        before do
+          @client = TCPSocket.new(ip_address, @server.connect_address.ip_port)
+        end
+
+        after do
+          @socket.close if @socket
+          @client.close
+        end
+
+        it 'returns a TCPSocket' do
+          IO.select([@server])
+          @socket = @server.accept_nonblock
+          @socket.should.instance_of?(TCPSocket)
+        end
+      end
+    end
+  end
+end