# encoding: utf-8

require File.expand_path("../helper.rb", __FILE__)

class TestZmqPoller < ZmqTestCase

  def test_alloc
    assert_instance_of ZMQ::Poller, ZMQ::Poller.new
  end

  def test_verbose
    poller = ZMQ::Poller.new
    poller.verbose = true
  end

  def test_poll_sockets
    ctx = ZMQ::Context.new
    poller = ZMQ::Poller.new
    poller.verbose = true
    assert_equal 0, poller.poll
    rep = ctx.socket(:REP)
    rep.linger = 0
    rep.bind("inproc://test.poll")
    req = ctx.socket(:REQ)
    req.linger = 0
    req.connect("inproc://test.poll")

    assert_raises TypeError do
      poller.poll :invalid
    end

    assert_equal 0, poller.poll_nonblock

    assert poller.register_readable(rep)
    assert req.send("request")
    sleep 0.1

    assert_equal 1, poller.poll(1)
    assert_equal [rep], poller.readables
    assert_equal [], poller.writables
    rep.recv

    assert poller.register(req)
    assert rep.send("reply")
    sleep 0.1

    assert_equal 1, poller.poll(1)
    assert_equal [req], poller.readables
    assert_equal [], poller.writables
  ensure
    ctx.destroy
  end

  def test_poll_ios
    poller = ZMQ::Poller.new
    r, w = IO.pipe

    poller.register(ZMQ::Pollitem(r, ZMQ::POLLIN))
    poller.register(ZMQ::Pollitem(w, ZMQ::POLLOUT))

    w.write("message")
    sleep 0.2

    assert_equal 2, poller.poll(1)
    assert_equal [r], poller.readables
    assert_equal [w], poller.writables

    assert_equal "message", r.read(7)
  end

  def test_poll_ruby_sockets
    poller = ZMQ::Poller.new
    server = TCPServer.new("127.0.0.1", 0)
    port = server.addr[1]
    client = TCPSocket.new("127.0.0.1", port)
    s = server.accept

    poller.register(ZMQ::Pollitem(s, ZMQ::POLLIN))
    poller.register(ZMQ::Pollitem(client, ZMQ::POLLOUT))

    client.send("message", 0)
    sleep 0.2

    assert_equal 2, poller.poll(1)
    assert_equal [s], poller.readables
    assert_equal [client], poller.writables

    assert_equal "message", s.recv_nonblock(7)
  end

  def test_register
    ctx = ZMQ::Context.new
    rep = ctx.bind(:REP, 'inproc://test.poller-register')
    req = ctx.connect(:REQ, 'inproc://test.poller-register')
    poller = ZMQ::Poller.new
    assert poller.register(rep)
    assert_raises ZMQ::Error do
      poller.register(ZMQ::Pollitem.new(req, 0))
    end
    unbound = ctx.socket(:REP)
    assert_raises ZMQ::Error do
      poller.register(unbound)
    end
  ensure
    ctx.destroy
  end

  def test_register_readable
    ctx = ZMQ::Context.new
    rep = ctx.bind(:REP, 'inproc://test.poller-register_readable')
    poller = ZMQ::Poller.new
    assert poller.register_readable(rep)
  ensure
    ctx.destroy
  end

  def test_register_writable
    ctx = ZMQ::Context.new
    rep = ctx.bind(:REP, 'inproc://test.poller-register_writable')
    poller = ZMQ::Poller.new
    assert poller.register_writable(rep)
  ensure
    ctx.destroy
  end

  def test_remove
    ctx = ZMQ::Context.new
    rep = ctx.bind(:REP, 'inproc://test.poller-remove')
    req = ctx.connect(:REQ, 'inproc://test.poller-remove')
    poller = ZMQ::Poller.new
    assert poller.register(req)
    assert poller.remove(req)
    assert !poller.remove(rep)

    io_poll_item = ZMQ::Pollitem.new(STDIN, ZMQ::POLLIN)
    poller.register(io_poll_item)
    assert poller.remove(io_poll_item)

    poller.register(io_poll_item)
    assert poller.remove(STDIN)
  ensure
    ctx.destroy
  end

  def test_readables
    poller = ZMQ::Poller.new
    assert_equal [], poller.readables
  end

  def test_writables
    poller = ZMQ::Poller.new
    assert_equal [], poller.writables
  end
end