my
$bot1
= POE::Component::IRC->spawn(
Flood
=> 1,
plugin_debug
=> 1,
);
my
$bot2
= POE::Component::IRC->spawn(
Flood
=> 1,
plugin_debug
=> 1,
);
my
$ircd
= POE::Component::Server::IRC->spawn(
Auth
=> 0,
AntiFlood
=> 0,
);
POE::Session->create(
package_states
=> [
main
=> [
qw(
_start
ircd_listener_add
ircd_listener_failure
_shutdown
irc_001
irc_join
irc_notice
irc_disconnected
)
],
],
);
$poe_kernel
->run();
sub
_start {
my
(
$kernel
) =
$_
[KERNEL];
$ircd
->yield(
'register'
,
'all'
);
$ircd
->yield(
'add_listener'
);
$kernel
->delay(
_shutdown
=> 60,
'Timed out'
);
}
sub
ircd_listener_failure {
my
(
$kernel
,
$op
,
$reason
) =
@_
[KERNEL, ARG1, ARG3];
$kernel
->yield(
'_shutdown'
,
"$op: $reason"
);
}
sub
ircd_listener_add {
my
(
$kernel
,
$port
) =
@_
[KERNEL, ARG0];
$bot1
->yield(
register
=>
'all'
);
$bot1
->yield(
connect
=> {
nick
=>
'TestBot1'
,
server
=>
'127.0.0.1'
,
port
=>
$port
,
});
$bot2
->yield(
register
=>
'all'
);
$bot2
->yield(
connect
=> {
nick
=>
'TestBot2'
,
server
=>
'127.0.0.1'
,
port
=>
$port
,
});
}
sub
_shutdown {
my
(
$kernel
,
$error
) =
@_
[KERNEL, ARG0];
fail(
$error
)
if
defined
$error
;
$kernel
->alarm_remove_all();
$ircd
->yield(
'shutdown'
);
$bot1
->yield(
'shutdown'
);
$bot2
->yield(
'shutdown'
);
}
sub
irc_001 {
my
$irc
=
$_
[SENDER]->get_heap();
pass(
'Logged in'
);
$irc
->yield(
join
=>
'#testchannel'
);
return
if
$irc
!=
$bot1
;
my
$plugin
= POE::Component::IRC::Plugin::BotCommand->new(
Auth_sub
=>
sub
{
return
1
if
$_
[3] eq
'help'
;
return
0;
}
);
ok(
$irc
->plugin_add(
BotCommand
=>
$plugin
),
'Add plugin with no commands'
);
}
sub
irc_join {
my
(
$heap
,
$sender
,
$who
,
$where
) =
@_
[HEAP, SENDER, ARG0, ARG1];
my
$nick
= (
split
/!/,
$who
)[0];
my
$irc
=
$sender
->get_heap();
return
if
$nick
ne
$irc
->nick_name();
pass(
'Joined channel'
);
$heap
->{joined}++;
return
if
$heap
->{joined} != 2;
$bot2
->yield(
privmsg
=>
$where
,
"TestBot1: help"
);
$bot2
->yield(
privmsg
=>
$where
,
"TestBot1: help foo"
);
}
sub
irc_notice {
my
(
$sender
,
$heap
,
$who
,
$where
,
$what
) =
@_
[SENDER, HEAP, ARG0..ARG2];
my
$irc
=
$sender
->get_heap();
my
$nick
= (
split
/!/,
$who
)[0];
return
if
$irc
!=
$bot2
;
$heap
->{replies}++;
if
(
$heap
->{replies} == 1) {
is(
$nick
,
$bot1
->nick_name(),
'Bot nickname'
);
like(
$what
,
qr/^No commands/
,
'Bot reply'
);
}
elsif
(
$heap
->{replies} == 2) {
is(
$nick
,
$bot1
->nick_name(),
'Bot nickname'
);
like(
$what
,
qr/^Unknown command:/
,
'Bot reply'
);
my
(
$p
) =
grep
{
$_
->isa(
'POE::Component::IRC::Plugin::BotCommand'
) }
values
%{
$bot1
->plugin_list() };
ok(
$p
->add(
foo
=>
'Test command'
),
'Add command foo'
);
$irc
->yield(
privmsg
=>
$where
,
"TestBot1: hlagh"
);
}
elsif
(
$heap
->{replies} == 4) {
is(
$nick
,
$bot1
->nick_name(),
'Bot nickname'
);
like(
$what
,
qr/^You are not authorized/
,
'Bot reply'
);
$bot1
->yield(
'quit'
);
$bot2
->yield(
'quit'
);
}
}
sub
irc_disconnected {
my
(
$kernel
,
$heap
) =
@_
[KERNEL, HEAP];
pass(
'irc_disconnected'
);
$heap
->{count}++;
$poe_kernel
->yield(
'_shutdown'
)
if
$heap
->{count} == 2;
}