require File.join(File.dirname(__FILE__), %w[spec_helper])

module ZMQ
  describe Context do
    context "when running basic push pull" do
      include APIHelper

      let(:string) { "booga-booga" }

      before(:each) do
        $stdout.flush
        @context = ZMQ::Context.new
        @push = @context.socket ZMQ::PUSH
        @pull = @context.socket ZMQ::PULL
        @link = "tcp://127.0.0.1:#{random_port}"
        @pull.connect @link
        @push.bind    @link
      end
      
      after(:each) do
        @push.close
        @pull.close
      end
      
      it "should receive an exact copy of the sent message using Message objects directly on one pull socket" do
        @push.send_string string
        received = @pull.recv_string
        received.should == string
      end
       
      it "should receive an exact string copy of the message sent when receiving in non-blocking mode and using Message objects directly" do
        sent_message = Message.new string
        received_message = Message.new

        @push.send sent_message
        sleep 0.1 # give it time for delivery
        @pull.recv received_message, ZMQ::NOBLOCK
        received_message.copy_out_string.should == string
      end

      it "should receive a single message for each message sent on each socket listening, when an equal number pulls to messages" do
        received = []
        threads  = []
        count    = 4
        @pull.close # close this one since we aren't going to use it below and we don't want it to receive a message
         
        count.times do
          threads << Thread.new do
            pull = @context.socket ZMQ::PULL
            rc = pull.connect @link
            received << pull.recv_string
            pull.close
          end
          sleep 0.001 # give each thread time to spin up
        end
        
        count.times { @push.send_string(string) }

        threads.each {|t| t.join}
        
        received.find_all {|r| r == string}.length.should == count
      end
      
    end # @context ping-pong
  end # describe
end # module ZMQ