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
|
use strict;
use warnings;
use lib 't/inc';
use POE qw(Wheel::SocketFactory);
use POE::Component::IRC;
use POE::Component::Server::IRC;
use Socket;
use Test::More tests => 14;
my $bot1 = POE::Component::IRC->spawn(Flood => 1);
my $bot2 = POE::Component::IRC->spawn(Flood => 1);
my $ircd = POE::Component::Server::IRC->spawn(
Auth => 0,
AntiFlood => 0,
);
POE::Session->create(
package_states => [
main => [qw(
_start
_config_ircd
_shutdown
irc_registered
irc_connected
irc_001
irc_error
irc_disconnected
irc_shutdown
)],
],
);
$poe_kernel->run();
sub _start {
my ($kernel) = $_[KERNEL];
my $ircd_port = get_port() or $kernel->yield(_shutdown => 'No free port');
$kernel->yield(_config_ircd => $ircd_port);
$kernel->delay(_shutdown => 60, 'Timed out');
}
sub get_port {
my $wheel = POE::Wheel::SocketFactory->new(
BindAddress => '127.0.0.1',
BindPort => 0,
SuccessEvent => '_fake_success',
FailureEvent => '_fake_failure',
);
return if !$wheel;
return unpack_sockaddr_in($wheel->getsockname()) if wantarray;
return (unpack_sockaddr_in($wheel->getsockname))[0];
}
sub _config_ircd {
my ($kernel, $heap, $session, $port) = @_[KERNEL, HEAP, SESSION, ARG0];
$ircd->yield(add_listener => Port => $port);
$kernel->signal($kernel, 'POCOIRC_REGISTER', $session, 'all');
$heap->{nickcounter} = 0;
$heap->{port} = $port;
}
sub irc_registered {
my ($heap, $irc) = @_[HEAP, ARG0];
$heap->{nickcounter}++;
pass('Registered ' . $heap->{nickcounter});
isa_ok($irc, 'POE::Component::IRC');
$irc->yield(connect => {
nick => 'TestBot' . $heap->{nickcounter},
server => '127.0.0.1',
port => $heap->{port},
});
}
sub irc_connected {
pass('Connected');
}
sub irc_001 {
my ($sender,$text) = @_[SENDER, ARG1];
my $irc = $sender->get_heap();
pass('Logged in');
$irc->yield('quit');
}
sub irc_error {
pass('irc_error');
}
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();
$kernel->signal($kernel, 'POCOIRC_SHUTDOWN');
$ircd->yield('shutdown');
}
sub irc_shutdown {
pass('irc_shutdown');
}
|
