require_relative '../spec_helper'
require_relative '../fixtures/classes'

# NATFIXME: CI issues on macos tests, disable them.
platform_is_not :darwin do
  describe "Socket::IPSocket#recvfrom" do
    before :each do
      @server = TCPServer.new("127.0.0.1", 0)
      @port = @server.addr[1]
      @client = TCPSocket.new("127.0.0.1", @port)
    end

    after :each do
      @server.close unless @server.closed?
      @client.close unless @client.closed?
    end

    # NATFIXME: Unreliable test
    xit "reads data from the connection" do
      data = nil
      t = Thread.new do
        client = @server.accept
        begin
          data = client.recvfrom(6)
        ensure
          client.close
        end
      end

      @client.send('hello', 0)
      @client.shutdown rescue nil
      # shutdown may raise Errno::ENOTCONN when sent data is pending.
      t.join

      data.first.should == 'hello'
    end

    it "reads up to len bytes" do
      data = nil
      t = Thread.new do
        client = @server.accept
        begin
          data = client.recvfrom(3)
        ensure
          client.close
        end
      end

      @client.send('hello', 0)
      @client.shutdown rescue nil
      t.join

      data.first.should == 'hel'
    end

    it "returns an array with the data and connection info" do
      data = nil
      t = Thread.new do
        client = @server.accept
        data = client.recvfrom(3)
        client.close
      end

      @client.send('hello', 0)
      @client.shutdown rescue nil
      t.join

      data.size.should == 2
      data.first.should == "hel"
      # This does not apply to every platform, dependent on recvfrom(2)
      # data.last.should == nil
    end
  end

  describe "Socket::IPSocket#recvfrom" do
    context "when recvfrom(2) returns 0 (if no messages are available to be received and the peer has performed an orderly shutdown)" do
      describe "stream socket" do
        before :each do
          @server = TCPServer.new("127.0.0.1", 0)
          port = @server.addr[1]
          @client = TCPSocket.new("127.0.0.1", port)
        end

        after :each do
          @server.close unless @server.closed?
          @client.close unless @client.closed?
        end

        ruby_version_is ""..."3.3" do
          it "returns an empty String as received data on a closed stream socket" do
            t = Thread.new do
              client = @server.accept
              message = client.recvfrom(10)
              message
            ensure
              client.close if client
            end

            Thread.pass while t.status and t.status != "sleep"
            t.status.should_not be_nil

            @client.close

            t.value.should.is_a? Array
            t.value[0].should == ""
          end
        end

        ruby_version_is "3.3" do
          # NATFIXME: Client gets closed before we can read, this results in errors in our spec runner
          xit "returns nil on a closed stream socket" do
            t = Thread.new do
              client = @server.accept
              message = client.recvfrom(10)
              message
            ensure
              client.close if client
            end

            Thread.pass while t.status and t.status != "sleep"
            t.status.should_not be_nil

            @client.close

            t.value.should be_nil
          end
        end
      end

      describe "datagram socket" do
        SocketSpecs.each_ip_protocol do |family, ip_address|
          before :each do
            @server = UDPSocket.new(family)
            @client = UDPSocket.new(family)
          end

          after :each do
            @server.close unless @server.closed?
            @client.close unless @client.closed?
          end

          it "returns an empty String as received data" do
            @server.bind(ip_address, 0)
            addr = @server.connect_address
            @client.connect(addr.ip_address, addr.ip_port)

            @client.send('', 0)
            message = @server.recvfrom(1)

            message.should.is_a? Array
            message[0].should == ""
          end
        end
      end
    end
  end

  describe 'Socket::IPSocket#recvfrom' do
    SocketSpecs.each_ip_protocol do |family, ip_address, family_name|
      before do
        @server = UDPSocket.new(family)
        @client = UDPSocket.new(family)

        @server.bind(ip_address, 0)
        @client.connect(ip_address, @server.connect_address.ip_port)

        @hostname = Socket.getaddrinfo(ip_address, nil)[0][2]
      end

      after do
        @client.close
        @server.close
      end

      it 'returns an Array containing up to N bytes and address information' do
        @client.write('hello')

        port = @client.local_address.ip_port
        ret  = @server.recvfrom(2)

        ret.should == ['he', [family_name, port, @hostname, ip_address]]
      end

      it 'allows specifying of flags when receiving data' do
        @client.write('hello')

        @server.recvfrom(2, Socket::MSG_PEEK)[0].should == 'he'

        @server.recvfrom(2)[0].should == 'he'
      end

      describe 'using reverse lookups' do
        before do
          @server.do_not_reverse_lookup = false

          @hostname = Socket.getaddrinfo(ip_address, nil, 0, 0, 0, 0, true)[0][2]
        end

        it 'includes the hostname in the address Array' do
          @client.write('hello')

          port = @client.local_address.ip_port
          ret  = @server.recvfrom(2)

          ret.should == ['he', [family_name, port, @hostname, ip_address]]
        end
      end
    end
  end
end
