diff options
| -rwxr-xr-x | bin/spamcat | 18 | ||||
| -rw-r--r-- | lib/SpamCat.pm | 2 | ||||
| -rw-r--r-- | t/lib.t | 5 |
3 files changed, 22 insertions, 3 deletions
diff --git a/bin/spamcat b/bin/spamcat index fd37f76..8549d95 100755 --- a/bin/spamcat +++ b/bin/spamcat @@ -24,6 +24,14 @@ Print a brief help message and exit. Load configuration from C<file> +=item get C<sender> + +Show database row for C<sender>. + +=item set C<sender> C<count> + +Set C<sender>'s remaining message count to C<count>. + =item dumpconfig Dump the current configuration. @@ -83,6 +91,16 @@ if ($#ARGV >= 0) { } print uc($k) . " = " . $v . "\n"; } + } elsif ($cmd eq 'get') { + my $sender = shift @ARGV; + pod2usage(1) unless $sender; + my $count = $sch->get_count($sender); + print "$sender has $count messages remaining.\n"; + } elsif ($cmd eq 'set') { + my ($sender, $count) = @ARGV; + pod2usage(1) unless $count && $count =~ /\d+/; + $sch->set_count($sender, $count); + print "$sender has $count messages remaining.\n"; } else { pod2usage(1); } diff --git a/lib/SpamCat.pm b/lib/SpamCat.pm index d470520..7a160cf 100644 --- a/lib/SpamCat.pm +++ b/lib/SpamCat.pm @@ -144,7 +144,7 @@ sub set_count_t { $q = 'INSERT INTO emails (count, sender) VALUES (?, ?)'; } else { # Otherwise update the record with the new count. - $q = 'UPDATE emails SET count = ? modified = CURRENT_TIMESTAMP WHERE sender = ?' + $q = 'UPDATE emails SET count = ?, modified = CURRENT_TIMESTAMP WHERE sender = ?' } my $sth = $self->{dbh}->prepare($q); $sth->execute($count, $sender); @@ -1,6 +1,6 @@ # -*- Mode: cperl -*- -use Test::More tests => 41; +use Test::More tests => 42; use strict; use warnings; @@ -39,7 +39,8 @@ ok(!defined $sch->get_count('doesntexist'), 'Non-existant sender has undefined count'); ok(SpamCat->can('set_count'), 'Has count setter'); -is($sch->set_count('bar', 1), 1, 'Setting count returns existing count'); +is($sch->set_count('bar', 10), 10, 'Setting count returns existing count'); +is($sch->set_count('bar', 1), 1, 'Updating existing count to 1 returns 1'); is($sch->decrement_count('bar'), 0, 'Decrementing count from 1 returns 0'); is($sch->decrement_count('bar'), 0, 'Decrementing count from 0 returns 0'); |