rubysl-socket 0.0.1 → 1.0.0

data/spec/udpsocket/recvfrom_nonblock_spec.rb

@@ -0,0 +1 @@
+require File.expand_path('../../fixtures/classes', __FILE__)

data/spec/udpsocket/send_spec.rb

@@ -0,0 +1,57 @@
+require File.expand_path('../../fixtures/classes', __FILE__)
+
+describe "UDPSocket.send" do
+  before :each do
+    @ready = false
+    @server_thread = Thread.new do
+      @server = UDPSocket.open
+      @server.bind(nil, SocketSpecs.port)
+      @ready = true
+      begin
+        @msg = @server.recvfrom_nonblock(64)
+      rescue Errno::EAGAIN
+        IO.select([@server])
+        retry
+      end
+      @server.close
+    end
+    Thread.pass while @server_thread.status and !@ready
+  end
+
+  it "sends data in ad hoc mode" do
+    @socket = UDPSocket.open
+    @socket.send("ad hoc", 0, SocketSpecs.hostname,SocketSpecs.port)
+    @socket.close
+    @server_thread.join
+
+    @msg[0].should == "ad hoc"
+    @msg[1][0].should == "AF_INET"
+    @msg[1][1].should be_kind_of(Fixnum)
+    @msg[1][3].should == "127.0.0.1"
+  end
+
+  it "sends data in ad hoc mode (with port given as a String)" do
+    @socket = UDPSocket.open
+    @socket.send("ad hoc", 0, SocketSpecs.hostname,SocketSpecs.str_port)
+    @socket.close
+    @server_thread.join
+
+    @msg[0].should == "ad hoc"
+    @msg[1][0].should == "AF_INET"
+    @msg[1][1].should be_kind_of(Fixnum)
+    @msg[1][3].should == "127.0.0.1"
+  end
+
+  it "sends data in connection mode" do
+    @socket = UDPSocket.open
+    @socket.connect(SocketSpecs.hostname,SocketSpecs.port)
+    @socket.send("connection-based", 0)
+    @socket.close
+    @server_thread.join
+
+    @msg[0].should == "connection-based"
+    @msg[1][0].should == "AF_INET"
+    @msg[1][1].should be_kind_of(Fixnum)
+    @msg[1][3].should == "127.0.0.1"
+  end
+end

data/spec/unixserver/accept_nonblock_spec.rb

@@ -0,0 +1,33 @@
+require File.expand_path('../../fixtures/classes', __FILE__)
+
+describe "UNIXServer#accept_nonblock" do
+
+  platform_is_not :windows do
+    before :each do
+      @path = SocketSpecs.socket_path
+      rm_r @path
+
+      @server = UNIXServer.open(@path)
+      @client = UNIXSocket.open(@path)
+
+      @socket = @server.accept_nonblock
+      @client.send("foobar", 0)
+    end
+
+    after :each do
+      @socket.close
+      @client.close
+      @server.close
+      rm_r @path
+    end
+
+    it "accepts a connection in a non-blocking way" do
+      data = @socket.recvfrom(6).first
+      data.should == "foobar"
+    end
+
+    it "returns a UNIXSocket" do
+      @socket.should be_kind_of(UNIXSocket)
+    end
+  end
+end

data/spec/unixserver/accept_spec.rb

@@ -0,0 +1,64 @@
+require File.expand_path('../../fixtures/classes', __FILE__)
+
+platform_is_not :windows do
+  describe "UNIXServer#accept" do
+    before :each do
+      @path = SocketSpecs.socket_path
+      rm_r @path
+    end
+
+    after :each do
+      rm_r @path
+    end
+
+    it "accepts what is written by the client" do
+      server = UNIXServer.open(SocketSpecs.socket_path)
+      client = UNIXSocket.open(SocketSpecs.socket_path)
+
+      client.send('hello', 0)
+
+      sock = server.accept
+      data, info = sock.recvfrom(5)
+
+      data.should == 'hello'
+
+      server.close
+      client.close
+      sock.close
+    end
+
+    it "can be interrupted by Thread#kill" do
+      server = UNIXServer.new(@path)
+      t = Thread.new {
+        server.accept
+      }
+      Thread.pass while t.status and t.status != "sleep"
+
+      # kill thread, ensure it dies in a reasonable amount of time
+      t.kill
+      a = 1
+      while a < 2000
+        break unless t.alive?
+        Thread.pass
+        sleep 0.2
+        a += 1
+      end
+      a.should < 2000
+      server.close
+    end
+
+    it "can be interrupted by Thread#raise" do
+      server = UNIXServer.new(@path)
+      t = Thread.new {
+        server.accept
+      }
+      Thread.pass while t.status and t.status != "sleep"
+
+      # raise in thread, ensure the raise happens
+      ex = Exception.new
+      t.raise ex
+      lambda { t.join }.should raise_error(Exception)
+      server.close
+    end
+  end
+end

