diff options
| -rwxr-xr-x | bin/spamcat | 2 | ||||
| -rwxr-xr-x[-rw-r--r--] | t/bin.t | 15 |
2 files changed, 10 insertions, 7 deletions
diff --git a/bin/spamcat b/bin/spamcat index 8549d95..25f04d2 100755 --- a/bin/spamcat +++ b/bin/spamcat @@ -94,7 +94,7 @@ if ($#ARGV >= 0) { } elsif ($cmd eq 'get') { my $sender = shift @ARGV; pod2usage(1) unless $sender; - my $count = $sch->get_count($sender); + my $count = $sch->get_count($sender) || $conf{default_count}; print "$sender has $count messages remaining.\n"; } elsif ($cmd eq 'set') { my ($sender, $count) = @ARGV; @@ -1,6 +1,6 @@ #!/usr/bin/env perl -use Test::More tests => 1; +use Test::More tests => 3; use IO::File; @@ -12,7 +12,7 @@ my $conffile = 't/fixtures/sample.conf'; # Add testlib which has createdb and possibly population of said db. -my @dumpconfig = `$spamcat -c t/fixtures/sample.conf dumpconfig`; +my @dumpconfig = `$spamcat -c $conffile dumpconfig`; my %got = parse_configdump(@dumpconfig); my %expected = (DBPATH => '/tmp/spamcat.sqlite3', DEFAULT_COUNT => 10, @@ -20,10 +20,13 @@ my %expected = (DBPATH => '/tmp/spamcat.sqlite3', DOMAINS => "spamcat.example.com, spamcat2.example.com, spamcat3"); is_deeply(\%got, \%expected); -# TODO: Test for proper exit codes. -#my $fh = IO::File->new("|$spamcat -c $conffile") || -# die "Couldn't open pipe to $spamcat: $!\n"; -#$fh->close; +# Unknown senders have the default number of mails left. +my @got = `$spamcat -c $conffile get unknown-sender`; +is_deeply(\@got, ["unknown-sender has $expected{DEFAULT_COUNT} messages remaining.\n"]); + +# Missing argument to set fails. +my $got = system("$spamcat -c $conffile set missing-argument") >> 8; +is_deeply($got, 1); sub parse_configdump { my %rc; |