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
|
#!/usr/bin/perl
# Test activity on idle connections in the pool.
use warnings;
use strict;
use lib qw(./mylib ../mylib);
use Test::More tests => 5;
sub POE::Kernel::ASSERT_DEFAULT () { 1 }
use POE;
use POE::Component::Client::Keepalive;
use POE::Component::Resolver;
use Socket qw(AF_INET);
use TestServer;
my $server_port = TestServer->spawn(0);
POE::Session->create(
inline_states => {
_child => sub { },
_start => \&start,
_stop => sub { },
check_for_input => \&check_for_input,
got_conn => \&got_conn,
got_conn2 => \&got_conn2,
got_error => \&got_error,
got_input => \&got_input,
got_timeout => \&got_timeout,
shutdown_server => \&shutdown_server,
}
);
# Start the connection manager, and allocate a connection to our test
# server.
sub start {
my $heap = $_[HEAP];
$heap->{cm} = POE::Component::Client::Keepalive->new(
resolver => POE::Component::Resolver->new(af_order => [ AF_INET ]),
);
$heap->{cm}->allocate(
scheme => "http",
addr => "localhost",
port => $server_port,
event => "got_conn",
context => "first",
);
}
# A connection has been allocated.
# Tell the test server to send us something.
# Discard the connection before we can retrieve from it.
sub got_conn {
my ($heap, $stuff) = @_[HEAP, ARG0..$#_];
my $conn = $stuff->{connection};
my $which = $stuff->{context};
ok(defined($conn), "$which connection established asynchronously");
ok(not (defined ($stuff->{from_cache})), "$which connection request deferred");
TestServer->send_something();
$_[KERNEL]->delay(check_for_input => 1);
# The connection goes free when it drops out of scope here.
# Everything that was sent to it remains unread.
}
# Reallocate the free socket. It should be asynchronous because there
# was data on the socket and the connection could not be reused.
sub check_for_input {
my ($kernel, $heap) = @_[KERNEL, HEAP];
$heap->{cm}->allocate(
scheme => "http",
addr => "localhost",
port => $server_port,
event => "got_conn2",
context => "first",
);
$kernel->delay(shutdown_server => 1);
}
sub got_conn2 {
my ($kernel, $heap, $stuff) = @_[KERNEL, HEAP, ARG0..$#_];
$heap->{conn} = $stuff->{connection};
is(
$stuff->{from_cache}, undef,
"second connection established assynchronously"
);
$heap->{conn}->start(
InputEvent => "got_input",
);
ok(defined($heap->{conn}->wheel()), "connection contains a wheel");
}
sub got_input {
$_[HEAP]->{got_input} = 1;
}
sub shutdown_server {
my ($kernel, $heap) = @_[KERNEL, HEAP];
ok(!$heap->{got_input}, "didn't receive any input");
delete $heap->{conn};
TestServer->shutdown();
$heap->{cm}->shutdown();
}
POE::Kernel->run();
exit;
|