data/spec/unixserver/for_fd_spec.rb

@@ -0,0 +1,32 @@
+require File.expand_path('../../fixtures/classes', __FILE__)
+
+describe "UNIXServer#for_fd" do
+  before :each do
+    @unix_path = tmp("unix_socket")
+    @unix = UNIXServer.new(@unix_path)
+  end
+
+  after :each do
+    # UG. We can't use the new_fd helper, because we need fds that are
+    # associated with sockets. But for_fd has the same issue as IO#new, it
+    # creates a fd aliasing issue with closing, causing EBADF errors.
+    #
+    # Thusly, the rescue for EBADF here. I'd love a better solution, but
+    # I'm not aware of one.
+
+    begin
+      @unix.close unless @unix.closed?
+    rescue Errno::EBADF
+      # I hate this API too
+    end
+
+    rm_r @unix_path
+  end
+
+  it "can calculate the path" do
+    b = UNIXServer.for_fd(@unix.fileno)
+
+    b.path.should == @unix_path
+    b.close
+  end
+end

data/spec/unixserver/new_spec.rb

@@ -0,0 +1,5 @@
+require File.expand_path('../shared/new', __FILE__)
+
+describe "UNIXServer.new" do
+  it_behaves_like :unixserver_new, :new
+end

data/spec/unixserver/open_spec.rb

@@ -0,0 +1,25 @@
+require File.expand_path('../shared/new', __FILE__)
+
+describe "UNIXServer.open" do
+  it_behaves_like :unixserver_new, :open
+
+  platform_is_not :windows do
+    before :each do
+      @path = tmp("unixserver_spec")
+      rm_r @path
+    end
+
+    after :each do
+      @server.close if @server
+      @server = nil
+      rm_r @path
+    end
+
+    it "yields the new UNIXServer object to the block, if given" do
+      UNIXServer.open(@path) do |unix|
+	unix.path.should == @path
+	unix.addr.should == ["AF_UNIX", @path]
+      end
+    end 
+  end
+end

data/spec/unixserver/shared/new.rb

@@ -0,0 +1,23 @@
+require File.expand_path('../../../fixtures/classes', __FILE__)
+require 'tempfile'
+
+describe :unixserver_new, :shared => true do
+  platform_is_not :windows do
+    before :each do
+      @path = tmp("unixserver_spec")
+      rm_r @path
+    end
+
+    after :each do
+      @server.close if @server
+      @server = nil
+      rm_r @path
+    end
+
+    it "creates a new UNIXServer" do
+      @server = UNIXServer.send(@method, @path)
+      @server.path.should == @path
+      @server.addr.should == ["AF_UNIX", @path]
+    end
+  end
+end

data/spec/unixsocket/addr_spec.rb

@@ -0,0 +1,37 @@
+require File.expand_path('../../fixtures/classes', __FILE__)
+
+describe "UNIXSocket#addr" do
+
+  platform_is_not :windows do
+    before :each do
+      @path = SocketSpecs.socket_path
+      rm_r @path
+
+      @server = UNIXServer.open(@path)
+      @client = UNIXSocket.open(@path)
+    end
+
+    after :each do
+      @client.close
+      @server.close
+      rm_r @path
+    end
+
+    it "returns the address family of this socket in an array" do
+      @client.addr[0].should == "AF_UNIX"
+    end
+
+    it "returns the path of the socket in an array if it's a server" do
+      @server.addr[1].should == @path
+    end
+
+    it "returns an empty string for path if it's a client" do
+      @client.addr[1].should == ""
+    end
+
+    it "returns an array" do
+      @client.addr.should be_kind_of(Array)
+    end
+  end
+
+end

data/spec/unixsocket/new_spec.rb

