Merge pull request #83 from joshuamcginnis/streaming_support

Add #size and #rewind to Net::SFTP::Operations::File

Author: mfazekas

lib/net/sftp/operations/file.rb

@@ -131,6 +131,14 @@ def print(*items)
       nil
     end
 
+    def size
+      stat.size
+    end
+
+    def rewind
+      self.pos = 0
+    end
+
     # Writes each argument to the stream, appending a newline to any item
     # that does not already end in a newline. Array arguments are flattened.
     def puts(*items)

test/test_file.rb

@@ -156,4 +156,20 @@ def test_stat_should_return_attributes_object_for_handle
     @sftp.expects(:fstat!).with("handle").returns(stat)
     assert_equal stat, @file.stat
   end
-end
+
+  def test_size_should_return_size_from_stat
+    stat = stub(size: 1024)
+    @sftp.expects(:fstat!).with("handle").returns(stat)
+    assert_equal 1024, @file.size
+  end
+
+  def test_rewind
+    @sftp.expects(:write!).with("handle", 0, "hello world\n")
+    @sftp.expects(:read!).with("handle", 12, 8192).returns("hello world\n")
+    @sftp.expects(:read!).with("handle", 0, 8192).returns("hello world\n")
+    @file.puts "hello world\n"
+    assert_equal "hello", @file.read(5)
+    @file.rewind
+    assert_equal "hello world", @file.read(11)
+  end
+end