logs_spec.rb in shelly-0.4.31

- old
+ new

@@ -64,11 +64,11 @@
     end
   end
 
   describe "#get" do
     before do
-      @bar = mock(:progress_callback => @callback)
+      @bar = mock(:progress_callback => @callback, :finish => true)
       Shelly::DownloadProgressBar.stub(:new).and_return(@bar)
       @client.stub(:download_application_logs_attributes).
         with("foo-production", {:date => "2013-05-31"}).
         and_return({"filename" => "foo-production.log.20130531.gz",
                    "url" => "http://example.com/foo-production/20130531",
@@ -98,10 +98,10 @@
 
     it "should fetch filename, url and size and initialize download progress bar" do
       @client.should_receive(:download_file).
         with("foo-production", "foo-production.log.20130531.gz",
              "http://example.com/foo-production/20130531", @callback)
-      Shelly::DownloadProgressBar.should_receive(:new).with(12345).and_return(@bar)
+      Shelly::DownloadProgressBar.should_receive(:new).and_return(@bar)
       invoke(@cli_logs, :get, "2013-05-31")
     end
 
     it "should fetch given log file itself" do
       @client.should_receive(:download_file).