lib/abstract_notifier/testing/rspec.rb in abstract_notifier-0.1.0 vs lib/abstract_notifier/testing/rspec.rb in abstract_notifier-0.1.1

@@ -2,11 +2,11 @@
 
 module AbstractNotifier
   class HaveSentNotification < RSpec::Matchers::BuiltIn::BaseMatcher
     attr_reader :payload
 
-    def initialize(payload)
+    def initialize(payload = nil)
       @payload = payload
       set_expected_number(:exactly, 1)
     end
 
     def exactly(count)
@@ -53,10 +53,10 @@
       proc.call
       in_block_deliveries = deliveries.drop(original_deliveries_count)
 
       @matching_deliveries, @unmatching_deliveries =
         in_block_deliveries.partition do |actual_payload|
-          payload === actual_payload
+          payload.nil? || (payload === actual_payload)
         end
 
       @matching_count = @matching_deliveries.size
 
       case @expectation_type