continuum-stager-api 0.1.8 → 0.1.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 12edd2b8b9c59ea9e1559d0d82a70dcd80974852
-  data.tar.gz: 4ff0c5b27bcce22bbfceeeb0efbbb42ec58f9156
+  metadata.gz: c915dbe247ca720afed7f0c88727537f61422181
+  data.tar.gz: b311c1b061e61b7c3bd082cbee449574ad24b1fb
 SHA512:
-  metadata.gz: 882efbf11bd13e8d214ab3da90bec6305fd6d77d3bd1bf8e1c02b5b428ebed875a79a1df9d92db6882b08d1fe51c960dcd059fd8a6f84ce04f742f4d30f66621
-  data.tar.gz: 428d0d08dd667c23809f312c89cbde645d0971b8f4605788faafe4c92e1a3ac3199667972172a16ccfa31b55e524f1b780eaa73d23bac41ada92dacf77290ae9
+  metadata.gz: f0e7376958039ab0c30e9b44f128a5985e83f9bf8897a21af5ac1c6b0aac2b2e6c1bb81c657ba4e39dccaf302d0f101617f5520949e499ec48356fd53bde305b
+  data.tar.gz: 7044cce662b548f8582f17a4cbb970ea082c14cc5ae9aea41788f1af2487617e2bf9e94ecf72bd0e860578953e7fd54115db485b6862c4efda3be093f46e1989

data/apcera-stager-api-contrib.gemspec

@@ -15,7 +15,7 @@ Gem::Specification.new do |gem|
   gem.test_files    = `git ls-files -- {spec}/*`.split("\n")
   gem.name          = "continuum-stager-api"
   gem.require_paths = ["lib"]
-  gem.version       = "0.1.8"
+  gem.version       = "0.1.9"
 
   gem.add_development_dependency 'rspec', '~> 2.6.0'
   gem.add_development_dependency 'rake'

data/lib/apcera/stager/error.rb

@@ -1,5 +1,6 @@
 module Apcera
   module Error
+    class DownloadError < StandardError; end
     class StagerURLRequired < StandardError; end
     class ExecuteError < StandardError; end
     class AppPathError < StandardError; end

data/lib/apcera/stager/stager.rb

@@ -15,10 +15,24 @@ module Apcera
     end
 
     # Download a package from the staging coordinator.
+    # We use Net::HTTP here because it supports streaming downloads.
     def download
-      response = RestClient.get(@stager_url + "/data")
-      File.open(@pkg_path, "wb") do |f|
-        f.write(response.to_str)
+      uri = URI(@stager_url + "/data")
+
+      Net::HTTP.start(uri.host, uri.port) do |http|
+        request = Net::HTTP::Get.new uri
+
+        http.request request do |response|
+          if response.code.to_i == 200
+            open @pkg_path, 'wb' do |io|
+              response.read_body do |chunk|
+                io.write chunk
+              end
+            end
+          else
+            raise Apcera::Error::DownloadError.new("package download failed.\n")
+          end
+        end
       end
     rescue => e
       fail e

data/spec/apcera/stager/stager_spec.rb

@@ -70,7 +70,7 @@ describe Apcera::Stager do
         @stager.should_receive(:exit0r).with(1) { raise }
 
         VCR.use_cassette('invalid/download') do
-          expect { @stager.download }.to raise_error(RestClient::ResourceNotFound, "404 Resource Not Found")
+          expect { @stager.download }.to raise_error(Apcera::Error::DownloadError, "package download failed.\n")
         end
       end
     end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: continuum-stager-api
 version: !ruby/object:Gem::Version
-  version: 0.1.8
+  version: 0.1.9
 platform: ruby
 authors:
 - Josh Ellithorpe