Fix behaviour on privmsg and add tests for privmsg

[poe-component-irc-plugin-seen.git] / t / POE-Component-IRC-Plugin-Seen.t

diff --git a/t/POE-Component-IRC-Plugin-Seen.t b/t/POE-Component-IRC-Plugin-Seen.t
index fdf55aebc6d041bf934b5116b868652ca9021649..607e5f66323ae266cbd8285a4f2f15c0b68a52de 100644 (file)
--- a/t/POE-Component-IRC-Plugin-Seen.t
+++ b/t/POE-Component-IRC-Plugin-Seen.t
@@ -2,7 +2,7 @@ use v5.14;
 use strict;
 use warnings;
 
-use Test::More tests => 8;
+use Test::More tests => 10;
 use Test::MockObject;
 
 BEGIN { *CORE::GLOBAL::localtime = sub { 'now' } }
@@ -21,9 +21,10 @@ my $rmgv = \'mgv!marius@ieval.ro';
 
 # Sub setup
 sub runtest{
-       my ($message, $expect, $comment) = @_;
+       my ($message, $expect, $comment, $privmsg) = @_;
        undef $last_msg;
-       $self->S_public($mockirc, $rmgv, \$channels, \$message);
+       $self->S_public($mockirc, $rmgv, \$channels, \$message) unless $privmsg;
+       $self->S_msg($mockirc, $rmgv, \$channels, \$message) if $privmsg;
        is($last_msg, $expect, $comment)
 }
 
@@ -43,3 +44,7 @@ $self->S_part($mockirc, $rmgv, \'#chan', \'buh-bye');
 runtest 'bot: seen mgv', "I last saw mgv now parting #chan with message 'buh-bye'", 'part with message';
 
 runtest 'bot: seen asd', "I haven't seen asd", "haven't seen";
+
+# Private messages
+runtest 'seen asd', "I haven't seen asd", "haven't seen", 1;
+runtest ' !seen asd', "I haven't seen asd", "haven't seen", 1;