A subscription can now "do_once"

pdcawley · Nov 4, 2011 · c396108 · c396108
1 parent e7bfd91
commit c396108
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 0 deletions.
diff --git a/lib/Announcements/Subscription.pm b/lib/Announcements/Subscription.pm
@@ -30,6 +30,20 @@ has _registry => (
     }
 );
 
+sub BUILDARGS {
+    my $class = shift;
+    my $params = $class->SUPER::BUILDARGS(@_);
+
+    if (my $oneshot = delete $params->{do_once}) {
+        $params->{do} = sub {
+            $_[2]->unsubscribe;
+            $oneshot->(@_);
+        };
+    }
+    return $params;
+}
+
+
 
 sub send {
     my $self         = shift;

diff --git a/t/unsubscribing.t b/t/unsubscribing.t
@@ -40,6 +40,23 @@ subtest "Basic unsubscription" => sub {
     is $announcement_count, 1;
 };
 
+subtest "Basic unsubscription" => sub {
+    my $nuke = PushedButton->new;
+    my $announcement_count = 0;
+
+    ok $nuke->add_subscription(
+        when => 'PushedButton',
+        do_once => sub {
+            $announcement_count++;
+        }
+    );
+
+    $nuke->push;
+    $nuke->push;
+
+    is $announcement_count, 1;
+};
+
 subtest "Double subscription is wrong, m'kay?" => sub {
     my $nuke = PushedButton->new;
     my $announcement_count = 0;