Contents

def fixture_dir
  Pathname.new File.expand_path("../../../fixtures", __FILE__)
end

# Would use Addrinfo(addr,0) but the class is missing/unstable on RBX.
def assign_port
  port = 12_000 + Random.rand(1024)
  attempts = 0
  begin
    socket = ::TCPServer.new(example_addr, port)
  rescue Errno::ECONNREFUSED, Errno::EADDRINUSE => ex
    raise ex.class.new("Tried #{attempts} times to assign port.") unless attempts < Specs::MAX_ATTEMPTS
    attempts += 1
    port += 1
    socket.close rescue nil
    sleep 0.126
    retry
  end
  return port
ensure
  socket.close rescue nil
end

def example_addr
  "127.0.0.1"
end

def example_unix_sock
  "/tmp/cell_sock"
end

def within_io_actor(&block)
  actor = WrapperActor.new
  actor.wrap(&block)
ensure
  actor.terminate if actor.alive? rescue nil
end

def with_tcp_server(port)
  server = Celluloid::IO::TCPServer.new(example_addr, port)
  begin
    yield server
  ensure
    server.close
  end
end

def with_unix_server
  server = Celluloid::IO::UNIXServer.open(example_unix_sock)
  begin
    yield server
  ensure
    server.close
    File.delete(example_unix_sock)
  end
end

def with_connected_sockets(port)
  with_tcp_server(port) do |server|
    client = Celluloid::IO::TCPSocket.new(example_addr, port)
    peer = server.accept

    begin
      yield peer, client
    ensure
      begin
        client.close
        peer.close
      rescue
      end
    end
  end
end

def with_connected_unix_sockets
  with_unix_server do |server|
    client = Celluloid::IO::UNIXSocket.new(example_unix_sock)
    peer = server.accept

    begin
      yield peer, client
    ensure
      begin
        client.close
        peer.close
      rescue
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
celluloid-io-0.17.3	spec/support/examples/methods.rb
celluloid-io-0.17.2	spec/support/examples/methods.rb