File: multipart_spec.rb

package info (click to toggle)
ruby-ffi-rzmq 2.0.7-1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, bullseye, forky, sid, trixie
  • size: 392 kB
  • sloc: ruby: 2,992; sh: 21; makefile: 2
file content (136 lines) | stat: -rw-r--r-- 4,098 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
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 "using #send_strings" do
        include APIHelper

        before(:all) do
          @receiver = Socket.new(@ctx.pointer, ZMQ::REP)
          port = bind_to_random_tcp_port(@receiver)

          @sender = Socket.new(@ctx.pointer, ZMQ::REQ)
          rc = @sender.connect("tcp://127.0.0.1:#{port}")
        end

        after(:all) do
          @sender.close
          @receiver.close
        end
        
        it "correctly handles a multipart message array with 1 element" do
          data = [ "1" ]

          @sender.send_strings(data)
          sleep 1
          strings = []
          rc = @receiver.recv_strings(strings)
          expect(strings).to eq(data)
        end
      end


      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)
          expect(strings).to eq(req_data)

          @rep.send_strings(rep_data)
          strings = []
          rc = @req.recv_strings(strings)
          expect(strings).to eq(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|
            expect(message.copy_out_string).to eq(req_data[index])
          end

          @rep.send_strings(rep_data)
          messages = []
          rc = @req.recvmsgs(messages)
          messages.each_with_index do |message, index|
            expect(message.copy_out_string).to eq(rep_data[index])
          end
        end
      end

        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)
            expect(strings).to eq([ @req.identity, "", "hello" ])

            @rep.send_strings(rep_data)
            string = ''
            rc = @req.recv_string(string)
            expect(string).to eq(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)
            expect(msgs[0].copy_out_string).to eq(@req.identity)
            expect(msgs[1].copy_out_string).to eq("")
            expect(msgs[2].copy_out_string).to eq("hello")

            @rep.send_strings(rep_data)
            msgs = []
            rc = @req.recvmsgs(msgs)
            expect(msgs[0].copy_out_string).to eq(rep_data.last)
          end
        end

    end
  end
end