third-party/mruby/mrbgems/mruby-io/test/io.rb

   1 ##
   2 # IO Test
   3
   4 MRubyIOTestUtil.io_test_setup
   5 $cr, $crlf, $cmd = MRubyIOTestUtil.win? ? [1, "\r\n", "cmd /c "] : [0, "\n", ""]
   6
   7 def assert_io_open(meth)
   8   assert "assert_io_open" do
   9     fd = IO.sysopen($mrbtest_io_rfname)
  10     assert_equal Fixnum, fd.class
  11     io1 = IO.__send__(meth, fd)
  12     begin
  13       assert_equal IO, io1.class
  14       assert_equal $mrbtest_io_msg, io1.read
  15     ensure
  16       io1.close
  17     end
  18
  19     io2 = IO.__send__(meth, IO.sysopen($mrbtest_io_rfname))do |io|
  20       if meth == :open
  21         assert_equal $mrbtest_io_msg, io.read
  22       else
  23         flunk "IO.#{meth} does not take block"
  24       end
  25     end
  26     io2.close unless meth == :open
  27
  28     assert_raise(RuntimeError) { IO.__send__(meth, 1023) } # For Windows
  29     assert_raise(RuntimeError) { IO.__send__(meth, 1 << 26) }
  30     assert_raise(RuntimeError) { IO.__send__(meth, 1 << 32) } if (1 << 32).kind_of?(Integer)
  31   end
  32 end
  33
  34 assert('IO.class', '15.2.20') do
  35   assert_equal(Class, IO.class)
  36 end
  37
  38 assert('IO.superclass', '15.2.20.2') do
  39   assert_equal(Object, IO.superclass)
  40 end
  41
  42 assert('IO.ancestors', '15.2.20.3') do
  43   assert_include(IO.ancestors, Enumerable)
  44 end
  45
  46 assert('IO.open', '15.2.20.4.1') do
  47   assert_io_open(:open)
  48 end
  49
  50 assert('IO#close', '15.2.20.5.1') do
  51   io = IO.new(IO.sysopen($mrbtest_io_rfname))
  52   assert_nil io.close
  53 end
  54
  55 assert('IO#closed?', '15.2.20.5.2') do
  56   io = IO.new(IO.sysopen($mrbtest_io_rfname))
  57   assert_false io.closed?
  58   io.close
  59   assert_true io.closed?
  60 end
  61
  62 #assert('IO#each', '15.2.20.5.3') do
  63 #assert('IO#each_byte', '15.2.20.5.4') do
  64 #assert('IO#each_line', '15.2.20.5.5') do
  65
  66 assert('IO#eof?', '15.2.20.5.6') do
  67   io = IO.new(IO.sysopen($mrbtest_io_wfname, 'w'), 'w')
  68   assert_raise(IOError) do
  69     io.eof?
  70   end
  71   io.close
  72
  73   # empty file
  74   io = IO.open(IO.sysopen($mrbtest_io_wfname, 'w'), 'w')
  75   io.close
  76   io = IO.open(IO.sysopen($mrbtest_io_wfname, 'r'), 'r')
  77   assert_true io.eof?
  78   io.close
  79
  80   # nonempty file
  81   io = IO.new(IO.sysopen($mrbtest_io_rfname))
  82   assert_false io.eof?
  83   io.readchar
  84   assert_false io.eof?
  85   io.read
  86   assert_true io.eof?
  87   io.close
  88 end
  89
  90 assert('IO#flush', '15.2.20.5.7') do
  91   # Note: mruby-io does not have any buffer to be flushed now.
  92   io = IO.new(IO.sysopen($mrbtest_io_wfname))
  93   assert_equal io, io.flush
  94   io.close
  95   assert_raise(IOError) do
  96     io.flush
  97   end
  98 end
  99
 100 assert('IO#getc', '15.2.20.5.8') do
 101   io = IO.new(IO.sysopen($mrbtest_io_rfname))
 102   $mrbtest_io_msg.split("").each { |ch|
 103     assert_equal ch, io.getc
 104   }
 105   assert_equal nil, io.getc
 106   io.close
 107 end
 108
 109 #assert('IO#gets', '15.2.20.5.9') do
 110 #assert('IO#initialize_copy', '15.2.20.5.10') do
 111 #assert('IO#print', '15.2.20.5.11') do
 112 #assert('IO#putc', '15.2.20.5.12') do
 113 #assert('IO#puts', '15.2.20.5.13') do
 114
 115 assert('IO#read', '15.2.20.5.14') do
 116   IO.open(IO.sysopen($mrbtest_io_rfname)) do |io|
 117     assert_raise(ArgumentError) { io.read(-5) }
 118     assert_raise(TypeError) { io.read("str") }
 119
 120     len = $mrbtest_io_msg.length
 121     assert_equal '', io.read(0)
 122     assert_equal 'mruby', io.read(5)
 123     assert_equal $mrbtest_io_msg[5,len], io.read(len)
 124
 125     assert_equal "", io.read
 126     assert_nil io.read(1)
 127   end
 128
 129   IO.open(IO.sysopen($mrbtest_io_rfname)) do |io|
 130     assert_equal $mrbtest_io_msg, io.read
 131   end
 132 end
 133
 134 assert "IO#read(n) with n > IO::BUF_SIZE" do
 135   skip "pipe is not supported on this platform" if MRubyIOTestUtil.win?
 136   IO.pipe do |r,w|
 137     n = IO::BUF_SIZE+1
 138     w.write 'a'*n
 139     assert_equal 'a'*n, r.read(n)
 140   end
 141 end
 142
 143 assert('IO#readchar', '15.2.20.5.15') do
 144   # almost same as IO#getc
 145   IO.open(IO.sysopen($mrbtest_io_rfname)) do |io|
 146     $mrbtest_io_msg.split("").each { |ch|
 147       assert_equal ch, io.readchar
 148     }
 149     assert_raise(EOFError) do
 150       io.readchar
 151     end
 152   end
 153 end
 154
 155 #assert('IO#readline', '15.2.20.5.16') do
 156 #assert('IO#readlines', '15.2.20.5.17') do
 157
 158 assert('IO#sync', '15.2.20.5.18') do
 159   io = IO.new(IO.sysopen($mrbtest_io_rfname))
 160   s = io.sync
 161   assert_true(s == true || s == false)
 162   io.close
 163   assert_raise(IOError) do
 164     io.sync
 165   end
 166 end
 167
 168 assert('IO#sync=', '15.2.20.5.19') do
 169   io = IO.new(IO.sysopen($mrbtest_io_rfname))
 170   io.sync = true
 171   assert_true io.sync
 172   io.sync = false
 173   assert_false io.sync
 174   io.close
 175   assert_raise(IOError) do
 176     io.sync = true
 177   end
 178 end
 179
 180 assert('IO#write', '15.2.20.5.20') do
 181   io = IO.open(IO.sysopen($mrbtest_io_wfname))
 182   assert_equal 0, io.write("")
 183   io.close
 184
 185   io = IO.open(IO.sysopen($mrbtest_io_wfname, "r+"), "r+")
 186   assert_equal 7, io.write("abcdefg")
 187   io.rewind
 188   assert_equal "ab", io.read(2)
 189   assert_equal 3, io.write("123")
 190   io.rewind
 191   assert_equal "ab123fg", io.read
 192   io.close
 193 end
 194
 195 assert('IO#<<') do
 196   io = IO.open(IO.sysopen($mrbtest_io_wfname))
 197   io << "" << ""
 198   assert_equal 0, io.pos
 199   io.close
 200 end
 201
 202 assert('IO#dup for readable') do
 203   io = IO.new(IO.sysopen($mrbtest_io_rfname))
 204   dup = io.dup
 205   assert_true io != dup
 206   assert_true io.fileno != dup.fileno
 207   begin
 208     assert_true dup.close_on_exec?
 209   rescue NotImplementedError
 210   end
 211   assert_equal 'm', dup.sysread(1)
 212   assert_equal 'r', io.sysread(1)
 213   assert_equal 'u', dup.sysread(1)
 214   assert_equal 'b', io.sysread(1)
 215   assert_equal 'y', dup.sysread(1)
 216   dup.close
 217   assert_false io.closed?
 218   io.close
 219 end
 220
 221 assert('IO#dup for writable') do
 222   io = IO.open(IO.sysopen($mrbtest_io_wfname, 'w+'), 'w+')
 223   dup = io.dup
 224   io.syswrite "mruby"
 225   assert_equal 5, dup.sysseek(0, IO::SEEK_CUR)
 226   io.sysseek 0, IO::SEEK_SET
 227   assert_equal 0, dup.sysseek(0, IO::SEEK_CUR)
 228   assert_equal "mruby", dup.sysread(5)
 229   dup.close
 230   io.close
 231 end
 232
 233 assert('IO.for_fd') do
 234   assert_io_open(:for_fd)
 235 end
 236
 237 assert('IO.new') do
 238   assert_io_open(:new)
 239 end
 240
 241 assert('IO gc check') do
 242   assert_nothing_raised { 100.times { IO.new(0) } }
 243 end
 244
 245 assert('IO.sysopen("./nonexistent")') do
 246   if Object.const_defined? :Errno
 247     eclass = Errno::ENOENT
 248   else
 249     eclass = RuntimeError
 250   end
 251   assert_raise eclass do
 252     fd = IO.sysopen "./nonexistent"
 253     IO._sysclose fd
 254   end
 255 end
 256
 257 assert('IO.sysopen, IO#sysread') do
 258   fd = IO.sysopen $mrbtest_io_rfname
 259   io = IO.new fd
 260   str1 = "     "
 261   str2 = io.sysread(5, str1)
 262   assert_equal $mrbtest_io_msg[0,5], str1
 263   assert_equal $mrbtest_io_msg[0,5], str2
 264   assert_raise EOFError do
 265     io.sysread(10000)
 266     io.sysread(10000)
 267   end
 268
 269   assert_raise RuntimeError do
 270     io.sysread(5, "abcde".freeze)
 271   end
 272
 273   io.close
 274   assert_equal "", io.sysread(0)
 275   assert_raise(IOError) { io.sysread(1) }
 276   assert_raise(ArgumentError) { io.sysread(-1) }
 277   io.closed?
 278
 279   fd = IO.sysopen $mrbtest_io_wfname, "w"
 280   io = IO.new fd, "w"
 281   assert_raise(IOError) { io.sysread(1) }
 282   io.close
 283 end
 284
 285 assert('IO.sysopen, IO#syswrite') do
 286   fd = IO.sysopen $mrbtest_io_wfname, "w"
 287   io = IO.new fd, "w"
 288   str = "abcdefg"
 289   len = io.syswrite(str)
 290   assert_equal str.size, len
 291   io.close
 292
 293   io = IO.new(IO.sysopen($mrbtest_io_rfname), "r")
 294   assert_raise(IOError) { io.syswrite("a") }
 295   io.close
 296 end
 297
 298 assert('IO#_read_buf') do
 299   fd = IO.sysopen $mrbtest_io_rfname
 300   io = IO.new fd
 301   def io._buf
 302     @buf
 303   end
 304   msg_len = $mrbtest_io_msg.size
 305   assert_equal '', io._buf
 306   assert_equal $mrbtest_io_msg, io._read_buf
 307   assert_equal $mrbtest_io_msg, io._buf
 308   assert_equal 'mruby', io.read(5)
 309   assert_equal 5, io.pos
 310   assert_equal msg_len - 5, io._buf.size
 311   assert_equal $mrbtest_io_msg[5,100], io.read
 312   assert_equal 0, io._buf.size
 313   assert_raise EOFError do
 314     io._read_buf
 315   end
 316   assert_equal true, io.eof
 317   assert_equal true, io.eof?
 318   io.close
 319 end
 320
 321 assert('IO#isatty') do
 322   skip "isatty is not supported on this platform" if MRubyIOTestUtil.win?
 323   begin
 324     f = File.open("/dev/tty")
 325   rescue RuntimeError => e
 326     skip e.message
 327   else
 328     assert_true f.isatty
 329   ensure
 330     f&.close
 331   end
 332   begin
 333     f = File.open($mrbtest_io_rfname)
 334     assert_false f.isatty
 335   ensure
 336     f&.close
 337   end
 338 end
 339
 340 assert('IO#pos=, IO#seek') do
 341   fd = IO.sysopen $mrbtest_io_rfname
 342   io = IO.new fd
 343   def io._buf
 344     @buf
 345   end
 346   assert_equal 'm', io.getc
 347   assert_equal 1, io.pos
 348   assert_equal 0, io.seek(0)
 349   assert_equal 0, io.pos
 350   io.close
 351 end
 352
 353 assert('IO#rewind') do
 354   fd = IO.sysopen $mrbtest_io_rfname
 355   io = IO.new fd
 356   assert_equal 'm', io.getc
 357   assert_equal 1, io.pos
 358   assert_equal 0, io.rewind
 359   assert_equal 0, io.pos
 360   io.close
 361 end
 362
 363 assert('IO#gets') do
 364   fd = IO.sysopen $mrbtest_io_rfname
 365   io = IO.new fd
 366
 367   # gets without arguments
 368   assert_equal $mrbtest_io_msg, io.gets, "gets without arguments"
 369   assert_equal nil, io.gets, "gets returns nil, when EOF"
 370
 371   # gets with limit
 372   io.pos = 0
 373   assert_equal $mrbtest_io_msg[0, 5], io.gets(5), "gets with limit"
 374
 375   # gets with rs
 376   io.pos = 0
 377   assert_equal $mrbtest_io_msg[0, 6], io.gets(' '), "gets with rs"
 378
 379   # gets with rs, limit
 380   io.pos = 0
 381   assert_equal $mrbtest_io_msg[0, 5], io.gets(' ', 5), "gets with rs, limit"
 382   io.close
 383   assert_equal true, io.closed?, "close success"
 384
 385   # reading many-lines file.
 386   fd = IO.sysopen $mrbtest_io_wfname, "w"
 387   io = IO.new fd, "w"
 388   io.write "0123456789" * 2 + "\na"
 389   assert_equal 22 + $cr, io.pos
 390   io.close
 391   assert_equal true, io.closed?
 392
 393   fd = IO.sysopen $mrbtest_io_wfname
 394   io = IO.new fd
 395   line = io.gets
 396
 397   # gets first line
 398   assert_equal "0123456789" * 2 + "\n", line, "gets first line"
 399   assert_equal 21, line.size
 400   assert_equal 21 + $cr, io.pos
 401
 402   # gets second line
 403   assert_equal "a", io.gets, "gets second line"
 404
 405   # gets third line
 406   assert_equal nil, io.gets, "gets third line; returns nil"
 407
 408   io.close
 409 end
 410
 411 assert('IO#gets - paragraph mode') do
 412   fd = IO.sysopen $mrbtest_io_wfname, "w"
 413   io = IO.new fd, "w"
 414   io.write "0" * 10 + "\n"
 415   io.write "1" * 10 + "\n\n"
 416   io.write "2" * 10 + "\n"
 417   assert_equal 34 + $cr * 4, io.pos
 418   io.close
 419
 420   fd = IO.sysopen $mrbtest_io_wfname
 421   io = IO.new fd
 422   para1 = "#{'0' * 10}\n#{'1' * 10}\n\n"
 423   text1 = io.gets("")
 424   assert_equal para1, text1
 425   para2 = "#{'2' * 10}\n"
 426   text2 = io.gets("")
 427   assert_equal para2, text2
 428   io.close
 429 end
 430
 431 assert('IO.popen') do
 432   begin
 433     $? = nil
 434     io = IO.popen("#{$cmd}echo mruby-io")
 435     assert_true io.close_on_exec?
 436     assert_equal Fixnum, io.pid.class
 437
 438     out = io.read
 439     assert_equal out.class, String
 440     assert_include out, 'mruby-io'
 441
 442     io.close
 443     if Object.const_defined? :Process
 444       assert_true $?.success?
 445     else
 446       assert_equal 0, $?
 447     end
 448
 449     assert_true io.closed?
 450   rescue NotImplementedError => e
 451     skip e.message
 452   end
 453 end
 454
 455 assert('IO.popen with in option') do
 456   begin
 457     IO.pipe do |r, w|
 458       w.write 'hello'
 459       w.close
 460       assert_equal "hello", IO.popen("cat", "r", in: r) { |i| i.read }
 461       assert_equal "", r.read
 462     end
 463     assert_raise(ArgumentError) { IO.popen("hello", "r", in: Object.new) }
 464   rescue NotImplementedError => e
 465     skip e.message
 466   end
 467 end
 468
 469 assert('IO.popen with out option') do
 470   begin
 471     IO.pipe do |r, w|
 472       IO.popen("echo 'hello'", "w", out: w) {}
 473       w.close
 474       assert_equal "hello\n", r.read
 475     end
 476   rescue NotImplementedError => e
 477     skip e.message
 478   end
 479 end
 480
 481 assert('IO.popen with err option') do
 482   begin
 483     IO.pipe do |r, w|
 484       assert_equal "", IO.popen("echo 'hello' 1>&2", "r", err: w) { |i| i.read }
 485       w.close
 486       assert_equal "hello\n", r.read
 487     end
 488   rescue NotImplementedError => e
 489     skip e.message
 490   end
 491 end
 492
 493 assert('IO.read') do
 494   # empty file
 495   fd = IO.sysopen $mrbtest_io_wfname, "w"
 496   io = IO.new fd, "w"
 497   io.close
 498   assert_equal "",  IO.read($mrbtest_io_wfname)
 499   assert_equal nil, IO.read($mrbtest_io_wfname, 1)
 500
 501   # one byte file
 502   fd = IO.sysopen $mrbtest_io_wfname, "w"
 503   io = IO.new fd, "w"
 504   io.write "123"
 505   io.close
 506   assert_equal "123", IO.read($mrbtest_io_wfname)
 507   assert_equal "",    IO.read($mrbtest_io_wfname, 0)
 508   assert_equal "1",   IO.read($mrbtest_io_wfname, 1)
 509   assert_equal "",    IO.read($mrbtest_io_wfname, 0, 10)
 510   assert_equal "23",  IO.read($mrbtest_io_wfname, 2, 1)
 511   assert_equal "23",  IO.read($mrbtest_io_wfname, 10, 1)
 512   assert_equal "",    IO.read($mrbtest_io_wfname, nil, 10)
 513   assert_equal nil,   IO.read($mrbtest_io_wfname, 1, 10)
 514 end
 515
 516 assert('IO#fileno') do
 517   fd = IO.sysopen $mrbtest_io_rfname
 518   io = IO.new fd
 519   assert_equal io.fileno, fd
 520   assert_equal io.to_i, fd
 521   io.close
 522 end
 523
 524 assert('IO#close_on_exec') do
 525   fd = IO.sysopen $mrbtest_io_wfname, "w"
 526   io = IO.new fd, "w"
 527   begin
 528     # IO.sysopen opens a file descripter with O_CLOEXEC flag.
 529     assert_true io.close_on_exec?
 530   rescue ScriptError
 531     io.close
 532     skip "IO\#close_on_exec is not implemented."
 533   end
 534
 535   io.close_on_exec = false
 536   assert_equal(false, io.close_on_exec?)
 537   io.close_on_exec = true
 538   assert_equal(true, io.close_on_exec?)
 539   io.close_on_exec = false
 540   assert_equal(false, io.close_on_exec?)
 541
 542   io.close
 543
 544   begin
 545     r, w = IO.pipe
 546     assert_equal(true, r.close_on_exec?)
 547     r.close_on_exec = false
 548     assert_equal(false, r.close_on_exec?)
 549     r.close_on_exec = true
 550     assert_equal(true, r.close_on_exec?)
 551
 552     assert_equal(true, w.close_on_exec?)
 553     w.close_on_exec = false
 554     assert_equal(false, w.close_on_exec?)
 555     w.close_on_exec = true
 556     assert_equal(true, w.close_on_exec?)
 557   ensure
 558     r.close unless r.closed?
 559     w.close unless w.closed?
 560   end
 561 end
 562
 563 assert('IO#sysseek') do
 564   IO.open(IO.sysopen($mrbtest_io_rfname)) do |io|
 565     assert_equal 2, io.sysseek(2)
 566     assert_equal 5, io.sysseek(3, IO::SEEK_CUR) # 2 + 3 => 5
 567     assert_equal $mrbtest_io_msg.size - 4, io.sysseek(-4, IO::SEEK_END)
 568   end
 569 end
 570
 571 assert('IO#pread') do
 572   skip "IO#pread is not implemented on this configuration" unless MRubyIOTestUtil::MRB_WITH_IO_PREAD_PWRITE
 573
 574   IO.open(IO.sysopen($mrbtest_io_rfname, 'r'), 'r') do |io|
 575     assert_equal $mrbtest_io_msg.byteslice(5, 8), io.pread(8, 5)
 576     assert_equal 0, io.pos
 577     assert_equal $mrbtest_io_msg.byteslice(1, 5), io.pread(5, 1)
 578     assert_equal 0, io.pos
 579     assert_raise(RuntimeError) { io.pread(20, -9) }
 580   end
 581 end
 582
 583 assert('IO#pwrite') do
 584   skip "IO#pwrite is not implemented on this configuration" unless MRubyIOTestUtil::MRB_WITH_IO_PREAD_PWRITE
 585
 586   IO.open(IO.sysopen($mrbtest_io_wfname, 'w+'), 'w+') do |io|
 587     assert_equal 6, io.pwrite("Warld!", 7)
 588     assert_equal 0, io.pos
 589     assert_equal 7, io.pwrite("Hello, ", 0)
 590     assert_equal 0, io.pos
 591     assert_equal "Hello, Warld!", io.read
 592     assert_equal 6, io.pwrite("world!", 7)
 593     assert_equal 13, io.pos
 594     io.pos = 0
 595     assert_equal "Hello, world!", io.read
 596   end
 597 end
 598
 599 assert('IO.pipe') do
 600   begin
 601     called = false
 602     IO.pipe do |r, w|
 603       assert_true r.kind_of?(IO)
 604       assert_true w.kind_of?(IO)
 605       assert_false r.closed?
 606       assert_false w.closed?
 607       assert_true FileTest.pipe?(r)
 608       assert_true FileTest.pipe?(w)
 609       assert_nil r.pid
 610       assert_nil w.pid
 611       assert_true 2 < r.fileno
 612       assert_true 2 < w.fileno
 613       assert_true r.fileno != w.fileno
 614       assert_false r.sync
 615       assert_true w.sync
 616       assert_equal 8, w.write('test for')
 617       assert_equal 'test', r.read(4)
 618       assert_equal ' for', r.read(4)
 619       assert_equal 5, w.write(' pipe')
 620       assert_equal nil, w.close
 621       assert_equal ' pipe', r.read
 622       called = true
 623       assert_raise(IOError) { r.write 'test' }
 624       # TODO:
 625       # This assert expect raise IOError but got RuntimeError
 626       # Because mruby-io not have flag for I/O readable
 627       # assert_raise(IOError) { w.read }
 628     end
 629     assert_true called
 630
 631     assert_nothing_raised do
 632       IO.pipe { |r, w| r.close; w.close }
 633     end
 634   rescue NotImplementedError => e
 635     skip e.message
 636   end
 637 end
 638
 639 assert('`cmd`') do
 640   begin
 641     assert_equal `#{$cmd}echo foo`, "foo#{$crlf}"
 642   rescue NotImplementedError => e
 643     skip e.message
 644   end
 645 end
 646
 647 MRubyIOTestUtil.io_test_cleanup