Merge pull request #44 from takahashim/io-addstr

support IO#<<

2 files changed, 13 insertions, 0 deletions

diff --git a/mrblib/io.rb b/mrblib/io.rb
index c838b96e5..30eb47404 100644
--- a/mrblib/io.rb
+++ b/mrblib/io.rb
@@ -108,6 +108,11 @@ class IO
     raise IOError
   end
 
+  def <<(str)
+    write(str)
+    self
+  end
+
   def eof?
     return true if @buf && @buf.size > 0
 
diff --git a/test/io.rb b/test/io.rb
index b828fef49..a5ce7e049 100644
--- a/test/io.rb
+++ b/test/io.rb
@@ -150,6 +150,14 @@ assert('IO#write', '15.2.20.5.20') do
   true
 end
 
+assert('IO#<<') do
+  io = IO.open(IO.sysopen($mrbtest_io_wfname))
+  io << "" << ""
+  assert_equal 0, io.pos
+  io.close
+  true
+end
+
 assert('IO.for_fd') do
   fd = IO.sysopen($mrbtest_io_rfname)
   io = IO.for_fd(fd)