ffi-rzmq 0.8.2 → 0.9.0

data/spec/multipart_spec.rb

@@ -0,0 +1,162 @@
+require File.join(File.dirname(__FILE__), %w[spec_helper])
+
+module ZMQ
+  describe Socket do
+    context "multipart messages" do
+      before(:all) { @ctx = Context.new }
+      after(:all) { @ctx.terminate }
+
+      context "without identity" do
+        include APIHelper
+
+        before(:all) do
+          @rep = Socket.new(@ctx.pointer, ZMQ::REP)
+          port = bind_to_random_tcp_port(@rep)
+
+          @req = Socket.new(@ctx.pointer, ZMQ::REQ)
+          @req.connect("tcp://127.0.0.1:#{port}")
+        end
+
+        after(:all) do
+          @req.close
+          @rep.close
+        end
+
+        it "should be delivered between REQ and REP returning an array of strings" do
+          req_data, rep_data = [ "1", "2" ], [ "2", "3" ]
+
+          @req.send_strings(req_data)
+          strings = []
+          rc = @rep.recv_strings(strings)
+          strings.should == req_data
+
+          @rep.send_strings(rep_data)
+          strings = []
+          rc = @req.recv_strings(strings)
+          strings.should == rep_data
+        end
+
+        it "should be delivered between REQ and REP returning an array of messages" do
+          req_data, rep_data = [ "1", "2" ], [ "2", "3" ]
+
+          @req.send_strings(req_data)
+          messages = []
+          rc = @rep.recvmsgs(messages)
+          messages.each_with_index do |message, index|
+            message.copy_out_string.should == req_data[index]
+          end
+
+          @rep.send_strings(rep_data)
+          messages = []
+          rc = @req.recvmsgs(messages)
+          messages.each_with_index do |message, index|
+            message.copy_out_string.should == rep_data[index]
+          end
+        end
+      end
+
+      if version2?
+        context "with identity" do
+          include APIHelper
+
+          before(:each) do # was :all
+            @rep = Socket.new(@ctx.pointer, ZMQ::XREP)
+            port = bind_to_random_tcp_port(@rep)
+
+            @req = Socket.new(@ctx.pointer, ZMQ::REQ)
+            @req.identity = 'foo'
+            @req.connect("tcp://127.0.0.1:#{port}")
+          end
+
+          after(:each) do # was :all
+            @req.close
+            @rep.close
+          end
+
+          it "should be delivered between REQ and REP returning an array of strings with an empty string as the envelope delimiter" do
+            req_data, rep_data = "hello", [ @req.identity, "", "ok" ]
+
+            @req.send_string(req_data)
+            strings = []
+            rc = @rep.recv_strings(strings)
+            strings.should == [ @req.identity, "", "hello" ]
+
+            @rep.send_strings(rep_data)
+            string = ''
+            rc = @req.recv_string(string)
+            string.should == rep_data.last
+          end
+
+          it "should be delivered between REQ and REP returning an array of messages with an empty string as the envelope delimiter" do
+            req_data, rep_data = "hello", [ @req.identity, "", "ok" ]
+
+            @req.send_string(req_data)
+            msgs = []
+            rc = @rep.recvmsgs(msgs)
+            msgs[0].copy_out_string.should == @req.identity
+            msgs[1].copy_out_string.should == ""
+            msgs[2].copy_out_string.should == "hello"
+
+            @rep.send_strings(rep_data)
+            msgs = []
+            rc = @req.recvmsgs(msgs)
+            msgs[0].copy_out_string.should == rep_data.last
+          end
+        end
+
+      elsif LibZMQ.version3? # version3
+
+        context "with identity" do
+          include APIHelper
+
+          before(:each) do # was :all
+            @rep = Socket.new(@ctx.pointer, ZMQ::ROUTER)
+            port = bind_to_random_tcp_port(@rep)
+
+            @req = Socket.new(@ctx.pointer, ZMQ::DEALER)
+            @req.identity = 'foo'
+            @req.connect("tcp://127.0.0.1:#{port}")
+          end
+
+          after(:each) do # was :all
+            @req.close
+            @rep.close
+          end
+
+          it "should be delivered between ROUTER and DEALER returning an array of strings" do
+            req_data, rep_data = "hello", [ @req.identity, "ok" ]
+
+            @req.send_string(req_data)
+            strings = []
+            rc = @rep.recv_strings(strings)
+            strings.should == [ @req.identity, "hello" ]
+
+            @rep.send_strings(rep_data)
+            string = ''
+            rc = @req.recv_string(string)
+            string.should == rep_data.last
+          end
+
+          it "should be delivered between ROUTER and DEALER returning an array of messages" do
+            req_data, rep_data = "hello", [ @req.identity, "ok" ]
+
+            @req.send_string(req_data)
+            msgs = []
+            rc = @rep.recvmsgs(msgs)
+            msgs[0].copy_out_string.should == @req.identity
+            msgs[1].copy_out_string.should == "hello"
+
+            @rep.send_strings(rep_data)
+            msgs = []
+            rc = @req.recvmsgs(msgs)
+            msgs[0].copy_out_string.should == rep_data.last
+          end
+
+        end
+
+      end # if version...
+
+
+    end
+  end
+end

