amelioration(balancer_delivery_method): implemente le fait de forcer la methode de delivery au niveau de notre balancer de fournisseur demail

spec/lib/balancer_delivery_method_spec.rb

@@ -5,6 +5,20 @@ RSpec.describe BalancerDeliveryMethod do
     end
   end
 
+  class ImportantEmail < ApplicationMailer
+    before_action :set_x_deliver_with
+
+    def greet(name)
+      mail(to: "smtp_to", from: "smtp_from", body: "Hello #{name}")
+    end
+
+    private
+
+    def set_x_deliver_with
+      headers['X-deliver-with'] = :mock_smtp
+    end
+  end
+
   class TestMail
     def self.deliveries
       @deliveries ||= []
@@ -46,6 +60,7 @@ RSpec.describe BalancerDeliveryMethod do
     ActionMailer::Base.add_delivery_method :balancer, BalancerDeliveryMethod
 
     ExampleMailer.delivery_method = :balancer
+    ImportantEmail.delivery_method = :balancer
   end
 
   context 'when a single delivery method is provided' do
@@ -101,6 +116,35 @@ RSpec.describe BalancerDeliveryMethod do
     end
   end
 
+  context 'SafeMailer.important_email_use_delivery_method is present' do
+    before do
+      allow(SafeMailer).to receive(:important_email_use_delivery_method).and_return(delivery_method)
+      ActionMailer::Base.balancer_settings = { mock_smtp: 10, mock_sendmail: 5 }
+
+      rng_sequence = [3, 14, 1]
+      BalancerDeliveryMethod.random = FixedSequence.new(rng_sequence)
+    end
+
+    after do
+      BalancerDeliveryMethod.random = Random.new
+    end
+
+    context 'known delivery_method & email is important' do
+      let(:delivery_method) { :mock_smtp }
+
+      it 'sends emails given the forced_delivery_method' do
+        mail1 = ImportantEmail.greet('Lucia').deliver_now
+        expect(mail1).to have_been_delivered_using(MockSmtp)
+
+        mail2 = ImportantEmail.greet('Damian').deliver_now
+        expect(mail2).to have_been_delivered_using(MockSmtp)
+
+        mail3 = ImportantEmail.greet('Rahwa').deliver_now
+        expect(mail3).to have_been_delivered_using(MockSmtp)
+      end
+    end
+  end
+
   # Helpers
 
   def have_been_delivered_using(delivery_class)