summaryrefslogtreecommitdiff
path: root/spec/ruby/library/socket/addrinfo/connect_from_spec.rb
blob: 55fce2e1595aec387dc68804718e106c6587909a (plain)
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
require_relative '../spec_helper'
require_relative '../fixtures/classes'

describe 'Addrinfo#connect_from' do
  SocketSpecs.each_ip_protocol do |family, ip_address|
    before do
      @server = TCPServer.new(ip_address, 0)
      @port   = @server.connect_address.ip_port
      @addr   = Addrinfo.tcp(ip_address, @port)
    end

    after do
      @socket.close if @socket
      @server.close
    end

    describe 'using separate arguments' do
      it 'returns a Socket when no block is given' do
        @socket = @addr.connect_from(ip_address, 0)
        @socket.should be_an_instance_of(Socket)
      end

      it 'yields the Socket when a block is given' do
        @addr.connect_from(ip_address, 0) do |socket|
          socket.should be_an_instance_of(Socket)
        end
      end

      it 'treats the last argument as a set of options if it is a Hash' do
        @socket = @addr.connect_from(ip_address, 0, timeout: 2)
        @socket.should be_an_instance_of(Socket)
      end

      it 'binds the socket to the local address' do
        @socket = @addr.connect_from(ip_address, 0)

        @socket.local_address.ip_address.should == ip_address

        @socket.local_address.ip_port.should > 0
        @socket.local_address.ip_port.should_not == @port
      end
    end

    describe 'using an Addrinfo as the 1st argument' do
      before do
        @from_addr = Addrinfo.tcp(ip_address, 0)
      end

      it 'returns a Socket when no block is given' do
        @socket = @addr.connect_from(@from_addr)
        @socket.should be_an_instance_of(Socket)
      end

      it 'yields the Socket when a block is given' do
        @addr.connect_from(@from_addr) do |socket|
          socket.should be_an_instance_of(Socket)
        end
      end

      it 'treats the last argument as a set of options if it is a Hash' do
        @socket = @addr.connect_from(@from_addr, timeout: 2)
        @socket.should be_an_instance_of(Socket)
      end

      it 'binds the socket to the local address' do
        @socket = @addr.connect_from(@from_addr)

        @socket.local_address.ip_address.should == ip_address

        @socket.local_address.ip_port.should > 0
        @socket.local_address.ip_port.should_not == @port
      end
    end
  end
end