data/spec/nonblocking_recv_spec.rb

@@ -0,0 +1,325 @@
+
+require File.join(File.dirname(__FILE__), %w[spec_helper])
+
+module ZMQ
+
+
+  describe Socket do
+    before(:all) do
+      @ctx = Context.new
+    end
+
+    after(:all) do
+      @ctx.terminate
+    end
+
+
+    shared_examples_for "any socket" do
+
+      it "returns -1 and gets EAGAIN when there are no messages to read" do
+        array = []
+        rc = @receiver.recvmsgs(array, NonBlockingFlag)
+        Util.resultcode_ok?(rc).should be_false
+        array.size.should == 0
+        ZMQ::Util.errno.should == ZMQ::EAGAIN
+      end
+
+    end
+
+    shared_examples_for "sockets without exposed envelopes" do
+
+      it "read the single message and returns a successful result code" do
+        rc = @sender.send_string('test')
+        Util.resultcode_ok?(rc).should be_true
+        sleep 0.1 # give it time to deliver to the receiver
+
+        array = []
+        rc = @receiver.recvmsgs(array, NonBlockingFlag)
+        Util.resultcode_ok?(rc).should be_true
+        array.size.should == 1
+      end
+
+      it "read all message parts transmitted and returns a successful result code" do
+        strings = Array.new(10, 'test')
+        rc = @sender.send_strings(strings)
+        Util.resultcode_ok?(rc).should be_true
+        sleep 0.1 # give it time to deliver to the sub socket
+
+        array = []
+        rc = @receiver.recvmsgs(array, NonBlockingFlag)
+        Util.resultcode_ok?(rc).should be_true
+        array.size.should == 10
+      end
+
+    end
+
+    shared_examples_for "sockets with exposed envelopes" do
+
+      it "read the single message and returns a successful result code" do
+        rc = @sender.send_string('test')
+        Util.resultcode_ok?(rc).should be_true
+        sleep 0.1 # give it time to deliver to the receiver
+
+        array = []
+        rc = @receiver.recvmsgs(array, NonBlockingFlag)
+        Util.resultcode_ok?(rc).should be_true
+        array.size.should == 1 + 1 # extra 1 for envelope
+      end
+
+      it "read all message parts transmitted and returns a successful result code" do
+        strings = Array.new(10, 'test')
+        rc = @sender.send_strings(strings)
+        Util.resultcode_ok?(rc).should be_true
+        sleep 0.1 # give it time to deliver to the sub socket
+
+        array = []
+        rc = @receiver.recvmsgs(array, NonBlockingFlag)
+        Util.resultcode_ok?(rc).should be_true
+        array.size.should == 10 + 1 # add 1 for the envelope
+      end
+
+    end
+
+    context "PUB" do
+
+      describe "non-blocking #recvmsgs where sender connects & receiver binds" do
+        include APIHelper
+
+        before(:each) do
+          @receiver = @ctx.socket ZMQ::SUB
+          port = bind_to_random_tcp_port(@receiver)
+          assert_ok(@receiver.setsockopt(ZMQ::SUBSCRIBE, ''))
+          @sender = @ctx.socket ZMQ::PUB
+          assert_ok(@sender.connect("tcp://127.0.0.1:#{port}"))
+          sleep 0.3
+        end
+
+        after(:each) do
+          @receiver.close
+          @sender.close
+        end
+
+        it_behaves_like "any socket"
+        it_behaves_like "sockets without exposed envelopes"
+
+      end # describe 'non-blocking recvmsgs'
+
+      describe "non-blocking #recvmsgs where sender binds & receiver connects" do
+        include APIHelper
+
+        before(:each) do
+          @receiver = @ctx.socket ZMQ::SUB
+          port = connect_to_random_tcp_port(@receiver)
+          assert_ok(@receiver.setsockopt(ZMQ::SUBSCRIBE, ''))
+          @sender = @ctx.socket ZMQ::PUB
+          assert_ok(@sender.bind("tcp://127.0.0.1:#{port}"))
+          sleep 0.3
+        end
+
+        after(:each) do
+          @receiver.close
+          @sender.close
+        end
+
+        it_behaves_like "any socket"
+        it_behaves_like "sockets without exposed envelopes"
+
+      end # describe 'non-blocking recvmsgs'
+
+    end # Pub
+
+    context "REQ" do
+
+      describe "non-blocking #recvmsgs where sender connects & receiver binds" do
+        include APIHelper
+
+        before(:each) do
+          @receiver = @ctx.socket ZMQ::REP
+          port = bind_to_random_tcp_port(@receiver)
+          @sender = @ctx.socket ZMQ::REQ
+          assert_ok(@sender.connect("tcp://127.0.0.1:#{port}"))
+          sleep 0.1
+        end
+
+        after(:each) do
+          @receiver.close
+          @sender.close
+        end
+
+        it_behaves_like "any socket"
+        it_behaves_like "sockets without exposed envelopes"
+
+      end # describe 'non-blocking recvmsgs'
+
+      describe "non-blocking #recvmsgs where sender binds & receiver connects" do
+        include APIHelper
+
+        before(:each) do
+          @receiver = @ctx.socket ZMQ::REP
+          port = connect_to_random_tcp_port(@receiver)
+          @sender = @ctx.socket ZMQ::REQ
+          assert_ok(@sender.bind("tcp://127.0.0.1:#{port}"))
+          sleep 0.1
+        end
+
+        after(:each) do
+          @receiver.close
+          @sender.close
+        end
+
+        it_behaves_like "any socket"
+        it_behaves_like "sockets without exposed envelopes"
+
+      end # describe 'non-blocking recvmsgs'
+
+    end # REQ
+
+
+    context "PUSH" do
+
+      describe "non-blocking #recvmsgs where sender connects & receiver binds" do
+        include APIHelper
+
+        before(:each) do
+          @receiver = @ctx.socket ZMQ::PULL
+          port = bind_to_random_tcp_port(@receiver)
+          @sender = @ctx.socket ZMQ::PUSH
+          assert_ok(@sender.connect("tcp://127.0.0.1:#{port}"))
+          sleep 0.1
+        end
+
+        after(:each) do
+          @receiver.close
+          @sender.close
+        end
+
+        it_behaves_like "any socket"
+        it_behaves_like "sockets without exposed envelopes"
+
+      end # describe 'non-blocking recvmsgs'
+
+      describe "non-blocking #recvmsgs where sender binds & receiver connects" do
+        include APIHelper
+
+        before(:each) do
+          @receiver = @ctx.socket ZMQ::PULL
+          port = connect_to_random_tcp_port(@receiver)
+          @sender = @ctx.socket ZMQ::PUSH
+          assert_ok(@sender.bind("tcp://127.0.0.1:#{port}"))
+          sleep 0.1
+        end
+
+        after(:each) do
+          @receiver.close
+          @sender.close
+        end
+
+        it_behaves_like "any socket"
+        it_behaves_like "sockets without exposed envelopes"
+
+      end # describe 'non-blocking recvmsgs'
+
+    end # PUSH
+
+
+    unless LibZMQ.version4?
+
+      context "DEALER" do
+
+        describe "non-blocking #recvmsgs where sender connects & receiver binds" do
+          include APIHelper
+
+          before(:each) do
+            @receiver = @ctx.socket ZMQ::ROUTER
+            port = bind_to_random_tcp_port(@receiver)
+            @sender = @ctx.socket ZMQ::DEALER
+            assert_ok(@sender.connect("tcp://127.0.0.1:#{port}"))
+            sleep 0.1
+          end
+
+          after(:each) do
+            @receiver.close
+            @sender.close
+          end
+
+          it_behaves_like "any socket"
+          it_behaves_like "sockets with exposed envelopes"
+
+        end # describe 'non-blocking recvmsgs'
+
+        describe "non-blocking #recvmsgs where sender binds & receiver connects" do
+          include APIHelper
+
+          before(:each) do
+            @receiver = @ctx.socket ZMQ::ROUTER
+            port = connect_to_random_tcp_port(@receiver)
+            @sender = @ctx.socket ZMQ::DEALER
+            assert_ok(@sender.bind("tcp://127.0.0.1:#{port}"))
+            sleep 0.1
+          end
+
+          after(:each) do
+            @receiver.close
+            @sender.close
+          end
+
+          it_behaves_like "any socket"
+          it_behaves_like "sockets with exposed envelopes"
+
+        end # describe 'non-blocking recvmsgs'
+
+      end # DEALER
+
+    end # unless version4?
+
+
+    context "XREQ" do
+
+      describe "non-blocking #recvmsgs where sender connects & receiver binds" do
+        include APIHelper
+
+        before(:each) do
+          @receiver = @ctx.socket ZMQ::XREP
+          port = bind_to_random_tcp_port(@receiver)
+          @sender = @ctx.socket ZMQ::XREQ
+          assert_ok(@sender.connect("tcp://127.0.0.1:#{port}"))
+          sleep 0.1
+        end
+
+        after(:each) do
+          @receiver.close
+          @sender.close
+        end
+
+        it_behaves_like "any socket"
+        it_behaves_like "sockets with exposed envelopes"
+
+      end # describe 'non-blocking recvmsgs'
+
+      describe "non-blocking #recvmsgs where sender binds & receiver connects" do
+        include APIHelper
+
+        before(:each) do
+          @receiver = @ctx.socket ZMQ::XREP
+          port = connect_to_random_tcp_port(@receiver)
+          @sender = @ctx.socket ZMQ::XREQ
+          assert_ok(@sender.bind("tcp://127.0.0.1:#{port}"))
+          sleep 0.1
+        end
+
+        after(:each) do
+          @receiver.close
+          @sender.close
+        end
+
+        it_behaves_like "any socket"
+        it_behaves_like "sockets with exposed envelopes"
+
+      end # describe 'non-blocking recvmsgs'
+
+    end # XREQ
+
+  end # describe Socket
+
+
+end # module ZMQ