socket_with_timeout_spec.rb

# frozen_string_literal: true

describe Kafka::SocketWithTimeout, ".open" do
  it "times out if the server doesn't accept the connection within the timeout" do
    host = "172.16.0.0" # this address is non-routable!
    port = 4444

    timeout = 0.1
    allowed_time = timeout + 0.1

    start = Time.now

    expect {
      Kafka::SocketWithTimeout.new(host, port, connect_timeout: timeout, timeout: 1)
    }.to raise_exception(Errno::ETIMEDOUT)

    finish = Time.now

    expect(finish - start).to be < allowed_time
  end

  describe "#read" do
    it "times out after the specified amount of time" do
      host = "localhost"
      server = TCPServer.new(host, 0)
      port = server.addr[1]

      timeout = 0.1
      allowed_time = timeout + 0.1

      socket = Kafka::SocketWithTimeout.new(host, port, connect_timeout: 1, timeout: timeout)

      start = Time.now

      expect {
        socket.read(4)
      }.to raise_exception(Errno::ETIMEDOUT)

      finish = Time.now

      expect(finish - start).to be < allowed_time
    end

    it "finishes immediately when #close is called" do
      host = "localhost"
      server = TCPServer.new(host, 0)
      port = server.addr[1]

      socket = Kafka::SocketWithTimeout.new(host, port, connect_timeout: 1, timeout: 5)

      thread = Thread.new do
        begin
          socket.read(4)
        rescue Errno::ETIMEDOUT
          :timeout
        end
      end

      socket.close

      expect { thread.value }.to raise_error(IOError)
    end
  end
end