lib/stripe_event/subscriber.rb in stripe_event-0.3.1 vs lib/stripe_event/subscriber.rb in stripe_event-0.4.0

@@ -1,26 +1,25 @@
 module StripeEvent
   class Subscriber
-    def initialize(*names, &block)
+    def initialize(*names)
       @names = names
-      @block = block
       ensure_valid_types!
     end
     
-    def register
+    def register(&block)
       ActiveSupport::Notifications.subscribe(pattern) do |*_, payload|
-        @block.call(payload[:event])
+        block.call(payload[:event])
       end
     end
     
-    private
-    
     def pattern
       Regexp.union(@names.empty? ? TYPE_LIST : @names)
     end
     
+    private
+    
     def ensure_valid_types!
       invalid_names = @names.select { |name| !TYPE_LIST.include?(name) }
-      raise InvalidEventType.new(invalid_names) if invalid_names.any?
+      raise InvalidEventTypeError.new(invalid_names) if invalid_names.any?
     end
   end
 end