1 files changed, 155 insertions, 0 deletions
diff --git a/spec/ruby/library/socket/basicsocket/shutdown_spec.rb b/spec/ruby/library/socket/basicsocket/shutdown_spec.rb
new file mode 100644
index 0000000000..c96c62abe1
--- /dev/null
+++ b/spec/ruby/library/socket/basicsocket/shutdown_spec.rb
@@ -0,0 +1,155 @@
+require_relative '../spec_helper'
+require_relative '../fixtures/classes'
+
+platform_is_not :windows do # hangs
+  describe "Socket::BasicSocket#shutdown" do
+    SocketSpecs.each_ip_protocol do |family, ip_address|
+      before do
+        @server = Socket.new(family, :STREAM)
+        @client = Socket.new(family, :STREAM)
+
+        @server.bind(Socket.sockaddr_in(0, ip_address))
+        @server.listen(1)
+
+        @client.connect(@server.getsockname)
+      end
+
+      after do
+        @client.close
+        @server.close
+      end
+
+      describe 'using an Integer' do
+        it 'shuts down a socket for reading' do
+          @client.shutdown(Socket::SHUT_RD)
+
+          @client.recv(1).to_s.should.empty?
+        end
+
+        it 'shuts down a socket for writing' do
+          @client.shutdown(Socket::SHUT_WR)
+
+          -> { @client.write('hello') }.should.raise(Errno::EPIPE)
+        end
+
+        it 'shuts down a socket for reading and writing' do
+          @client.shutdown(Socket::SHUT_RDWR)
+
+          @client.recv(1).to_s.should.empty?
+
+          -> { @client.write('hello') }.should.raise(Errno::EPIPE)
+        end
+
+        it 'raises ArgumentError when using an invalid option' do
+          -> { @server.shutdown(666) }.should.raise(ArgumentError)
+        end
+      end
+
+      describe 'using a Symbol' do
+        it 'shuts down a socket for reading using :RD' do
+          @client.shutdown(:RD)
+
+          @client.recv(1).to_s.should.empty?
+        end
+
+        it 'shuts down a socket for reading using :SHUT_RD' do
+          @client.shutdown(:SHUT_RD)
+
+          @client.recv(1).to_s.should.empty?
+        end
+
+        it 'shuts down a socket for writing using :WR' do
+          @client.shutdown(:WR)
+
+          -> { @client.write('hello') }.should.raise(Errno::EPIPE)
+        end
+
+        it 'shuts down a socket for writing using :SHUT_WR' do
+          @client.shutdown(:SHUT_WR)
+
+          -> { @client.write('hello') }.should.raise(Errno::EPIPE)
+        end
+
+        it 'shuts down a socket for reading and writing' do
+          @client.shutdown(:RDWR)
+
+          @client.recv(1).to_s.should.empty?
+
+          -> { @client.write('hello') }.should.raise(Errno::EPIPE)
+        end
+
+        it 'raises ArgumentError when using an invalid option' do
+          -> { @server.shutdown(:Nope) }.should.raise(SocketError)
+        end
+      end
+
+      describe 'using a String' do
+        it 'shuts down a socket for reading using "RD"' do
+          @client.shutdown('RD')
+
+          @client.recv(1).to_s.should.empty?
+        end
+
+        it 'shuts down a socket for reading using "SHUT_RD"' do
+          @client.shutdown('SHUT_RD')
+
+          @client.recv(1).to_s.should.empty?
+        end
+
+        it 'shuts down a socket for writing using "WR"' do
+          @client.shutdown('WR')
+
+          -> { @client.write('hello') }.should.raise(Errno::EPIPE)
+        end
+
+        it 'shuts down a socket for writing using "SHUT_WR"' do
+          @client.shutdown('SHUT_WR')
+
+          -> { @client.write('hello') }.should.raise(Errno::EPIPE)
+        end
+
+        it 'raises ArgumentError when using an invalid option' do
+          -> { @server.shutdown('Nope') }.should.raise(SocketError)
+        end
+      end
+
+      describe 'using an object that responds to #to_str' do
+        before do
+          @dummy = mock(:dummy)
+        end
+
+        it 'shuts down a socket for reading using "RD"' do
+          @dummy.stub!(:to_str).and_return('RD')
+
+          @client.shutdown(@dummy)
+
+          @client.recv(1).to_s.should.empty?
+        end
+
+        it 'shuts down a socket for reading using "SHUT_RD"' do
+          @dummy.stub!(:to_str).and_return('SHUT_RD')
+
+          @client.shutdown(@dummy)
+
+          @client.recv(1).to_s.should.empty?
+        end
+
+        it 'shuts down a socket for reading and writing' do
+          @dummy.stub!(:to_str).and_return('RDWR')
+
+          @client.shutdown(@dummy)
+
+          @client.recv(1).to_s.should.empty?
+
+          -> { @client.write('hello') }.should.raise(Errno::EPIPE)
+        end
+      end
+
+      describe 'using an object that does not respond to #to_str' do
+        it 'raises TypeError' do
+          -> { @server.shutdown(mock(:dummy)) }.should.raise(TypeError)
+        end
+      end
+    end
+  end
+end