test/test_epoll_optimizations.rb

   1 require 'test/unit'
   2 begin
   3   require 'strace'
   4 rescue LoadError
   5 end
   6 $-w = true
   7
   8 require 'sleepy_penguin'
   9
  10 class TestEpollOptimizations < Test::Unit::TestCase
  11   include SleepyPenguin
  12   RBX = defined?(RUBY_ENGINE) && (RUBY_ENGINE == 'rbx')
  13   IO_PURGATORY = []
  14
  15   def setup
  16     @rd, @wr = IO.pipe
  17     @ep = Epoll.new
  18   end
  19
  20   def teardown
  21     [ @ep, @rd, @wr ].each { |io| io.close unless io.closed? }
  22   end
  23
  24   def test_set
  25     io, err = Strace.me do
  26       @ep.set(@wr, Epoll::OUT)
  27       @ep.set(@wr, Epoll::OUT)
  28     end
  29     assert_nil err
  30     lines = io.readlines; io.close
  31     assert_equal 1, lines.grep(/^epoll_ctl/).size
  32     assert_match /EPOLL_CTL_ADD/, lines.grep(/^epoll_ctl/)[0]
  33
  34     io, err = Strace.me { @ep.set(@wr, Epoll::OUT | Epoll::ONESHOT) }
  35     assert_nil err
  36     lines = io.readlines; io.close
  37     assert_equal 1, lines.grep(/^epoll_ctl/).size
  38     assert_match /EPOLL_CTL_MOD/, lines.grep(/^epoll_ctl/)[0]
  39
  40     io, err = Strace.me { @ep.set(@wr, Epoll::OUT) }
  41     assert_nil err
  42     lines = io.readlines; io.close
  43     assert_equal 1, lines.grep(/^epoll_ctl/).size
  44     assert_match /EPOLL_CTL_MOD/, lines.grep(/^epoll_ctl/)[0]
  45     @wr.close
  46     @rd.close
  47
  48     @rd, @wr = IO.pipe
  49     io, err = Strace.me { @ep.set(@wr, Epoll::OUT) }
  50     assert_nil err
  51     lines = io.readlines; io.close
  52     assert_equal 1, lines.grep(/^epoll_ctl/).size
  53     assert_match /EPOLL_CTL_ADD/, lines.grep(/^epoll_ctl/)[0]
  54   end
  55
  56   def test_delete
  57     @ep.set(@wr, Epoll::OUT)
  58     rv = true
  59     io, err = Strace.me { rv = @ep.delete(@wr) }
  60     assert_equal @wr, rv
  61     assert_nil err
  62     lines = io.readlines; io.close
  63     assert_equal 1, lines.grep(/^epoll_ctl/).size
  64     assert_match %r{=\s+0$}, lines.grep(/^epoll_ctl/)[0]
  65
  66     rv = true
  67     io, err = Strace.me { rv = @ep.delete(@wr) }
  68     assert_nil rv
  69     assert_nil err
  70     lines = io.readlines; io.close
  71     assert_equal 0, lines.grep(/^epoll_ctl/).size
  72   end
  73
  74   def test_delete_closed
  75     a = @wr.fileno
  76     @ep.set(@wr, Epoll::OUT)
  77     @rd.close
  78     @wr.close
  79     @rd, @wr = IO.pipe
  80     assert_equal a, @wr.fileno
  81     rv = true
  82     io, err = Strace.me { rv = @ep.delete(@wr) }
  83     lines = io.readlines; io.close
  84     assert_nil err
  85     assert_nil rv
  86     assert_equal 0, lines.grep(/^epoll_ctl/).size
  87   end
  88
  89   def test_delete_closed_proxy
  90     obj = Struct.new(:to_io).new(@wr)
  91     rv = nil
  92     @ep.add(obj, Epoll::OUT)
  93     @wr.close
  94     io, err = Strace.me { rv = @ep.delete(obj) }
  95     lines = io.readlines; io.close
  96     assert_kind_of IOError, err
  97     assert_nil rv
  98     assert_equal 0, lines.grep(/^epoll_ctl/).size
  99   end
 100
 101   def test_delete_closed_fileno
 102     fileno = @wr.fileno
 103     @ep.add(fileno, Epoll::OUT)
 104     @wr.close
 105     rv = nil
 106     io, err = Strace.me { rv = @ep.delete(fileno) }
 107     lines = io.readlines; io.close
 108     assert_nil err
 109     assert_equal fileno, rv
 110     assert_equal 0, lines.grep(/^epoll_ctl/).size
 111   end
 112
 113   def test_delete_aliased_a
 114     tmp = IO.for_fd @wr.fileno
 115     IO_PURGATORY << tmp
 116     @ep.set(tmp, Epoll::OUT)
 117     rv = nil
 118     io, err = Strace.me { rv = @ep.delete(@wr) }
 119     lines = io.readlines; io.close
 120     assert_equal @wr, rv
 121     assert_nil err
 122     assert_equal 1, lines.grep(/^epoll_ctl/).size
 123     assert_match %r{=\s+0$}, lines.grep(/^epoll_ctl/)[0]
 124     assert_equal 0, lines.grep(/ENOENT/).size
 125   end
 126
 127   def test_delete_aliased_b
 128     tmp = IO.for_fd @wr.fileno
 129     IO_PURGATORY << tmp
 130     @ep.set(@wr, Epoll::OUT)
 131     rv = nil
 132     io, err = Strace.me { rv = @ep.delete(tmp) }
 133     lines = io.readlines; io.close
 134     assert_equal tmp, rv
 135     assert_nil err
 136     assert_equal 1, lines.grep(/^epoll_ctl/).size
 137     assert_match %r{=\s+0$}, lines.grep(/^epoll_ctl/)[0]
 138     assert_equal 0, lines.grep(/ENOENT/).size
 139   end
 140
 141   def test_delete_aliased_closed
 142     tmp = IO.for_fd @wr.fileno
 143     IO_PURGATORY << tmp
 144     @ep.set(tmp, Epoll::OUT)
 145     @wr.close
 146     rv = nil
 147     io, err = Strace.me { rv = @ep.delete(tmp) }
 148     lines = io.readlines; io.close
 149     assert_nil rv
 150     assert_nil err
 151     assert_equal 1, lines.grep(/^epoll_ctl/).size
 152     assert_equal 1, lines.grep(/EBADF/).size
 153   end
 154 end if defined?(Strace)