Return default email count for unknown senders on CLI.

1 files changed, 9 insertions, 6 deletions

diff --git a/t/bin.t b/t/bin.t
index b27e994..5431242 100644..100755
--- a/t/bin.t
+++ b/t/bin.t
@@ -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;