1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141
|
# This make sures that we can close a DCC connection right after sending
# some data over it. The original bug was that the DCC plugin didn't post
# a delayed close event correctly so it ended up checking if there was data
# left to be sent on an undefined value rather than the wheel in question.
use strict;
use warnings FATAL => 'all';
use lib 't/inc';
use POE::Component::IRC;
use POE::Component::Server::IRC;
use POE;
use Test::More tests => 12;
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_disconnected
irc_dcc_request
irc_dcc_done
irc_dcc_chat
irc_dcc_start
)],
],
);
$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 irc_001 {
my $irc = $_[SENDER]->get_heap();
pass('Logged in');
$irc->yield(join => '#testchannel');
}
sub irc_join {
my ($sender, $who, $where) = @_[SENDER, ARG0, ARG1];
my $nick = ( split /!/, $who )[0];
my $irc = $sender->get_heap();
if ($nick eq $irc->nick_name()) {
is($where, '#testchannel', 'Joined Channel Test');
if ($nick eq 'TestBot2') {
$irc->yield(dcc => TestBot1 => CHAT => '' => '' => 5);
}
}
}
sub irc_dcc_request {
my ($sender, $cookie) = @_[SENDER, ARG3];
pass('Got dcc request');
$sender->get_heap()->yield(dcc_accept => $cookie);
}
sub irc_dcc_start {
my ($sender, $id) = @_[SENDER, ARG0];
my $irc = $sender->get_heap();
pass('DCC started');
if ($irc->nick_name() eq 'TestBot2') {
$irc->yield(dcc_chat => $id => 'MOO');
$irc->yield(dcc_close => $id);
}
}
sub irc_dcc_chat {
my ($sender, $what) = @_[SENDER, ARG3];
is($what, 'MOO', 'DCC CHAT test');
}
sub irc_dcc_done {
pass('Got dcc close');
$_[SENDER]->get_heap()->yield('quit');
}
sub irc_disconnected {
my ($kernel, $heap) = @_[KERNEL, HEAP];
pass('irc_disconnected');
$heap->{count}++;
$kernel->yield('_shutdown') if $heap->{count} == 2;
}
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');
}
|