vertica 0.10.4 → 0.10.5

data/VERSION

@@ -1 +1 @@
-0.10.4
+0.10.5

data/lib/vertica/connection.rb

@@ -214,7 +214,8 @@ class Vertica::Connection
   end
 
   def write_bytes(bytes)
-    socket.write_nonblock bytes
+    bytes_written = socket.write_nonblock(bytes)
+    write_bytes(bytes[bytes_written...bytes.size]) if bytes_written < bytes.size
   rescue IO::WaitReadable, IO::WaitWritable => wait_error
     io_select(wait_error)
     retry

data/test/functional/query_test.rb

@@ -128,6 +128,15 @@ class QueryTest < Test::Unit::TestCase
       @connection.query("BLAH")
     end
   end
+
+  def test_copy_in_alot_of_data_with_customer_handler
+    @connection.copy "COPY test_ruby_vertica_table FROM STDIN" do |data|
+      data.write "11|#{"a" * 1_000_000}\n"
+    end
+    
+    result = @connection.query("SELECT id FROM test_ruby_vertica_table ORDER BY id", :row_style => :array)
+    assert_equal 2, result.length
+  end
   
   def test_copy_in_with_customer_handler
     @connection.copy "COPY test_ruby_vertica_table FROM STDIN" do |data|
@@ -159,7 +168,7 @@ class QueryTest < Test::Unit::TestCase
     assert_equal 4, result.length
     assert_equal [[1, "matt"], [11, "Stuff"], [12, "More stuff"], [13, "Final stuff"]], result.rows
   end
-  
+
   def test_copy_in_with_io
     io = StringIO.new("11|Stuff\r\n12|More stuff\n13|Final stuff\n")
     @connection.copy "COPY test_ruby_vertica_table FROM STDIN", io

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: vertica
 version: !ruby/object:Gem::Version
-  version: 0.10.4
+  version: 0.10.5
   prerelease: 
 platform: ruby
 authors:
@@ -11,7 +11,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-08-07 00:00:00.000000000 Z
+date: 2013-08-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake