spec/shelly/cli/logs_spec.rb in shelly-0.5.0 vs spec/shelly/cli/logs_spec.rb in shelly-0.5.1

@@ -122,12 +122,13 @@
       invoke(@cli_logs, :get, "2013-05-31")
     end
 
     context "on log file not found" do
       it "should display error message" do
-        exception = RestClient::ResourceNotFound.new
+        exception = Shelly::Client::NotFoundException.new
         @client.stub(:download_file).and_raise(exception)
-        $stdout.should_receive(:puts).with(red "Log file not found")
+        @bar.should_receive(:clear)
+        $stdout.should_receive(:puts).with(red "Log file for 2013-05-31 not found")
         invoke(@cli_logs, :get, "2013-05-31")
       end
     end
 
     context "on invalid date format" do