td-client 1.0.3 → 1.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f8123cb6d1ba413d6e482a5396d2675340256e1b
-  data.tar.gz: 3e8541a2b1070b50dc270c54bb5f76a24e29cd24
+  metadata.gz: 10f1e7ce3ea3f8b94d5c2d2cce131ef2968c534f
+  data.tar.gz: 24fa95fd8ebb8751ba502265a7d15942e6f5a978
 SHA512:
-  metadata.gz: 7bdbacf95cd257c635dd6d72486b32974408d9f09e4f81e1ec1233c4db0b709bbb0b48991c2820c80d1a94c305295b7809771dbc34ca5e8b0e885a1146053277
-  data.tar.gz: 901e31f20e95e0aa83fabaa4daf18470376aee5396cbe09cad474e5680ab1e56fee43ceae3e466b91f92d1b60b7de1a4b2ad777e86e309807bf3d5a733007fcd
+  metadata.gz: e3dce86a68061a523fd51215e1fb37d6f0b654660dbc4c2fd87420d0b155bf8bd5de084b8b0aa4b2a932f130012bf7d7ee49214bcb589fa2d3d4463ba72f8439
+  data.tar.gz: 16ab57b46b6d3c208bcbeb690c1578b30e64d2643ff518f57889ca20c9980be6e387846d2d308b8dbcbdee63c3d25fea768d9c84dcfb3427c1ea63a0d4652e74

data/lib/td/client/api/job.rb

@@ -313,8 +313,12 @@ module Job
     current_total_chunk_size = 0
     infl = nil
     begin # LOOP of Network/Server errors
+      first_chunk_p = true
       response = client.get(url, params, header) do |res, chunk|
-        validate_response_status(res, current_total_chunk_size)
+        # Validate only on first chunk
+        validate_response_status(res, current_total_chunk_size) if first_chunk_p
+        first_chunk_p = false
+
         if infl.nil? && autodecode
           case res.header['Content-Encoding'][0].to_s.downcase
           when 'gzip'

data/lib/td/client/version.rb

@@ -1,5 +1,5 @@
 module TreasureData
   class Client
-    VERSION = '1.0.3'
+    VERSION = '1.0.4'
   end
 end

data/spec/td/client/job_api_spec.rb

@@ -341,6 +341,21 @@ describe 'Job API' do
       expect(api.job_result(12345)).to eq ['hello', 'world']
     end
 
+    it 'has multiple chunks' do
+      client = double('client')
+      allow(api).to receive(:new_client).and_return([client, {}])
+      allow(client).to receive(:send_timeout=)
+      allow(client).to receive(:receive_timeout=)
+      res = HTTP::Message.new_response(nil, HTTP::Message::Headers.new)
+      sz = 3
+      chunk1 = packed[0, sz]
+      chunk2 = packed[sz, packed.bytesize-sz]
+      expect(client).to receive(:get).
+        and_yield(res, chunk1).
+        and_yield(res, chunk2).
+        and_return(res)
+      expect(api.job_result(12345)).to eq ['hello', 'world']
+    end
   end
 
   describe 'job_result_format' do

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: td-client
 version: !ruby/object:Gem::Version
-  version: 1.0.3
+  version: 1.0.4
 platform: ruby
 authors:
 - Treasure Data, Inc.
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-09-15 00:00:00.000000000 Z
+date: 2017-09-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: msgpack