@@ -0,0 +1,5 @@
+require File.expand_path('../shared/new', __FILE__)
+
+describe "UNIXSocket.new" do
+  it_behaves_like :unixsocket_new, :new
+end

data/spec/unixsocket/open_spec.rb

@@ -0,0 +1,26 @@
+require File.expand_path('../shared/new', __FILE__)
+
+describe "UNIXSocket.open" do
+  it_behaves_like :unixsocket_new, :open
+
+  platform_is_not :windows do
+    before :each do
+      @path = SocketSpecs.socket_path
+      rm_r @path
+
+      @server = UNIXServer.open(@path)
+    end
+
+    after :each do
+      @server.close
+      rm_r @path
+    end
+
+    it "opens a unix socket on the specified file and yields it to the block" do
+      UNIXSocket.send(@method, @path) do |client|
+        client.addr[0].should == "AF_UNIX"
+        client.closed?.should == false
+      end
+    end
+  end
+end

data/spec/unixsocket/pair_spec.rb

@@ -0,0 +1,38 @@
+require File.expand_path('../../fixtures/classes', __FILE__)
+require File.expand_path('../../shared/partially_closable_sockets', __FILE__)
+
+describe "UNIXSocket#pair" do
+  platform_is_not :windows do
+
+    it_should_behave_like "partially closable sockets"
+
+    before :each do
+      @s1, @s2 = UNIXSocket.pair
+    end
+
+    after :each do
+      @s1.close
+      @s2.close
+    end
+
+    it "returns a pair of connected sockets" do
+      @s1.puts "foo"
+      @s2.gets.should == "foo\n"
+    end
+
+    it "returns sockets with no name" do
+      @s1.path.should == @s2.path
+      @s1.path.should == ""
+    end
+
+    it "returns sockets with no address" do
+      @s1.addr.should == ["AF_UNIX", ""]
+      @s2.addr.should == ["AF_UNIX", ""]
+    end
+
+    it "returns sockets with no peeraddr" do
+      @s1.peeraddr.should == ["AF_UNIX", ""]
+      @s2.peeraddr.should == ["AF_UNIX", ""]
+    end
+  end
+end

data/spec/unixsocket/partially_closable_spec.rb

@@ -0,0 +1,25 @@
+require File.expand_path('../../fixtures/classes', __FILE__)
+require File.expand_path('../../shared/partially_closable_sockets', __FILE__)
+
+platform_is_not :windows do
+  describe "UNIXSocket partial closability" do
+
+    before :each do
+      @path = SocketSpecs.socket_path
+      rm_r @path
+      @server = UNIXServer.open(@path)
+      @s1 = UNIXSocket.new(@path)
+      @s2 = @server.accept
+    end
+
+    after :each do
+      @server.close
+      @s1.close
+      @s2.close
+      rm_r @path
+    end
+
+    it_should_behave_like "partially closable sockets"
+
+  end
+end

data/spec/unixsocket/path_spec.rb

@@ -0,0 +1,29 @@
+require File.expand_path('../../fixtures/classes', __FILE__)
+
+describe "UNIXSocket#path" do
+
+  platform_is_not :windows do
+    before :each do
+      @path = SocketSpecs.socket_path
+      rm_r @path
+
+      @server = UNIXServer.open(@path)
+      @client = UNIXSocket.open(@path)
+    end
+
+    after :each do
+      @client.close
+      @server.close
+      rm_r @path
+    end
+
+    it "returns the path of the socket if it's a server" do
+      @server.path.should == @path
+    end
+
+    it "returns an empty string for path if it's a client" do
+      @client.path.should == ""
+    end
+  end
+
+end

data/spec/unixsocket/peeraddr_spec.rb

@@ -0,0 +1,29 @@
+require File.expand_path('../../fixtures/classes', __FILE__)
+
+describe "UNIXSocket#peeraddr" do
+
+  platform_is_not :windows do
+    before :each do
+      @path = SocketSpecs.socket_path
+      rm_r @path
+
+      @server = UNIXServer.open(@path)
+      @client = UNIXSocket.open(@path)
+    end
+
+    after :each do
+      @client.close
+      @server.close
+      rm_r @path
+    end
+
+    it "returns the address familly and path of the server end of the connection" do
+      @client.peeraddr.should == ["AF_UNIX", @path]
+    end
+
+    it "raises an error in server sockets" do
+      lambda { @server.peeraddr }.should raise_error
+    end
+  end
+
+end