From ef8f6fda82f19d5e8e7f56b390447c75299547c5 Mon Sep 17 00:00:00 2001
From: Kat <katrina@kosiak.co.uk>
Date: Fri, 9 Dec 2022 11:12:07 +0000
Subject: [PATCH] Update test

---
 spec/requests/locations_controller_spec.rb | 17 +++--------------
 spec/requests/schemes_controller_spec.rb   | 11 +++--------
 2 files changed, 6 insertions(+), 22 deletions(-)

diff --git a/spec/requests/locations_controller_spec.rb b/spec/requests/locations_controller_spec.rb
index 1aca81a00..13e66672f 100644
--- a/spec/requests/locations_controller_spec.rb
+++ b/spec/requests/locations_controller_spec.rb
@@ -1344,7 +1344,6 @@ RSpec.describe LocationsController, type: :request do
         before do
           FactoryBot.create_list(:lettings_log, 1, :sh, location:, scheme:, startdate:, created_by: user_a)
           FactoryBot.create_list(:lettings_log, 3, :sh, location:, scheme:, startdate:, created_by: user_b)
-          allow(LocationOrSchemeDeactivationMailer).to receive_message_chain(:send_deactivation_mail, :deliver_later).and_return(true)
 
           Timecop.freeze(Time.utc(2022, 10, 10))
           sign_in user
@@ -1385,19 +1384,9 @@ RSpec.describe LocationsController, type: :request do
 
         context "and the users need to be notified" do
           it "sends E-mails to the creators of affected logs with counts" do
-            expect(LocationOrSchemeDeactivationMailer).to receive(:send_deactivation_mail).with(user_a,
-                                                                                                1,
-                                                                                                url_for(controller: "lettings_logs", action: "update_logs"),
-                                                                                                location.scheme.service_name,
-                                                                                                location.postcode)
-
-            expect(LocationOrSchemeDeactivationMailer).to receive(:send_deactivation_mail).with(user_b,
-                                                                                                3,
-                                                                                                url_for(controller: "lettings_logs", action: "update_logs"),
-                                                                                                location.scheme.service_name,
-                                                                                                location.postcode)
-
-            patch "/schemes/#{scheme.id}/locations/#{location.id}/deactivate", params:
+            expect {
+              patch "/schemes/#{scheme.id}/locations/#{location.id}/deactivate", params:
+            }.to enqueue_job(ActionMailer::MailDeliveryJob).at_least(2).times
           end
         end
 
diff --git a/spec/requests/schemes_controller_spec.rb b/spec/requests/schemes_controller_spec.rb
index 8de025ee7..af74d90f0 100644
--- a/spec/requests/schemes_controller_spec.rb
+++ b/spec/requests/schemes_controller_spec.rb
@@ -1865,8 +1865,6 @@ RSpec.describe SchemesController, type: :request do
         let(:mailer) { instance_double(LocationOrSchemeDeactivationMailer) }
 
         before do
-          allow(LocationOrSchemeDeactivationMailer).to receive_message_chain(:send_deactivation_mail, :deliver_later).and_return(true)
-
           Timecop.freeze(Time.utc(2022, 10, 10))
           sign_in user
         end
@@ -1925,12 +1923,9 @@ RSpec.describe SchemesController, type: :request do
 
         context "and the users need to be notified" do
           it "sends E-mails to the creators of affected logs with counts" do
-            expect(LocationOrSchemeDeactivationMailer).to receive(:send_deactivation_mail).with(user,
-                                                                                                1,
-                                                                                                url_for(controller: "lettings_logs", action: "update_logs"),
-                                                                                                scheme.service_name)
-
-            patch "/schemes/#{scheme.id}/deactivate", params:
+            expect {
+              patch "/schemes/#{scheme.id}/deactivate", params:
+            }.to enqueue_job(ActionMailer::MailDeliveryJob)
           end
         end
       end