Validate address on route definition

1 files changed, 9 insertions, 3 deletions

diff --git a/lib/action_mailbox/router/route.rb b/lib/action_mailbox/router/route.rb
index e123a93bb1..34cc684381 100644
--- a/lib/action_mailbox/router/route.rb
+++ b/lib/action_mailbox/router/route.rb
@@ -1,10 +1,10 @@
 class ActionMailbox::Router::Route
-  class InvalidAddressError < StandardError; end
-
   attr_reader :address, :mailbox_name
 
   def initialize(address, to:)
     @address, @mailbox_name = address, to
+
+    ensure_valid_address
   end
 
   def match?(inbound_email)
@@ -16,7 +16,7 @@ class ActionMailbox::Router::Route
     when Proc
       address.call(inbound_email)
     else
-      address.try(:match?, inbound_email) || raise(InvalidAddressError)
+      address.match?(inbound_email)
     end
   end
 
@@ -25,6 +25,12 @@ class ActionMailbox::Router::Route
   end
 
   private
+    def ensure_valid_address
+      unless [ String, Regexp, Proc ].any? { |klass| address.is_a?(klass) } || address.respond_to?(:match?)
+        raise ArgumentError, "Expected a String, Regexp, Proc, or matchable, got #{address.inspect}"
+      end
+    end
+
     def recipients_from(mail)
       Array(mail.to) + Array(mail.cc) + Array(mail.bcc)
     end