test_armailer.rb in adzap-ar_mailer-2.1.6

- old
+ new

@@ -5,18 +5,20 @@
 
   def mail
     @mail = Object.new
     def @mail.encoded() 'email' end
     def @mail.from() ['nobody@example.com'] end
+    def @mail.[](key) {'return-path' => $return_path, 'from' => 'nobody@example.com'}[key] end
     def @mail.destinations() %w[user1@example.com user2@example.com] end
   end
 
 end
 
 class TestARMailer < Test::Unit::TestCase
 
   def setup
+    $return_path = nil
     Mailer.email_class = Email
 
     Email.records.clear
     Newsletter.records.clear
   end
@@ -25,9 +27,18 @@
     Mailer.email_class = Newsletter
 
     Mailer.deliver_mail
 
     assert_equal 2, Newsletter.records.length
+  end
+  
+  def test_perform_delivery_activerecord_when_return_path_is_present
+    $return_path = stub(:spec => 'return-path@example.com')
+    Mailer.deliver_mail
+
+    assert_equal 2, Email.records.length
+    record = Email.records.first
+    assert_equal 'return-path@example.com', record.from
   end
 
   def test_perform_delivery_activerecord
     Mailer.deliver_mail
 