Skip to content
This repository
tree: db4af11304
Fetching contributors…

Octocat-spinner-32-eaf2f5

Cannot retrieve contributors at this time

file 61 lines (59 sloc) 1.67 kb
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60
module ActionMailer
  module TestHelper
    # Asserts that the number of emails sent matches the given number.
    #
    # def test_emails
    # assert_emails 0
    # ContactMailer.welcome.deliver
    # assert_emails 1
    # ContactMailer.welcome.deliver
    # assert_emails 2
    # end
    #
    # If a block is passed, that block should cause the specified number of
    # emails to be sent.
    #
    # def test_emails_again
    # assert_emails 1 do
    # ContactMailer.welcome.deliver
    # end
    #
    # assert_emails 2 do
    # ContactMailer.welcome.deliver
    # ContactMailer.welcome.deliver
    # end
    # end
    def assert_emails(number)
      if block_given?
        original_count = ActionMailer::Base.deliveries.size
        yield
        new_count = ActionMailer::Base.deliveries.size
        assert_equal original_count + number, new_count, "#{number} emails expected, but #{new_count - original_count} were sent"
      else
        assert_equal number, ActionMailer::Base.deliveries.size
      end
    end

    # Assert that no emails have been sent.
    #
    # def test_emails
    # assert_no_emails
    # ContactMailer.welcome.deliver
    # assert_emails 1
    # end
    #
    # If a block is passed, that block should not cause any emails to be sent.
    #
    # def test_emails_again
    # assert_no_emails do
    # # No emails should be sent from this block
    # end
    # end
    #
    # Note: This assertion is simply a shortcut for:
    #
    # assert_emails 0
    def assert_no_emails(&block)
      assert_emails 0, &block
    end
  end
end
Something went wrong with that request. Please try again.