summaryrefslogtreecommitdiff
path: root/spec/ruby/library/socket/socket/accept_loop_spec.rb
blob: 78e8c3fa4af965a06e7617eaab46bdeb6e36b06a (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
76
77
78
79
80
81
82
83
84
require_relative '../spec_helper'

describe 'Socket.accept_loop' do
  before do
    @server = Socket.new(:INET, :STREAM)
    @client = Socket.new(:INET, :STREAM)

    @server.bind(Socket.sockaddr_in(0, '127.0.0.1'))
    @server.listen(1)
  end

  after do
    @client.close
    @server.close
  end

  describe 'using an Array of Sockets' do
    describe 'without any available connections' do
      # FIXME windows randomly hangs here forever
      # https://ci.appveyor.com/project/ruby/ruby/builds/20817932/job/dor2ipny7ru4erpa
      platform_is_not :windows do
        it 'blocks the caller' do
          -> { Socket.accept_loop([@server]) }.should block_caller
        end
      end
    end

    describe 'with available connections' do
      before do
        @client.connect(@server.getsockname)
      end

      it 'yields a Socket and an Addrinfo' do
        conn = nil
        addr = nil

        Socket.accept_loop([@server]) do |connection, address|
          conn = connection
          addr = address
          break
        end

        begin
          conn.should be_an_instance_of(Socket)
          addr.should be_an_instance_of(Addrinfo)
        ensure
          conn.close
        end
      end
    end
  end

  describe 'using separate Socket arguments' do
    describe 'without any available connections' do
      it 'blocks the caller' do
        -> { Socket.accept_loop(@server) }.should block_caller
      end
    end

    describe 'with available connections' do
      before do
        @client.connect(@server.getsockname)
      end

      it 'yields a Socket and an Addrinfo' do
        conn = nil
        addr = nil

        Socket.accept_loop(@server) do |connection, address|
          conn = connection
          addr = address
          break
        end

        begin
          conn.should be_an_instance_of(Socket)
          addr.should be_an_instance_of(Addrinfo)
        ensure
          conn.close
        end
      end
    end
  end
end