Support with blocks

Author: Joel Lubrano

lib/rspec/rails/matchers/have_enqueued_mail.rb

@@ -24,9 +24,12 @@ def description
           "enqueues #{@mailer_class.name}.#{@method_name}"
         end
 
-        def with(*args)
+        def with(*args, &block)
           @mail_args = args
-          super(*mailer_args)
+
+          super(*mailer_args) do |*job_args|
+            block.call(*(job_args - base_mailer_args)) if block.present?
+          end
         end
 
         def matches?(block)
@@ -62,10 +65,8 @@ def expected_count_message
         end
 
         def mailer_args
-          base_args = [@mailer_class.name, @method_name.to_s, 'deliver_now']
-
           if @mail_args.any?
-            base_args + @mail_args
+            base_mailer_args + @mail_args
           else
             mailer_method_arity = @mailer_class.instance_method(@method_name).arity
 
@@ -75,10 +76,14 @@ def mailer_args
                                mailer_method_arity
                              end
 
-            base_args + Array.new(number_of_args) { anything }
+            base_mailer_args + Array.new(number_of_args) { anything }
           end
         end
 
+        def base_mailer_args
+          [@mailer_class.name, @method_name.to_s, 'deliver_now']
+        end
+
         def check_active_job_adapter
           return if ::ActiveJob::QueueAdapters::TestAdapter === ::ActiveJob::Base.queue_adapter
           raise StandardError, "To use HaveEnqueuedMail matcher set `ActiveJob::Base.queue_adapter = :test`"

spec/rspec/rails/matchers/have_enqueued_mail_spec.rb

@@ -242,5 +242,40 @@ def email_with_optional_args(required_arg, optional_arg = nil); end
 
       ActiveJob::Base.queue_adapter = queue_adapter
     end
+
+    it "fails with with block with incorrect data" do
+      expect {
+        expect {
+          TestMailer.email_with_args('asdf', 'zxcv').deliver_later
+        }.to have_enqueued_mail(TestMailer, :email_with_args).with { |first_arg, second_arg|
+          expect(first_arg).to eq("zxcv")
+        }
+      }.to raise_error { |e|
+        expect(e.message).to match(/expected: "zxcv"/)
+        expect(e.message).to match(/got: "asdf"/)
+      }
+    end
+
+    it "passes multiple arguments to with block" do
+      expect {
+        TestMailer.email_with_args('asdf', 'zxcv').deliver_later
+      }.to have_enqueued_mail(TestMailer, :email_with_args).with { |first_arg, second_arg|
+        expect(first_arg).to eq("asdf")
+        expect(second_arg).to eq("zxcv")
+      }
+    end
+
+    it "only calls with block if other conditions are met" do
+      noon = Date.tomorrow.noon
+      midnight = Date.tomorrow.midnight
+
+      expect {
+        TestMailer.email_with_args('high', 'noon').deliver_later(wait_until: noon)
+        TestMailer.email_with_args('midnight', 'rider').deliver_later(wait_until: midnight)
+      }.to have_enqueued_mail(TestMailer, :email_with_args).at(noon).with { |first_arg, second_arg|
+        expect(first_arg).to eq('high')
+        expect(second_arg).to eq('noon')
+      }
+    end
   end
 end