spec/timber/cli/installers/root_spec.rb in timber-2.1.0.rc3 vs spec/timber/cli/installers/root_spec.rb in timber-2.1.0.rc4

@@ -23,18 +23,19 @@
 
   describe ".run" do
     it "should run properly" do
       input.string = "y\n"
 
-      expect(installer).to receive(:install_platform).with(app).exactly(1).times.and_return(true)
-      expect(installer).to receive(:run_sub_installer).with(app).exactly(1).times.and_return(true)
-      expect(installer).to receive(:send_test_messages).exactly(1).times.and_return(true)
-      expect(installer).to receive(:confirm_log_delivery).exactly(1).times.and_return(true)
-      expect(installer).to receive(:assist_with_git).exactly(1).times.and_return(true)
+      expect(installer).to receive(:install_platform).with(app).exactly(1).times
+      expect(installer).to receive(:run_sub_installer).with(app).exactly(1).times
+      expect(installer).to receive(:send_test_messages).exactly(1).times
+      expect(installer).to receive(:confirm_log_delivery).exactly(1).times
+      expect(installer).to receive(:wrap_up).with(app).exactly(1).times
       expect(api).to receive(:event).with(:success).exactly(1).times
-      expect(installer).to receive(:collect_feedback).exactly(1).times.and_return(true)
+      expect(installer).to receive(:collect_feedback).exactly(1).times
+      expect(installer).to receive(:free_data).exactly(1).times
 
-      expect(installer.run(app)).to eq(true)
+      installer.run(app)
     end
   end
 
   describe ".install_platform" do
     context "non-heroku" do
\ No newline at end of file