Add default options to whitelist

dewski · Mar 14, 2012 · 27854ca · 27854ca
1 parent 3325a34
commit 27854ca
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/test/test_helper.rb b/test/test_helper.rb
@@ -28,7 +28,7 @@ class Test::Unit::TestCase
   # Returns instance of Mail::Message.
   def with_whitelist(options={})
     filter = MailGate::Filter.new \
-      :whitelist => options.delete(:whitelist),
+      :whitelist => options.fetch(:whitelist, /.*/),
       :subject_prefix => options.delete(:subject_prefix),
       :append_emails => options.delete(:append_emails)
 

diff --git a/test/unit/mail_gate/filter_test.rb b/test/unit/mail_gate/filter_test.rb
@@ -9,7 +9,7 @@ def test_without_whitelist_regex
   def test_with_whitelist_as_string
     mail = with_whitelist \
       :to => 'garrett@site.com',
-      :whitelist => '@site.com'
+      :whitelist => /\@site.com/
     assert_equal %w{ garrett@site.com }, mail.to
   end